中文

NAM:基于归一化的注意力模块

计算机视觉与模式识别 2021-11-25 v1 图像与视频处理

摘要

识别不那么显著的特征是模型压缩的关键。然而,这一点在革命性的注意力机制中尚未得到研究。在这项工作中,我们提出了一种新颖的基于归一化的注意力模块(NAM),该模块抑制不那么显著的权重。它对注意力模块施加权重稀疏性惩罚,从而使它们在保持相似性能的同时计算效率更高。在 ResNet 和 MobileNet 上与另外三种注意力机制的比较表明,我们的方法带来了更高的准确率。本文的代码可在 https://github.com/Christian-lyc/NAM 公开获取。

关键词

引用

@article{arxiv.2111.12419,
  title  = {NAM: Normalization-based Attention Module},
  author = {Yichao Liu and Zongru Shao and Yueyang Teng and Nico Hoffmann},
  journal= {arXiv preprint arXiv:2111.12419},
  year   = {2021}
}

备注

3 pages, 2 figures, 2 tables, 2 tables in the appendix