Efficient and Effective Model Extraction

Code for the paper "Efficient and Effective Model Extraction" accepted at IEEE ICASSP 2025.

Usage

The core implementation is now available. Experimental parameters on CIFAR-10 are listed in the corresponding .py files.

To run E3:

First, execute query_selection.py to generate the query set from the candidate out-of-distribution (OOD) set.
Then, run model_extraction.py to perform the extraction attack.
Finally, if desired, execute distribution_alignment.py to apply test-time distribution alignment.