USENIX Security2024Top-tier venue
SoK: All You Need to Know About On-Device ML Model Extraction - The Gap Between Research and Practice
Tushar Nayan, Qiming Guo, Mohammed Alduniawi, Marcus Botacin, A. Selcuk Uluagac, Ruimin Sun
Abstract
On-device ML is increasingly used in different applications. It brings convenience to offline tasks and avoids sending userprivate data through the network. On-device ML models are valuable and may suffer from model extraction attacks from different categories. Existing studies lack a deep understanding of on-device ML model security, which creates a gap between research and practice. This paper provides a systematization approach to classify existing model extraction attacks and defenses based on different threat models. We evaluated well known research projects from existing work with real-world ML models, and discussed their reproducibility, computation complexity, and power consumption. We identified the challenges for research projects in wide adoption in practice. We also provided directions for future research in ML model extraction security.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers7
- Peering Inside the Black-Box: Long-Range and Scalable Model Architecture Snooping via GPU Electromagnetic Side-ChannelRui Xiao, Sibo Feng, Soundarya Ramesh, Jun Han et al.NDSS 2026 · 3 citations
- STIP: Three-Party Privacy-Preserving and Lossless Inference for Large Transformers in ProductionMu Yuan, Lan Zhang, Yihang Cheng, Miao-Hui Song et al.NDSS 2026 · 2 citations
- ASGARD: Protecting On-Device Deep Neural Networks with Virtualization-Based Trusted Execution EnvironmentsMyungsuk Moon, Minhee Kim, Joonkyo Jung, Dokyung SongNDSS 2025
- CRISP: An Efficient Cryptographic Framework for ML Inference Against Malicious ClientsXiaoyu Fang, Shihui Zheng, Lize GuNDSS 2026
- Understanding the Security Boundary of Obfuscation-based On-Device LLM ProtectionHanyi Zhou, Chenyang Li, Yuanzhe Pang, Ke Xu et al.CCS 2026
Builds on35
- Deep Learning with Differential PrivacyMartín Abadi, Andy Chu, Ian J. Goodfellow, H. Brendan McMahan et al.CCS 2016 · 7,620 citations
- Membership Inference Attacks Against Machine Learning ModelsReza Shokri, Marco Stronati, Congzheng Song, Vitaly ShmatikovS&P 2017 · 5,137 citations
- Stealing Machine Learning Models via Prediction APIsFlorian Tramèr, Fan Zhang, Ari Juels, Michael K. Reiter et al.USENIX Security 2016 · 2,088 citations
- Stealing Hyperparameters in Machine LearningBinghui Wang, Neil Zhenqiang GongS&P 2018 · 504 citations
- BERT Loses Patience: Fast and Robust Inference with Early ExitWangchunshu Zhou, Canwen Xu, Tao Ge, Julian J. McAuley et al.NeurIPS 2020 · 473 citations
Related papers
- Mind Your Weight(s): A Large-scale Study on Insufficient Machine Learning Model Protection in Mobile AppsZhichuang Sun, Ruimin Sun, Long Lu, Alan MisloveUSENIX Security 2021 · 101 citations
- DEMISTIFY: Identifying On-device Machine Learning Models Stealing and Reuse Vulnerabilities in Mobile AppsPengcheng Ren, Chaoshun Zuo, Xiaofeng Liu, Wenrui Diao et al.ICSE 2024 · 10 citations
- Model-less Is the Best Model: Generating Pure Code Implementations to Replace On-Device DL ModelsMingyi Zhou, Xiang Gao, Pei Liu, John Grundy et al.ISSTA 2024 · 4 citations
- ModelObfuscator: Obfuscating Model Information to Protect Deployed ML-Based SystemsMingyi Zhou, Xiang Gao, Jing Wu, John C. Grundy et al.ISSTA 2023 · 11 citations
- SAME: Sample Reconstruction against Model Extraction AttacksYi Xie, Jie Zhang, Shiqian Zhao, Tianwei Zhang et al.AAAI 2024 · 6 citations
