vix.ing · top · new · best · stats · spec

Mix-Pooling Strategy for Attention Mechanism

2022/08/22 by Shanshan Zhong, Zhong, Shanshan, Wushao Wen +3
Computer Science · Neuroscience · #Advanced Neural Network Applications #Brain Tumor Detection and Classification #FOS: Computer and information sciences #Machine Learning (cs.LG) #Machine Learning and ELM

paper · pdf · doi:10.48550/arxiv.2208.10322

openalex publication_date 2022/08/22 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

Recently many effective attention modules are proposed to boot the model performance by exploiting the internal information of convolutional neural networks in computer vision. In general, many previous works ignore considering the design of the pooling strategy of the attention mechanism since they adopt the global average pooling for granted, which hinders the further improvement of the performance of the attention mechanism. However, we empirically find and verify a phenomenon that the simple linear combination of global max-pooling and global min-pooling can produce pooling strategies that match or exceed the performance of global average pooling. Based on this empirical observation, we propose a simple-yet-effective attention module SPEM, which adopts a self-adaptive pooling strategy based on global max-pooling and global min-pooling and a lightweight module for producing the attention map. The effectiveness of SPEM is demonstrated by extensive experiments on widely-used benchmark datasets and popular attention networks.

Related