通过逐位翻转攻击GNN模型,仅改一个比特就让分类准确率降17%。
Bit-Flip Fault Attack: Crushing Graph Neural Networks via Gradual Bit Search
- 分层定位脆弱比特,逐步翻转以最小代价破坏GNN性能。
- 在Cora数据集上,单次比特翻转使GraphSAGE准确率下降17%。
- 适合研究GNN硬件安全的学者和对抗攻击防御设计者。
图神经网络(GNN)已成为处理图结构数据的强大机器学习方法,大量硬件加速器被引入以满足真实场景中的性能需求。然而,基于硬件的攻击安全性普遍被忽视。本文研究了GNN模型对硬件故障攻击的脆弱性,即攻击者通过内存设备注入故障修改训练好的权重参数,从而导致输出误分类。为此,我们提出渐进式比特翻转故障攻击(GBFA),一种分层感知的比特翻转攻击方法,通过逐个选择目标权重中的脆弱比特,在最少比特翻转下破坏GNN性能。GBFA分为两步:首先利用马尔可夫模型根据内存访问模式特征预测层执行顺序,实现特定层内攻击;随后在选定权重中通过梯度排序进行层内搜索,识别脆弱比特。我们在节点分类任务上评估了GBFA在Cora和PubMed数据集上的有效性。结果表明,GBFA显著降低预测准确率,且不同层间影响差异明显,凸显分层攻击策略的重要性。例如,在Cora数据集上,仅对最后一层进行一次比特翻转,GraphSAGE的准确率便下降17%。
原文摘要 · Abstract (English)
Graph Neural Networks (GNNs) have emerged as a powerful machine learning method for graph-structured data. A plethora of hardware accelerators has been introduced to meet the performance demands of GNNs in real-world applications. However, security challenges of hardware-based attacks have been generally overlooked. In this paper, we investigate the vulnerability of GNN models to hardware-based fault attack, wherein an attacker attempts to misclassify output by modifying trained weight parameters through fault injection in a memory device. Thus, we propose Gradual Bit-Flip Fault Attack (GBFA), a layer-aware bit-flip fault attack, selecting a vulnerable bit in each selected weight gradually to compromise the GNN's performance by flipping a minimal number of bits. To achieve this, GBFA operates in two steps. First, a Markov model is created to predict the execution sequence of layers based on features extracted from memory access patterns, enabling the launch of the attack within a specific layer. Subsequently, GBFA identifies vulnerable bits within the selected weights using gradient ranking through an in-layer search. We evaluate the effectiveness of the proposed GBFA attack on various GNN models for node classification tasks using the Cora and PubMed datasets. Our findings show that GBFA significantly degrades prediction accuracy, and the variation in its impact across different layers highlights the importance of adopting a layer-aware attack strategy in GNNs. For example, GBFA degrades GraphSAGE's prediction accuracy by 17% on the Cora dataset with only a single bit flip in the last layer.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。