Multi-Task Learning for Left Atrial Segmentation on GE-MRI

Segmentation of the left atrium (LA) is crucial for assessing its anatomy in both pre-operative atrial fibrillation (AF) ablation planning and post-operative follow-up studies. In this paper, we present a fully automated framework for left atrial segmentation in gadolinium-enhanced magnetic resonance images (GE-MRI) based on deep learning. We propose a fully convolutional neural network and explore the benefits of multi-task learning for performing both atrial segmentation and pre/post ablation classification. Our results show that, by sharing features between related tasks, the network can gain additional anatomical information and achieve more accurate atrial segmentation, leading to a mean Dice score of 0.901 on a test set of 20 3D MRI images. Code of our proposed algorithm is available at this https URL.

[1]  Keith E. Muller,et al.  Contrast-limited adaptive histogram equalization: speed and effectiveness , 1990, [1990] Proceedings of the First Conference on Visualization in Biomedical Computing.

[2]  Bram van Ginneken,et al.  A survey on deep learning in medical image analysis , 2017, Medical Image Anal..

[3]  Mert R. Sabuncu,et al.  Robust Atlas-Based Segmentation of Highly Variable Anatomy: Left Atrium Segmentation , 2010, STACOM/CESC.

[4]  Leo Breiman,et al.  Bagging Predictors , 1996, Machine Learning.

[5]  V. Palanisamy,et al.  A Proficient Clustering Technique to Detect CSF Level in MRI Brain Images Using PSO Algorithm , 2013 .

[6]  Soumya Ghose,et al.  A Random Forest Based Classification Approach to Prostate Segmentation in MRI , 2012 .

[7]  José Jalife,et al.  2017 HRS/EHRA/ECAS/APHRS/SOLAECE expert consensus statement on catheter and surgical ablation of atrial fibrillation , 2012, Heart rhythm.

[8]  Jian Sun,et al.  Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition , 2014, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[9]  P. Kalavathi,et al.  MEDICAL IMAGE CONTRAST ENHANCEMENT BASED ON GAMMA CORRECTION , 2012 .

[10]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.

[11]  Jürgen Weese,et al.  Benchmark for Algorithms Segmenting the Left Atrium From 3D CT and MRI Datasets , 2015, IEEE Transactions on Medical Imaging.

[12]  Wolters Kluwer,et al.  Hrs/ehra/ecas Expert Consensus Statement on Catheter and Surgical Ablation of Atrial Fibrillation: Recommendations for Patient Selection, Procedural Techniques, Patient Management and Follow-up, Definitions, Endpoints, and Research Trial Design: a Report of the Heart Rhythm Society (hrs) Task Force , 2022 .

[13]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[14]  Karel J. Zuiderveld,et al.  Contrast Limited Adaptive Histogram Equalization , 1994, Graphics Gems.

[15]  Jason Weston,et al.  Curriculum learning , 2009, ICML '09.

[16]  Felix Lau,et al.  FastVentricle: Cardiac Segmentation with ENet , 2017, FIMH.

[17]  D. Rueckert,et al.  Automated analysis of atrial late gadolinium enhancement imaging that correlates with endocardial voltage and clinical outcomes: A 2-center study , 2013, Heart rhythm.

[18]  Geoffrey E. Hinton,et al.  Learning representations by back-propagating errors , 1986, Nature.

[19]  Nitish Srivastava,et al.  Dropout: a simple way to prevent neural networks from overfitting , 2014, J. Mach. Learn. Res..

[20]  Sergey Ioffe,et al.  Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift , 2015, ICML.