A Semantic-Wise Convolutional Neural Network Approach for 3-D Left Atrium Segmentation from Late Gadolinium Enhanced Magnetic Resonance Imaging

Abstract

Several studies suggest that the assessment of viable left atrial (LA) tissue is a relevant information to support catheter ablation in atrial fibrillation (AF). Late gadolinium enhanced magnetic resonance imaging (LGE MRI) is a new emerging technique which is employed for the non-invasive quantification of LA fibrotic tissue. The analysis of LGE MRI relies on manual tracing of LA boundaries. This procedure is time-consuming and prone to high inter-observer variability given the different degrees of observers’ experience, LA wall thickness and data resolution. Therefore, an automatic approach for the LA wall detection would be highly desirable. This work focuses on the design and development of a semantic-wise convolutional neural network based on the successful architecture U-Net (U-SWCNN). Batch normalization, early stopping and parameter initializers consistent with the activation functions chosen were used; a loss function based on the Dice coefficient was employed. The U-SWCNN was trained end-to-end with the 3-D data available from the 2018 Atrial Segmentation Challenge. The training was completed using 80 LGE MRI data and a post-processing step based on the 3-D morphology was then applied. After the post-processing step, the average Dice coefficient on the validation set (20 LGE MRI data) was 0.911, while on the test set (54 LGE MRI data) was 0.898.