Ze Zhang ¹ Zhen Wang ² Xiqian Wang ² Kuanquan Wang ¹ Yongfeng Yuan ¹ Qince Li ^1*

• ¹ Harbin Institute of Technology, Harbin, China
• ² Zibo Central Hospital, Shandong, China

Automatic segmentation of the left atrium (LA) constitutes a crucial pre-processing step in evaluating heart structure and function during clinical interventions, such as image-guided radiofrequency ablation of atrial fibrillation. Despite prior research on LA segmentation, the low contrast in medical images exacerbates the challenge of distinguishing various tissues, rendering accurate boundary delineation of the target area formidable. Moreover, class imbalance due to the small target size further complicates segmentation. This study aims to devise an architecture that augments edge information for LA segmentation from late gadolinium enhancement magnetic resonance imaging. To intensify edge information within image features, this study introduces an Edge Information Enhancement Module (EIEM) to the foundational network. The design of EIEM is grounded in exploring edge details within target region features learned from images. Additionally, it incorporates a Spatially Weighted Cross-Entropy loss function tailored for EIEM, introducing constraints on different regions based on the importance of pixels to edge segmentation, while also mitigating class imbalance through weighted treatment of positive and negative samples. The proposed method is validated on the 2018 Atrial Segmentation Challenge dataset. Compared with other state-of-the-art algorithms, the proposed algorithm demonstrated a significant improvement with an average symmetric surface distance of 0.684mm and achieved a commendable Dice coefficient of 0.924, implicating the effectiveness of enhancing edge information. The method offers a practical framework for precise LA localization and segmentation, particularly strengthening the algorithm's effectiveness in improving segmentation outcomes for irregular protrusions and discrete multiple targets. Additionally, the generalizability of our model was evaluated on the heart dataset from the Medical Segmentation Decathlon (MSD) challenge, confirming its robustness across different clinical scenarios involving LA segmentation.