MAS-UNet: a U-shaped network for prostate segmentation

YuQi Hong1, Zhao Qiu1, Huajing Chen2

  • 1School of Computer Science and Technology, Hainan University, Haikou, China.

PubMed

Insights

This study introduces an improved Attention UNet model for precise prostate MRI segmentation, crucial for early prostate cancer detection. The enhanced model significantly boosts segmentation accuracy in key prostate regions.

Area of Science:

  • Medical Imaging
  • Artificial Intelligence
  • Oncology

Background:

  • Prostate cancer poses a significant health risk to middle-aged and elderly men.
  • Accurate segmentation of prostate magnetic resonance imaging (MRI) is vital for cancer diagnosis.
  • Existing segmentation methods require further accuracy improvements.

Purpose of the Study:

  • To develop a novel, highly accurate prostate MRI segmentation model.
  • To enhance the Attention UNet architecture for superior performance.

Main Methods:

  • Proposed a modified Attention UNet incorporating Group Normalization (GN), dropout, and an Atrous Spatial Pyramid Pooling (ASPP) module.
  • Integrated channel attention into the attention gate module.
  • Utilized distinct output channels for segmenting different prostate regions.

Main Results:

  • The enhanced model achieved Dice scores of 0.807 for the transition zone and 0.907 for the peripheral zone.
  • Comparative experiments demonstrated superior performance over five existing UNet-based models.

Conclusions:

  • The proposed Attention UNet-based model offers improved accuracy for prostate MRI segmentation.
  • This advancement holds promise for more reliable prostate cancer diagnosis and assessment.

Related Concept Videos