arXiv:2602.20418cs.LG2026-02被引 4

提出CITED方法,用决策边界感知签名防御GNN模型盗取

CITED: A Decision Boundary-Aware Signature for GNNs Towards Model Extraction Defense

  • 基于决策边界设计新型签名,同时保护嵌入和标签输出
  • 在多种攻击下仍保持高验证准确率,且不影响下游任务性能
  • 无需额外模型或计算开销,适合实际部署的GNN服务

图神经网络(GNN)在推荐系统和金融风险管控等场景中表现优异。然而,将大规模GNN模型本地部署对用户而言面临巨大计算资源与数据需求挑战,因此机器学习即服务(MLaaS)日益流行,使用户可便捷访问各类模型,包括GNN。但随之而来的模型提取攻击(MEAs)威胁显著:攻击者通过反复查询子图输入并获取响应,构建低成本的替代模型。现有防御方法多局限于特定输出层级(如嵌入或标签),且存在技术缺陷。为此,本文提出首个在嵌入与标签双层面实现所有权验证的框架CITED。该方法为基于签名的新型防御机制,不损害下游性能,也无需引入辅助模型以保证效率,且优于所有已有水印与指纹方案。大量实验验证了其有效性和鲁棒性。代码已开源:https://github.com/LabRAI/CITED。

原文摘要 · Abstract (English)

Graph neural networks (GNNs) have demonstrated superior performance in various applications, such as recommendation systems and financial risk management. However, deploying large-scale GNN models locally is particularly challenging for users, as it requires significant computational resources and extensive property data. Consequently, Machine Learning as a Service (MLaaS) has become increasingly popular, offering a convenient way to deploy and access various models, including GNNs. However, an emerging threat known as Model Extraction Attacks (MEAs) presents significant risks, as adversaries can readily obtain surrogate GNN models exhibiting similar functionality. Specifically, attackers repeatedly query the target model using subgraph inputs to collect corresponding responses. These input-output pairs are subsequently utilized to train their own surrogate models at minimal cost. Many techniques have been proposed to defend against MEAs, but most are limited to specific output levels (e.g., embedding or label) and suffer from inherent technical drawbacks. To address these limitations, we propose a novel ownership verification framework CITED which is a first-of-its-kind method to achieve ownership verification on both embedding and label levels. Moreover, CITED is a novel signature-based method that neither harms downstream performance nor introduces auxiliary models that reduce efficiency, while still outperforming all watermarking and fingerprinting approaches. Extensive experiments demonstrate the effectiveness and robustness of our CITED framework. Code is available at: https://github.com/LabRAI/CITED.

GNN安全模型防御签名验证

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。