arXiv:2410.04916cs.LGcs.AI2024-10被引 2

无需模型细节,用子图信息抵御图神经网络后门攻击

Defense-as-a-Service: Black-box Shielding against Backdoored Graph Models

  • 通过特征-拓扑聚类识别并剔除异常子图(触发器)
  • 在六大数据集上将后门攻击成功率降至10%以下,常规准确率不受损
  • 适合隐私受限场景下使用第三方模型的业务方

随着大规模图学习模型的发展,企业常依赖第三方提供的模型服务。但这些模型可能被植入后门,恶意用户可通过含触发器的输入操纵预测结果。现有防御方法存在依赖模型细节、需微调模型或依赖可解释性工具等局限,难以在严格隐私政策下应用。为此,我们提出GraphProt,一种仅依赖输入图的模型无关防御机制。其核心思想是利用子图信息进行预测,从而削弱触发器引发的后门影响。GraphProt包含两部分:基于特征-拓扑聚类的触发器消除与鲁棒子图集成。我们设计了基于该聚类的子图采样策略,通过多数投票构建稳健分类器。在三种后门攻击和六个基准数据集上的实验表明,GraphProt显著降低后门攻击成功率,同时保持正常任务的分类准确率。

原文摘要 · Abstract (English)

With the trend of large graph learning models, business owners tend to employ a model provided by a third party to deliver business services to users. However, these models might be backdoored, and malicious users can submit trigger-embedded inputs to manipulate the model predictions. Current graph backdoor defenses have several limitations: 1) depending on model-related details, 2) requiring additional model fine-tuning, and 3) relying upon extra explainability tools, all of which are infeasible under stringent privacy policies. To address those limitations, we propose GraphProt, which allows resource-constrained business owners to rely on third parties to avoid backdoor attacks on GNN-based graph classifiers. Our GraphProt is model-agnostic and only relies on the input graph. The key insight is to leverage subgraph information for prediction, thereby mitigating backdoor effects induced by triggers. GraphProt comprises two components: clustering-based trigger elimination and robust subgraph ensemble. Specifically, we first propose feature-topology clustering that aims to remove most of the anomalous subgraphs (triggers). Moreover, we design subgraph sampling strategies based on feature-topology clustering to build a robust classifier via majority vote. Experimental results across three backdoor attacks and six benchmark datasets demonstrate that GraphProt significantly reduces the backdoor attack success rate while preserving the model accuracy on regular graph classification tasks.

图神经网络后门攻击防御机制隐私保护

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