Triplet attention fusion module: A concise and efficient channel attention module for medical image segmentation

Yanlin Wu,Guanglei Wang,Zhongyang Wang,Hongrui Wang,Yan Li
DOI: https://doi.org/10.1016/j.bspc.2022.104515
IF: 5.1
2022-12-29
Biomedical Signal Processing and Control
Abstract:Concise segmentation of medical images is vital for diagnosing and treating diseases. In recent years, convolutional neural networks (CNNs) have yielded satisfactory results in the field of medical image segmentation, and researchers are striving to improve the segmentation performance of CNNs by using the channel attention mechanism. However, most existing channel attention modules have numerous parameters and use a single compression and activation function in the process of attention realization. This limits the effect of attention promotion. To solve this problem, in this paper, we propose the triplet attention fusion (TAF) module. It combines direct and indirect mapping to achieve effective fusion of global-local information. In addition, this method has low computational complexity. Because of its lightweight and convenient advantages, the TAF module can be integrated into the existing semantic segmentation networks. The experimental results on ISIC-2018 and LiTS datasets show that TAF can greatly improve the performance of the network. Compared with other attention modules, it introduces fewer parameters. In addition, the performance of UNet with TAF module is close to or even surpasses that of the other state-of-the-art network models. It proves the superiority and effectiveness of our proposed module in medical image segmentation.
engineering, biomedical
What problem does this paper attempt to address?