arXiv:2501.13776cs.LG2025-01AAAI被引 1

提出跨火策略,用零训练修复图神经网络的位翻转攻击。

Crossfire: An Elastic Defense Framework for Graph Neural Networks Under Bit Flip Attacks

  • 结合哈希、诱饵与位级纠错,利用权重稀疏性修复被攻击模型
  • 在六大数据集上平均提升21.8%的恢复成功率,最多修复55位翻转
  • 无需重训练和标签数据,开销极小,适合实际部署

位翻转攻击(BFAs)是针对卷积神经网络的经典对抗攻击,近期已被扩展至图神经网络(GNNs),暴露出显著漏洞。现有防御方法无法有效应对,且缺乏可验证的恢复机制。本文首次评估了源自计算机视觉领域的蜜罐与哈希防御在GNN上的效果,发现其存在明显局限。为此,提出Crossfire:一种无需重训练、不依赖标签数据的混合防御框架,利用权重稀疏性,结合哈希、蜜罐与位级纠错,对分布外权重元素进行修正,以恢复网络完整性。在六个基准数据集上共2160次实验中,平均比现有方案高21.8%的恢复率,最高支持55位翻转;修复后预测质量提升10.85%。计算与存储开销相对于最简GNN模型可忽略。

原文摘要 · Abstract (English)

Bit Flip Attacks (BFAs) are a well-established class of adversarial attacks, originally developed for Convolutional Neural Networks within the computer vision domain. Most recently, these attacks have been extended to target Graph Neural Networks (GNNs), revealing significant vulnerabilities. This new development naturally raises questions about the best strategies to defend GNNs against BFAs, a challenge for which no solutions currently exist. Given the applications of GNNs in critical fields, any defense mechanism must not only maintain network performance, but also verifiably restore the network to its pre-attack state. Verifiably restoring the network to its pre-attack state also eliminates the need for costly evaluations on test data to ensure network quality. We offer first insights into the effectiveness of existing honeypot- and hashing-based defenses against BFAs adapted from the computer vision domain to GNNs, and characterize the shortcomings of these approaches. To overcome their limitations, we propose Crossfire, a hybrid approach that exploits weight sparsity and combines hashing and honeypots with bit-level correction of out-of-distribution weight elements to restore network integrity. Crossfire is retraining-free and does not require labeled data. Averaged over 2,160 experiments on six benchmark datasets, Crossfire offers a 21.8% higher probability than its competitors of reconstructing a GNN attacked by a BFA to its pre-attack state. These experiments cover up to 55 bit flips from various attacks. Moreover, it improves post-repair prediction quality by 10.85%. Computational and storage overheads are negligible compared to the inherent complexity of even the simplest GNNs.

图神经网络对抗攻击防御框架位翻转

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