CEGA: A Cost-Effective Approach for Graph-Based Model Extraction and Acquisition
Zebin Wang, Menghan Lin, Bolin Shen, Ken Anderson, Molei Liu, Tianxi Cai, Yushun Dong
Abstract
Graph Neural Networks (GNNs) have demonstrated remarkable utility across diverse applications, and their growing complexity has made Machine Learning as a Service (MLaaS) a viable platform for scalable deployment. However, this accessibility also exposes GNN to serious security threats, most notably model extraction attacks (MEAs), in which adversaries strategically query a deployed model to construct a high-fidelity replica. In this work, we evaluate the vulnerability of GNNs to MEAs and explore their potential for cost-effective model acquisition in non-adversarial research settings. Importantly, adaptive node querying strategies can also serve a critical role in research, particularly when labeling data is expensive or timeconsuming. By selectively sampling informative nodes, researchers can train high-performing GNNs with minimal supervision, which is particularly valuable in domains such as biomedicine, where annotations often require expert input. To address this, we propose a node querying strategy tailored to a highly practical yet underexplored scenario, where bulk queries are prohibited, and only a limited set of initial nodes is available. Our approach iteratively refines the node selection mechanism over multiple learning cycles, leveraging historical feedback to improve extraction efficiency. Extensive experiments on benchmark graph datasets demonstrate our superiority over comparable baselines on accuracy,
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 9f011246-152c-43ab-ad57-aaa62369e6d7Cited by top-tier papers2
- CREDIT: Certified Ownership Verification of Deep Neural Networks Against Model Extraction AttacksBolin Shen, Zhan Cheng, Neil Gong, Fan Yao et al.ICML 2026 · 3 citations
- AGDN: Learning to Solve Traveling Salesman Problem with Anisotropic Graph Diffusion NetworkBolin Shen, Ziwei Huang, Zhiguang Cao, Yushun DongKDD 2026
Builds on18
- Stealing Machine Learning Models via Prediction APIsFlorian Tramèr, Fan Zhang, Ari Juels, Michael K. Reiter et al.USENIX Security 2016 · 2,088 citations
- ActiveThief: Model Extraction Using Active Learning and Unannotated Public DataSoham Pal, Yash Gupta, Aditya Shukla, Aditya Kanade et al.AAAI 2020 · 164 citations
- When Do Graph Neural Networks Help with Node Classification? Investigating the Homophily Principle on Node DistinguishabilitySitao Luan, Chenqing Hua, Minkai Xu, Qincheng Lu et al.NeurIPS 2023 · 118 citations
- Does GNN Pretraining Help Molecular Representation?Ruoxi Sun, Hanjun Dai, Adams Wei YuNeurIPS 2022 · 102 citations
- Model Stealing Attacks Against Inductive Graph Neural NetworksYun Shen, Xinlei He, Yufei Han, Yang ZhangS&P 2022 · 88 citations
Related papers
- ATOM: A Framework of Detecting Query-Based Model Extraction Attacks for Graph Neural NetworksZhan Cheng, Bolin Shen, Tianming Sha, Yuan Gao et al.KDD 2025 · 2 citations
- On Stealing Graph Neural Network ModelsMarcin Podhajski, Jan Dubinski, Franziska Boenisch, Adam Dziedzic et al.AAAI 2026
- Defending against Model Extraction for GNNs with Model ReprogrammingYan Wen, Zhenyi Wang, Heng HuangKDD 2026
- Securing Graph Neural Networks in MLaaS: A Comprehensive Realization of Query-based Integrity VerificationBang Wu, Xingliang Yuan, Shuo Wang, Qi Li et al.S&P 2024 · 13 citations
- Exploring Connections Between Active Learning and Model ExtractionVarun Chandrasekaran, Kamalika Chaudhuri, Irene Giacomelli, Somesh Jha et al.USENIX Security 2020
