AM-AN: Adversarial Network Based on Attention Mechanism for Medical Image Segmentation

Qin Liu,Wenxin Hu,Jun Zheng
DOI: https://doi.org/10.18178/wcse.2020.06.030
2020-01-01
Abstract:Image segmentation is crucial to a series of medical applications. One of the current problems is that imbalanced pixel classes have a negative effect on the results of medical image segmentation. However, the method of employing weighted loss cannot address this problem well. Therefore, we are motivated to propose a novel method based on Generative Adversarial Networks (GANs) and attention mechanisms to train our segmentation model with effective loss functions. Firstly, the proposed model consists of the segmentor network and the critic network, which are trained by adversarial learning. Moreover, the attention module we used in each residual block takes not only channel attention but also spatial attention into consideration and makes the segmentor perform better. The overhead of computation and parameters costed by our attention module is negligible. In addition, our work extends other methods by means of using pixel-wise loss functions, which include multi-scale loss and binary cross entropy loss. Finally, our work presents the results of comparing three different methods on medical image segmentation and our method yields a higher performance than theirs on MICCAI PROMISE12, DSB2018 and the lung nodule datasets.
What problem does this paper attempt to address?