arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2404.14996cs.CV

CA-Stream:用于可解释图像识别的基于注意力的池化

CA-Stream: Attention-based pooling for interpretable image recognition

Centrale Marseille(马赛中央理工学院);Aix Marseille Univ(艾克斯-马赛大学);CNRS(法国国家科学研究中心);Institute of Intelligent Software(智能软件研究所);Institute of Advanced Research on Artificial Intelligence (IARAI)(高级人工智能研究所(IARAI))

Felipe Torres, Hanwei Zhang, Ronan Sicre, Stéphane Ayache, Yannis Avrithis

AI总结:

本文提出CA-Stream,一种基于注意力的池化机制,通过交叉注意力块与不同深度特征交互来替代GAP,在保持识别性能的同时增强模型的可解释性。

摘要:

从基于transformer的架构中获得的原始注意力解释,可被视为类别不可知的显著性图。此外,基于注意力的池化在特征空间中充当一种掩蔽形式。受此观察启发,我们设计了一种基于注意力的池化机制,旨在推理时替代全局平均池化(GAP)。该机制称为交叉注意力流(CA-Stream),由一系列与不同网络深度的特征交互的交叉注意力块组成。CA-Stream增强了模型的可解释性,同时保持了识别性能。

英文摘要:

Explanations obtained from transformer-based architectures in the form of raw attention, can be seen as a class-agnostic saliency map. Additionally, attention-based pooling serves as a form of masking the in feature space. Motivated by this observation, we design an attention-based pooling mechanism intended to replace Global Average Pooling (GAP) at inference. This mechanism, called Cross-Attention Stream (CA-Stream), comprises a stream of cross attention blocks interacting with features at different network depths. CA-Stream enhances interpretability in models, while preserving recognition performance.

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

↑