让图神经网络更安全又高效,实现高精度与可证明鲁棒性兼顾
AuditVotes: Elevating Provable Defense for GNNs with Efficient Augmentation and Conditional Smoothing
- 通过图重连增强和条件平滑,提升随机平滑的鲁棒性
- 在Cora-ML上抗20条边攻击时,纯净准确率提升437.1%,认证准确率提升409.3%
- 适用于未见节点,可推广至图像等其他平滑方法
尽管图神经网络(GNN)取得进展,自适应攻击仍对其鲁棒性构成挑战。基于随机平滑的认证鲁棒性虽能提供可证明保证,但面临严重精度-鲁棒性权衡,限制实际应用。为此,我们提出AuditVotes,首个同时实现高纯净准确率与强认证鲁棒性的框架。该框架将两项新组件融入随机平滑流程:(1) 图重连增强,对随机化图进行去噪以恢复数据质量;(2) 条件平滑,过滤低置信度投票以确保预测一致性。我们建立了新理论结果,证明认证鲁棒性在任意过滤函数下仍被保留。本框架面向归纳学习,可泛化至未见节点,并广泛适配其他平滑方案,包括图的去随机平滑与图像的高斯平滑。大量实验表明,AuditVotes表现显著提升:在Cora-ML上面对20条边攻击时,纯净准确率提升437.1%,认证准确率提升409.3%,且运行时间与原始平滑相当。作为通用、高效的即插即用模块,AuditVotes为安全敏感场景中的实用可认证鲁棒GNN提供了可能。
原文摘要 · Abstract (English)
Despite advancements in Graph Neural Networks (GNNs), adaptive attacks continue to challenge their robustness. Certified robustness via randomized smoothing offers provable guarantees but suffers from a severe accuracy-robustness trade-off, limiting its practical use. To bridge this gap, we introduce AuditVotes, the first framework that simultaneously achieves high clean accuracy and strong certified robustness. AuditVotes seamlessly integrates two novel components into the randomized smoothing pipeline: (1) graph rewiring augmentation, which denoises randomized graphs to recover data quality, and (2) conditional smoothing, which filters low-confidence votes to ensure prediction consistency. We establish a novel theoretical result, proving that certified robustness is preserved under arbitrary filtering functions. Designed for inductive learning, our framework generalizes to unseen nodes and applies broadly to other smoothing schemes, including de-randomized smoothing for graphs and Gaussian smoothing for images. Extensive experiments show AuditVotes delivers substantial gains: on Cora-ML under 20-edge attacks, it improves clean accuracy by 437.1% and certified accuracy by 409.3%, while maintaining comparable runtime to vanilla smoothing. As a widely applicable and efficient plug-in, AuditVotes offers higher accuracy and stronger guarantees, enabling the practical and certifiably robust GNNs in security-sensitive domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。