arXiv:2503.00065cs.CRcs.LG2025-03中稿 · AsiaCCS 2026被引 2

提出首个主动防御GNN模型盗取的通用方案,通过监控查询多样性实时干扰输出。

ADAGE: Active Defenses Against GNN Extraction

  • 基于查询多样性监测,动态扰动输出以阻断模型泄露
  • 在6个数据集、4种模型上使攻击者无法完成盗取
  • 适合需安全共享高价值GNN模型的科研与工业场景

图神经网络(GNN)在药物发现、交通预测和推荐系统等实际应用中表现优异。由于构建高性能GNN需大量训练数据、算力及人力,使其成为模型盗取攻击的高价值目标。现有研究显示,攻击者可利用节点标签、高维节点嵌入等多种异构信号,在远低于原始训练成本下复现目标GNN。这种威胁向量的多样性使得有效且通用的防御设计极为困难,而现有防御多仅针对特定攻击场景,且仅能识别被盗模型而非阻止攻击。为此,本文提出首个通用的主动防御框架——ADAGE。其核心思想是:完整复制模型功能需覆盖输入空间的多样化查询以泄露行为。因此,ADAGE持续监控查询多样性,并在累积泄漏量增加时逐步扰动输出。实验在6个基准数据集、4种GNN模型及3类自适应攻击者上验证,结果表明ADAGE可将攻击成本提升至不可行水平,同时保持下游任务的预测性能。该工作为未来安全共享高价值GNN模型提供了可行路径。

原文摘要 · Abstract (English)

Graph Neural Networks (GNNs) achieve high performance in various real-world applications, such as drug discovery, traffic states prediction, and recommendation systems. The fact that building powerful GNNs requires a large amount of training data, powerful computing resources, and human expertise turns the models into lucrative targets for model stealing attacks. Prior work has revealed that the threat vector of stealing attacks against GNNs is large and diverse, as an attacker can leverage various heterogeneous signals ranging from node labels to high-dimensional node embeddings to create a local copy of the target GNN at a fraction of the original training costs. This diversity in the threat vector renders the design of effective and general defenses challenging and existing defenses usually focus on one particular stealing setup. Additionally, they solely provide means to identify stolen model copies rather than preventing the attack. To close this gap, we propose the first and general Active Defense Against GNN Extraction (ADAGE). ADAGE builds on the observation that stealing a model's full functionality requires highly diverse queries to leak its behavior across the input space. Our defense monitors this query diversity and progressively perturbs outputs as the accumulated leakage grows. In contrast to prior work, ADAGE can prevent stealing across all common attack setups. Our extensive experimental evaluation using six benchmark datasets, four GNN models, and three types of adaptive attackers shows that ADAGE penalizes attackers to the degree of rendering stealing impossible, whilst preserving predictive performance on downstream tasks. ADAGE, thereby, contributes towards securely sharing valuable GNNs in the future.

GNN安全模型窃取主动防御

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