SAP‐cGAN: Adversarial learning for breast mass segmentation in digital mammogram based on superpixel average pooling

Yamei Li,Guohua Zhao,Qian Zhang,Yusong Lin,Meiyun Wang
DOI: https://doi.org/10.1002/mp.14671
IF: 4.506
2021-01-10
Medical Physics
Abstract:<section class="article-section__content"><h3 class="article-section__sub-title section1"> Purpose</h3><p>Breast mass segmentation is a prerequisite step in the use of computer‐aided tools designed for breast cancer diagnosis and treatment planning. However, mass segmentation remains challenging due to the low contrast, irregular shapes, and fuzzy boundaries of masses. In this work, we propose a mammography mass segmentation model for improving segmentation performance.</p></section><section class="article-section__content"><h3 class="article-section__sub-title section1"> Methods</h3><p>We propose a mammography mass segmentation model called SAP‐cGAN, which is based on an improved conditional Generative Adversarial Network (cGAN). We introduce a superpixel average pooling layer into the cGAN decoder, which utilizes superpixels as a pooling layout to improve boundary segmentation. In addition, we adopt a multiscale input strategy to enable the network to learn scale‐invariant features with increased robustness. The performance of the model is evaluated with two public datasets: CBIS‐DDSM and INbreast. Moreover, ablation analysis is conducted to evaluate further the individual contribution of each block to the performance of the network.</p></section><section class="article-section__content"><h3 class="article-section__sub-title section1"> Results</h3><p>Dice and Jaccard scores of 93.37% and 87.57%, respectively, are obtained for the CBIS‐DDSM dataset. The Dice and Jaccard scores for the INbreast dataset are 91.54% and 84.40%, respectively. These results indicate that our proposed model outperforms current state‐of‐the‐art breast mass segmentation methods. The super‐pixel average pooling layer and multiscale input strategy has improved the Dice and Jaccard scores of the original cGAN by 7.8% and 12.79%, respectively.</p></section><section class="article-section__content"><h3 class="article-section__sub-title section1"> Conclusions</h3><p>Adversarial learning with the addition of a superpixel average pooling layer and multiscale input strategy can encourage the Generator network to generate masks with increased realism and improve breast mass segmentation performance through the minimax game between the Generator network and Discriminator network.</p></section>
radiology, nuclear medicine & medical imaging
What problem does this paper attempt to address?