Skip to main navigation Skip to search Skip to main content

Double Spatial Attention for Convolution Neural Networks

  • Hang Wei*
  • , Zulin Wang
  • , Gengxin Hua
  • , Yunfu Zhao
  • *Corresponding author for this work
  • Beihang University
  • CAS - Beijing Institute of Control Engineering

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

In recent years, attention technology has attracted a lot attention in improving the performance of convolution neural networks (CNNs). However, most existing methods mainly consider the global information including spatial and channel dimensions, which ignore the selectively augmenting the key features like human eyes. This paper proposes an efficient Double Spatial Attention (DSA) mechanism, which does not involve convolution operation, avoiding non-linear operation. DSA includes two parts: (1) calculating the spatial attention coefficients of foreground and background respectively; (2) dynamic pruning the redundant spatial pixels according to foreground and background attention coefficients. The proposed DSA module not only can effectively focus on the main features in feature maps to improve the performance, but also maximally remove the spatial feature map redundancy and reduces the redundant computation. Experiments show that our method could improve 0.03%~2.49% accuracy on MobileNetV2 without increasing additional parameters.

Original languageEnglish
Title of host publication2022 7th International Conference on Big Data Analytics, ICBDA 2022
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages149-153
Number of pages5
ISBN (Electronic)9781665479387
DOIs
StatePublished - 2022
Event7th International Conference on Big Data Analytics, ICBDA 2022 - Guangzhou, China
Duration: 4 Mar 20226 Mar 2022

Publication series

Name2022 7th International Conference on Big Data Analytics, ICBDA 2022

Conference

Conference7th International Conference on Big Data Analytics, ICBDA 2022
Country/TerritoryChina
CityGuangzhou
Period4/03/226/03/22

Keywords

  • neural network
  • pruning
  • spatial attention

Fingerprint

Dive into the research topics of 'Double Spatial Attention for Convolution Neural Networks'. Together they form a unique fingerprint.

Cite this