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

Jignesh Chowdary, Pratheepan Yogarajah

Research output: Contribution to journalArticlepeer-review

2 Citations (Scopus)
75 Downloads (Pure)

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.
Original languageEnglish
Pages (from-to)1-11
Number of pages11
JournalIEEE Journal of Biomedical and Health Informatics
Early online date12 Apr 2023
DOIs
Publication statusPublished online - 12 Apr 2023

Bibliographical note

Publisher Copyright:
IEEE

Keywords

  • Feature extraction
  • Image segmentation
  • Breast
  • Mammography
  • Decoding
  • Data mining
  • Medical diagnostic imaging
  • Atrous convolutions
  • U-shaped networks
  • Breast mass segmentation
  • Digital Mammogram
  • Ultrasound
  • Medical image segmentation

Fingerprint

Dive into the research topics of 'EU-Net: Enhanced U-shaped Network for Breast Mass Segmentation'. Together they form a unique fingerprint.

Cite this