3D shallow deep neural network for fast and precise segmentation of left atrium

Abstract

Knowledge of the underlying anatomy of the left atrium can promote improved diagnostic protocols and clinical interventions. Hence, an automatic segmentation of the left atrium on magnetic resonance imaging (MRI) can support diagnosis, treatment and surgery planning of heart. However, due to the small size of left atrium with respect to the whole MRI volume, accurate segmentation of left atrium is challenging. Most of the existing deep learning approaches are based on cropping or cascading networks. In this work, we present a novel deep learning architecture for the segmentation of left atrium from MRI volume which incorporates the residual learning based encoder-decoder network. We introduce a loss function and parameter adjustments to deal with the issue of class imbalance and unavailability of large medical imaging dataset. To facilitate the high quality segmentation, we present a three-dimensional multi-scale residual learning based architecture that maintains coarse and fine level features throughout the network. Experimental results have shown a considerable improvement in segmentation performance by surpassing the current benchmarks (especially the winner of Left Atrial Segmentation Challenge-2018) with fewer parameters compared to the state-of-the-art approaches, thus potentially supporting cardiac diagnosis and surgery without adding any extensive pre-processing of input volumes or any post-processing on the base network’s output.