用费舍尔信息引导优化,高效清除模型后门且不损失准确率。
Fisher Information guided Purification against Backdoor Attacks
- 基于费舍尔信息矩阵设计正则项,引导模型向平滑极小值优化
- 在14种攻击下保持90%以上清洁准确率,优于现有方法
- 快速版FIP提速近5倍,适合实际部署
近年来研究表明,攻击者可通过操纵少量训练样本破坏深度神经网络的完整性。我们分析发现,此类操作会使模型收敛至更尖锐的局部极小值,即不良极小值。直观上,通过重新优化使模型进入更平滑的极小值可实现后门净化。然而,直接采用任何追求平滑性的优化方法可能损害干净测试准确率。为此,受后门移除与损失函数平滑性关联的新视角启发,本文提出费舍尔信息引导净化(FIP)框架。FIP引入若干新型正则项,利用费舍尔信息矩阵知识,在净化过程中抑制后门影响并保留对干净数据分布的学习能力。此外,提出高效变体Fast FIP,显著减少可调参数,实现接近5倍的运行时提升。大量实验表明,该方法在广泛基准上达到最先进性能:涵盖5类任务(图像识别、目标检测、视频动作识别、3D点云、语言生成)、11个数据集(如ImageNet、PASCAL VOC、UCF101)、多种架构(CNN与视觉变换器),以及14种攻击(如Dynamic、WaNet、LIRA、ISSBA等)。
原文摘要 · Abstract (English)
Studies on backdoor attacks in recent years suggest that an adversary can compromise the integrity of a deep neural network (DNN) by manipulating a small set of training samples. Our analysis shows that such manipulation can make the backdoor model converge to a bad local minima, i.e., sharper minima as compared to a benign model. Intuitively, the backdoor can be purified by re-optimizing the model to smoother minima. However, a naïve adoption of any optimization targeting smoother minima can lead to sub-optimal purification techniques hampering the clean test accuracy. Hence, to effectively obtain such re-optimization, inspired by our novel perspective establishing the connection between backdoor removal and loss smoothness, we propose Fisher Information guided Purification (FIP), a novel backdoor purification framework. Proposed FIP consists of a couple of novel regularizers that aid the model in suppressing the backdoor effects and retaining the acquired knowledge of clean data distribution throughout the backdoor removal procedure through exploiting the knowledge of Fisher Information Matrix (FIM). In addition, we introduce an efficient variant of FIP, dubbed as Fast FIP, which reduces the number of tunable parameters significantly and obtains an impressive runtime gain of almost $5\times$. Extensive experiments show that the proposed method achieves state-of-the-art (SOTA) performance on a wide range of backdoor defense benchmarks: 5 different tasks -- Image Recognition, Object Detection, Video Action Recognition, 3D point Cloud, Language Generation; 11 different datasets including ImageNet, PASCAL VOC, UCF101; diverse model architectures spanning both CNN and vision transformer; 14 different backdoor attacks, e.g., Dynamic, WaNet, LIRA, ISSBA, etc.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。