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

Authors

Abstract

<b>Background:</b> Cardiac magnetic resonance (MRI) images often exhibit low contrast, anatomical variability, and indistinct boundaries, particularly in the myocardium (MYO) and right ventricle (RV). These challenges can reduce the reliability of both manual and automated segmentation, highlighting the need for more robust and boundary-aware approaches. <b>Methods:</b> In this study, an Aspect-Aware Boundary-Resilient UNet3D (ABR-UNet3D) architecture is proposed for cardiac MRI segmentation. The model incorporates an Aspect-Aware Complementary Attention (AAC) module that combines multi-planar contextual information with a complementary gating mechanism to enhance boundary representation. The method was evaluated on the ACDC dataset under consistent training conditions. In addition to Dice Similarity Coefficient (DSC) and Intersection over Union (IoU), boundary-based metrics, including the 95th percentile Hausdorff Distance (HD95), Average Surface Distance (ASD), and Surface Dice, were employed. Furthermore, a five-fold cross-validation protocol and detailed ablation studies were conducted to assess robustness and analyze the contribution of individual AAC components. <b>Results:</b> The proposed method achieved a mean DSC of 0.9603 in single-run experiments on the ACDC dataset and showed consistent performance in anatomically challenging regions, particularly for RV and MYO segmentation. In addition, five-fold cross-validation experiments resulted in an average DSC of 0.952 ± 0.009 and IoU of 0.908 ± 0.012, indicating stable performance across different data splits within the evaluated dataset. Boundary-based metrics also showed improved surface agreement and lower boundary errors compared with the evaluated baseline models. Ablation studies further indicated that the combined use of multi-planar contextual information and complementary gating contributes more effectively to segmentation performance than the individual components used separately. <b>Conclusions:</b> The results suggest that the proposed ABR-UNet3D architecture provides a stable and competitive segmentation framework for cardiac MRI images within the scope of the ACDC dataset. By jointly modeling contextual information and boundary refinement, the method improves segmentation reliability in challenging regions while maintaining competitive and consistent performance with respect to existing approaches.

Tags

Topics