Code for the paper 'Efficient and Effective Model Extraction' accepted at IEEE ICASSP 2025.
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.