ABR-UNet3D: Aspect-Aware Boundary-Resilient Attention for Robust Cardiac MRI Segmentation

Serdar Akyel1, Zeki Cetinkaya2, Fatih Topaloglu3

  • 1Afyonkarahisar State Hospital, Ministry of Health, 03030 Afyonkarahisar, Türkiye.

Insights

A new Aspect-Aware Boundary-Resilient UNet3D (ABR-UNet3D) model improves cardiac MRI segmentation, especially in challenging right ventricle (RV) and myocardium (MYO) regions, by enhancing boundary detection.

Area of Science:

  • Medical Imaging
  • Artificial Intelligence
  • Biomedical Engineering

Background:

  • Cardiac MRI segmentation faces challenges due to low contrast and indistinct boundaries in myocardium (MYO) and right ventricle (RV).
  • Existing segmentation methods struggle with anatomical variability, impacting reliability.
  • Robust, boundary-aware approaches are crucial for accurate cardiac image analysis.

Purpose of the Study:

  • To introduce a novel Aspect-Aware Boundary-Resilient UNet3D (ABR-UNet3D) architecture for improved cardiac MRI segmentation.
  • To enhance the model's ability to capture contextual information and refine boundaries.
  • To evaluate the proposed method's performance and robustness on a standard dataset.

Main Methods:

  • Developed an Aspect-Aware Complementary Attention (AAC) module combining multi-planar context and complementary gating.
  • Implemented the ABR-UNet3D architecture for cardiac MRI segmentation.
  • Evaluated performance using Dice Similarity Coefficient (DSC), Intersection over Union (IoU), and boundary metrics (HD95, ASD, Surface Dice) on the ACDC dataset.
  • Conducted five-fold cross-validation and ablation studies to assess robustness and component contributions.

Main Results:

  • Achieved a mean DSC of 0.9603 in single-run experiments and 0.952 ± 0.009 in cross-validation on the ACDC dataset.
  • Demonstrated consistent performance in challenging RV and MYO segmentation.
  • Boundary metrics showed improved surface agreement and reduced errors compared to baseline models.
  • Ablation studies confirmed the synergistic benefit of AAC module components.

Conclusions:

  • The ABR-UNet3D architecture offers a stable and competitive framework for cardiac MRI segmentation.
  • Jointly modeling context and boundary refinement enhances segmentation reliability in difficult regions.
  • The proposed method achieves consistent and improved performance over existing approaches for cardiac MRI analysis.

Related Concept Videos