GPCA: A Probabilistic Framework for Gaussian Process Embedded Channel Attention

Jiyang Xie,Zhanyu Ma,Dongliang Chang,Guoqiang Zhang,Jun Guo
DOI: https://doi.org/10.1109/tpami.2021.3102955
IF: 23.6
2021-01-01
IEEE Transactions on Pattern Analysis and Machine Intelligence
Abstract:Channel attention mechanisms have been commonly applied in many visual tasksfor effective performance improvement. It is able to reinforce the informativechannels as well as to suppress the useless channels. Recently, differentchannel attention modules have been proposed and implemented in various ways.Generally speaking, they are mainly based on convolution and poolingoperations. In this paper, we propose Gaussian process embedded channelattention (GPCA) module and further interpret the channel attention schemes ina probabilistic way. The GPCA module intends to model the correlations amongthe channels, which are assumed to be captured by beta distributed variables.As the beta distribution cannot be integrated into the end-to-end training ofconvolutional neural networks (CNNs) with a mathematically tractable solution,we utilize an approximation of the beta distribution to solve this problem. Tospecify, we adapt a Sigmoid-Gaussian approximation, in which the Gaussiandistributed variables are transferred into the interval [0,1]. The Gaussianprocess is then utilized to model the correlations among different channels. Inthis case, a mathematically tractable solution is derived. The GPCA module canbe efficiently implemented and integrated into the end-to-end training of theCNNs. Experimental results demonstrate the promising performance of theproposed GPCA module. Codes are available at https://github.com/PRIS-CV/GPCA.
What problem does this paper attempt to address?