A Two-stage Method with a Shared 3D U-Net for Left Atrial Segmentation of Late Gadolinium-Enhanced MRI Images

Abstract

Objective: This study was aimed at validating the accuracy of a proposed algorithm for fully automatic 3D left atrial segmentation and to compare its performance with existing deep learning algorithms. Methods: A two-stage method with a shared 3D U-Net was proposed to segment the 3D left atrium. In this architecture, the 3D U-Net was used to extract 3D features, a two-stage strategy was used to decrease segmentation error caused by the class imbalance problem, and the shared network was designed to decrease model complexity. Model performance was evaluated with the DICE score, Jaccard index and Hausdorff distance. Results: Algorithm development and evaluation were performed with a set of 100 late gadolinium-enhanced cardiovascular magnetic resonance images. Our method achieved a DICE score of 0.918, a Jaccard index of 0.848 and a Hausdorff distance of 1.211, thus, outperforming existing deep learning algorithms. The best performance of the proposed model (DICE: 0.851; Jaccard: 0.750; Hausdorff distance: 4.382) was also achieved on a publicly available 2013 image data set. Conclusion: The proposed two-stage method with a shared 3D U-Net is an efficient algorithm for fully automatic 3D left atrial segmentation. This study provides a solution for processing large datasets in resource-constrained applications. Significance Statement: Studying atrial structure directly is crucial for comprehending and managing atrial fibrillation (AF). Accurate reconstruction and measurement of atrial geometry for clinical purposes remains challenging, despite potential improvements in the visibility of AF-associated structures with late gadolinium-enhanced magnetic resonance imaging. This difficulty arises from the varying intensities caused by increased tissue enhancement and artifacts, as well as variability in image quality. Therefore, an efficient algorithm for fully automatic 3D left atrial segmentation is proposed in the present study.