跳到主要导航 跳到搜索 跳到主要内容

CPSAM: Channel and Position Squeeze Attention Module

  • Yuchen Gong
  • , Zhihao Gu*
  • , Zhenghao Zhang
  • , Lizhuang Ma
  • *此作品的通讯作者
  • Shanghai Jiao Tong University
  • East China Normal University

科研成果: 书/报告/会议事项章节会议稿件同行评审

摘要

In deep neural networks, how to model the remote dependency on time or space has always been a problem for scholars. By aggregatingpioneering method of capturing remote dependencies. However, the NL network faces many problems; 1) For different query positions in the image, the long-range dependency modeled by the NL network is quite similar so that it’s a wates of computation cost to build pixel-level pairwise relations. 2) The NL network only focuses on capturing spatial-wise lo a ng-range dependencies and neglects channel-wise attention. Therefore, in response to thesquery-specific global context of each query location, Non-Local (NL) networks propose e problems, we propose the Channel and Position Squeeze Attention Module (CPSAM). Specifically, for a feature map of the middle layer, our module infers attention maps along channel and spatial dimensions in parallel. The Channel Squeeze Attention Module selectively joins the feature of different position by a query-independent feature map. Meanwhile, the Position Squeeze Attention Module uses both avg and max pooling to compress the spatial dimension and Integrate the correlation characteristics between all channel maps. Finally, the outputs of two attention modules are combine together through the conv layer to further enhance feature representation. We have achieved higher accuracy and fewer parameters on the cifar100 and ImageNet1k compared to the NL network. The code will be publicly available soon.

源语言英语
主期刊名Neural Information Processing - 28th International Conference, ICONIP 2021, Proceedings
编辑Teddy Mantoro, Minho Lee, Media Anugerah Ayu, Kok Wai Wong, Achmad Nizar Hidayanto
出版商Springer Science and Business Media Deutschland GmbH
190-202
页数13
ISBN(印刷版)9783030921842
DOI
出版状态已出版 - 2021
活动28th International Conference on Neural Information Processing, ICONIP 2021 - Virtual, Online
期限: 8 12月 202112 12月 2021

出版系列

姓名Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
13108 LNCS
ISSN(印刷版)0302-9743
ISSN(电子版)1611-3349

会议

会议28th International Conference on Neural Information Processing, ICONIP 2021
Virtual, Online
时期8/12/2112/12/21

指纹

探究 'CPSAM: Channel and Position Squeeze Attention Module' 的科研主题。它们共同构成独一无二的指纹。

引用此