arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.04819cs.CVcs.AI

Top-GAP:在 CNN 中整合尺寸先验以提升可解释性、鲁棒性并缓解偏差

Top-GAP: Integrating Size Priors in CNNs for more Interpretability, Robustness, and Bias Mitigation

Fraunhofer ITWM(弗劳恩霍夫信息技术与微系统研究所);Offenburg University(奥芬堡大学)

Lars Nieradzik, Henrike Stephani, Janis Keuper

AI总结:

提出 Top-GAP 正则化方法,通过约束 CNN 特征表示的空间尺寸,使其关注目标区域,从而提升可解释性、对抗鲁棒性、分布偏移性能和定位精度。

摘要:

本文介绍 Top-GAP,这是一种新颖的正则化技术,可增强卷积神经网络的可解释性和鲁棒性。通过约束所学特征表示的空间尺寸,我们的方法迫使网络关注最显著的图像区域,从而有效降低背景影响。利用对抗攻击和 Effective Receptive Field,我们表明 Top-GAP 将更多注意力引向目标像素而非背景,这带来了更强的可解释性和鲁棒性。在 CIFAR-10 上,使用 PGD(ε=8/255、20 次迭代)时,我们实现了超过 50% 的鲁棒准确率,同时保持原有的干净样本准确率。此外,面对分布偏移,准确率最高提升 5%。我们的方法还产生了更精确的目标定位:与 GradCAM 和 Recipro-CAM 等方法相比,Intersection over Union(IOU)最高提升 25%。

英文摘要:

This paper introduces Top-GAP, a novel regularization technique that enhances the explainability and robustness of convolutional neural networks. By constraining the spatial size of the learned feature representation, our method forces the network to focus on the most salient image regions, effectively reducing background influence. Using adversarial attacks and the Effective Receptive Field, we show that Top-GAP directs more attention towards object pixels rather than the background. This leads to enhanced interpretability and robustness. We achieve over 50% robust accuracy on CIFAR-10 with PGD $ε=\frac{8}{255}$ and $20$ iterations while maintaining the original clean accuracy. Furthermore, we see increases of up to 5% accuracy against distribution shifts. Our approach also yields more precise object localization, as evidenced by up to 25% improvement in Intersection over Union (IOU) compared to methods like GradCAM and Recipro-CAM.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑