Non-Local Attention Learning for Medical Image Classification

Yang Wen,Leiting Chen,Haisheng Chen,Ximan Tang,Yu Deng,Yongbiao Chen,Chuan Zhou
DOI: https://doi.org/10.1109/icme51207.2021.9428267
2021-01-01
Abstract:In recent years, deep convolutional neural networks (CNNs) have been used with great success in medical image classification. However, within CNNs, the convolutional operation only considers localized regions and the stacking of pooling layers can lead to the loss of information about tiny lesions. In this paper, we propose a non-local attention learning method that models long-range dependencies between pixels to pre-serve global information and help CNNs better identify the tiny lesions. It consists of two main parts, the non-local attention module and the non-local visual context fusion module, one for improving global understanding of the visual scene and one for aggregating non-local visual features. These two modules are used in parallel with the CNN backbone as an auxiliary branch. We conducted extensive experiments on two public medical image datasets and showed that our model achieves the best performance in terms of accuracy, precision, and sensitivity on both datasets compared to other recent models.
What problem does this paper attempt to address?