Polarized Self-Attention: Towards High-quality Pixel-wise Mapping

July 2022

Polarized Self-Attention: Towards High-quality Pixel-wise Mapping

Authors:

Huajun Liu, Fuqiang Liu, Xinyi Fan, and Dong Huang

Abstract:

We address the pixel-wise mapping problem that commonly exists in the fine-grained computer vision tasks, such as estimating keypoint heatmaps and segmentation masks. These tasks require, at low computation overheads, modeling the long-range dependencies among high-resolution inputs and estimating the highly nonlinear pixel-wise outputs. While the attention mechanism added to Deep Convolutional Neural Networks(DCNNs) can boost long-range dependencies, the element-specific attention, such as the Nonlocal block, is highly complex and noise-sensitive to learn, and most of the simplified attention blocks are designed for image-wise classification purposes and simply applied to pixel-wise tasks. In this paper, we present the Polarized Self-Attention(PSA) block targeting the high-quality pixel-wise mapping with: (1) Polarized filtering: keeping high internal resolution in both channel and spatial attention computation while completely collapsing input tensors along their counterpart dimensions. (2) Enhancement: composing non-linearity that directly fits the output distribution of typical pixel-wise mappings, such as the 2D Gaussian distribution (keypoint heatmaps), or the 2D Binormial distribution (binary segmentation masks). Experimental results show that PSA boosts standard baselines by 2-4 points, and boosts state-of-the-arts by 1-2 points on 2D pose estimation and semantic segmentation benchmarks. Codes are available at (https://github.com/DeLightCMU/PSA)
@article{Liu and Huang-2022-132503,
author = {Huajun Liu And Fuqiang Liu And Xinyi Fan And Dong Huang},
title = {Polarized Self-Attention: Towards High-quality Pixel-wise Mapping},
journal = {Proceedings of Neurocomputing},
year = {2022},
month = {July},
keywords = {Pixel-wise Mapping, Self-Attention, Polarization, Convolution},
}
Copyright notice: This material is presented to ensure timely dissemination of scholarly and technical work. Copyright and all rights therein are retained by authors or by other copyright holders. All persons copying this information are expected to adhere to the terms and constraints invoked by each author's copyright. These works may not be reposted without the explicit permission of the copyright holder.