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.pyto generate the query set from the candidate out-of-distribution (OOD) set. - Then, run
model_extraction.pyto perform the extraction attack. - Finally, if desired, execute
distribution_alignment.pyto apply test-time distribution alignment.