提出新方法ArcGen,让后门检测模型跨架构通用。
ArcGen: Generalizing Neural Backdoor Detection Across Diverse Architectures
- 用对齐层和损失函数提取与架构无关的特征。
- 在未见过的16896个模型上检测性能提升42.5%(AUC)。
- 适合需要跨模型部署检测能力的安全研究者。
后门攻击严重威胁深度学习模型的安全性与可靠性。现有基于学习的神经网络后门检测方法在未见架构上泛化能力差。本文分析根源并提出新方法ArcGen,旨在获取与架构无关的模型特征(即对齐特征)。不同于直接使用模型输出作为特征,弧生成引入额外对齐层处理特征,降低架构信息影响。设计两种对齐损失,强制具有相似后门行为但不同架构的模型特征在分布和样本层面一致。大规模实验涵盖16,896个模型,覆盖多种数据集、攻击方式和模型结构,结果表明该方法在未见架构上检测性能最高提升42.5%(如AUC指标)。代码已开源。
原文摘要 · Abstract (English)
Backdoor attacks pose a significant threat to the security and reliability of deep learning models. To mitigate such attacks, one promising approach is to learn to extract features from the target model and use these features for backdoor detection. However, we discover that existing learning-based neural backdoor detection methods do not generalize well to new architectures not seen during the learning phase. In this paper, we analyze the root cause of this issue and propose a novel black-box neural backdoor detection method called ArcGen. Our method aims to obtain architecture-invariant model features, i.e., aligned features, for effective backdoor detection. Specifically, in contrast to existing methods directly using model outputs as model features, we introduce an additional alignment layer in the feature extraction function to further process these features. This reduces the direct influence of architecture information on the features. Then, we design two alignment losses to train the feature extraction function. These losses explicitly require that features from models with similar backdoor behaviors but different architectures are aligned at both the distribution and sample levels. With these techniques, our method demonstrates up to 42.5% improvements in detection performance (e.g., AUC) on unseen model architectures. This is based on a large-scale evaluation involving 16,896 models trained on diverse datasets, subjected to various backdoor attacks, and utilizing different model architectures. Our code is available at https://github.com/SeRAlab/ArcGen.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。