EU-Net: Enhanced U-shaped Network for Breast Mass Segmentation.

Abstract

Segmentation of breast masses in digital mammograms is very challenging due to its complexity. The recent U-shaped encoder-decoder networks achieved remarkable performance in medical image segmentation. However, these networks have some limitations: a) The multi-scale context information is required to accurately segment mass but is not effectively extracted and utilized. b) The global context information is often ignored by the skip connection. To overcome these limitations and achieve better segmentation, we propose an Enhanced U-shaped Network (EU-Net). The proposed EU-Net comprises of 3 novel components: 1) dense-block, which is employed in the encoder and the decoder in place of convolutional layers to achieve the multi-scale features. 2) Multi-Scale Feature Extraction and Fusion, which is used in the junction between the encoder and the decoder for further extracting and fusing the multi-scale context information. 3) Skip Connection Reconstruction, which is inserted between the encoder and the decoder at each stage, to redesign the skip connection and emphasize the global context information. Extensive experimental results under different settings show that the proposed EU-Net achieves superior performances than the previous state-of-the-art segmentation models, and other existing approaches on IN-Breast and CBIS-DDSM mammogram datasets. The generalization ability of the proposed EU-Net is evidenced through cross-dataset and ternary dataset evaluation performance. In the ternary dataset evaluation, the model is trained and evaluated on the UDIAT breast ultrasound dataset without fine-tuning. The EU-Net achieves higher generalization performance in both evaluation experiments. These experiments collectively indicate the efficiency and high generalization ability of the proposed EU-Net.