Abstract:Widely adopted in modern Vision Transformer designs, Softmax attention can effectively capture long-range visual information; however, it incurs excessive computational cost when dealing with high-resolution inputs. In contrast, linear attention naturally enjoys linear complexity and has great potential to scale up to higher-resolution images. Nonetheless, the unsatisfactory performance of linear attention greatly limits its practical application in various scenarios. In this paper, we take a step forward to close the gap between the linear and Softmax attention with novel theoretical analyses, which demystify the core factors behind the performance deviations. Specifically, we present two key perspectives to understand and alleviate the limitations of linear attention: the injective property and the local modeling ability. Firstly, we prove that linear attention is not injective, which is prone to assign identical attention weights to different query vectors, thus adding to severe semantic confusion since different queries correspond to the same outputs. Secondly, we confirm that effective local modeling is essential for the success of Softmax attention, in which linear attention falls short. The aforementioned two fundamental differences significantly contribute to the disparities between these two attention paradigms, which is demonstrated by our substantial empirical validation in the paper. In addition, more experiment results indicate that linear attention, as long as endowed with these two properties, can outperform Softmax attention across various tasks while maintaining lower computation complexity. Code is available at <a class="link-external link-https" href="https://github.com/LeapLabTHU/InLine" rel="external noopener nofollow">this https URL</a>.

Linear Attention Mechanism: An Efficient Attention for Semantic Segmentation

EHANet: Efficient Hybrid Attention Network Towards Real-time Semantic Segmentation

Efficient Attention: Attention with Linear Complexities

Multistage Attention ResU-Net for Semantic Segmentation of Fine-Resolution Remote Sensing Images

A LiDAR Point Cloud Semantic Segmentation Algorithm Based on Attention Mechanism and Hybrid CNN-LSTM

Expectation-Maximization Attention Networks for Semantic Segmentation

Embedded Attention Network for Semantic Segmentation

An Empirical Study of Attention Networks for Semantic Segmentation

Multiattention Network for Semantic Segmentation of Fine-Resolution Remote Sensing Images

Channelized Axial Attention for Semantic Segmentation -- Considering Channel Relation within Spatial Attention for Semantic Segmentation

Bridging the Divide: Reconsidering Softmax and Linear Attention

Improving performance of deep learning models for 3D point cloud semantic segmentation via attention mechanisms

Lightweight Attention Network for Very High-Resolution Image Semantic Segmentation

Bilateral Network with Residual U-blocks and Dual-Guided Attention for Real-time Semantic Segmentation

MEDANet: More Efficient Dual Attention Network for Scene Segmentation

Threshold Attention Network for Semantic Segmentation of Remote Sensing Images

SegNeXt: Rethinking Convolutional Attention Design for Semantic Segmentation

CAA : Channelized Axial Attention for Semantic Segmentation.

Point Attention Network for Point Cloud Semantic Segmentation.

Partial Class Activation Attention for Semantic Segmentation

Research on Image Semantic Segmentation Based on Hybrid Cascade Feature Fusion and Detailed Attention Mechanism