FS-UNet: Mass segmentation in mammograms using an encoder-decoder architecture with feature strengthening

Jiande Pi,Yunliang Qi,Meng Lou,Xiaorong Li,Yiming Wang,Chunbo Xu,Yide Ma
DOI: https://doi.org/10.1016/j.compbiomed.2021.104800
IF: 7.7
2021-01-01
Computers in Biology and Medicine
Abstract:Breast mass segmentation in mammograms is still a challenging and clinically valuable task. In this paper, we propose an effective and lightweight segmentation model based on convolutional neural networks to automatically segment breast masses in whole mammograms. Specifically, we first developed feature strengthening modules to enhance relevant information about masses and other tissues and improve the representation power of low-resolution feature layers with high-resolution feature maps. Second, we applied a parallel dilated convolution module to capture the features of different scales of masses and fully extract information about the edges and internal texture of the masses. Third, a mutual information loss function was employed to optimise the accuracy of the prediction results by maximising the mutual information between the prediction results and the ground truth. Finally, the proposed model was evaluated on both available INbreast and CBIS-DDSM datasets, and the experimental results indicated that our method achieved excellent segmentation performance in terms of dice coefficient, intersection over union, and sensitivity metrics.
What problem does this paper attempt to address?