Skip-connected 3D DenseNet for volumetric infant brain MRI segmentation

• Published in: Biomedical signal processing and control 2019-09, Vol.54, p.101613, Article 101613
• Main Authors: Bui, Toan Duc, Shin, Jitae, Moon, Taesup
• Format: Article
• Language: English
• Subjects: DenseNet; Fully convolutional neural networks; Infant brain segmentation; skip-connection
• ISSN: 1746-8094; 1746-8108
• DOI: 10.1016/j.bspc.2019.101613

•3D fully convolutional densely connected network with skip connections from different dense blocks.•Efficiency of the bottleneck with compression (BC) and 2 × 2 ×2 convolution layers for infant brain segmentation.•State-of-the-art performance on the 6-month infant brain MRI segmentation dataset (iSeg-2017). Automatic 6-month infant brain tissue segmentation of magnetic resonance imaging (MRI) is still less accurate owing to the low intensity contrast among tissues. To tackle the problem, we introduce an accurate segmentation method for volumetric infant brain MRI built upon a densely connected network that achieves state-of-the-art accuracy. Specifically, we carefully design a fully convolutional densely connected network with skip connections such that the information from different levels of dense blocks can be directly combined to achieve highly accurate segmentation results. The proposed network, called 3D-SkipDenseSeg, exploits the advantage of the recently DenseNet for classification task and extends this to segment the 6-month infant brain tissue segmentation of magnetic resonance imaging (MRI). Experimental results demonstrate a competitive performance with regard to both segmentation accuracy and parameter efficiency of the proposed method over the existing methods; namely, the proposed 3D-SkipDenseSeg achieved the best dice similarity coefficient (DSC) of 90.37 ± 1.38% (WM), 92.27 ± 0.81% (GM), and 95.79 ± 0.54% (CSF) among the 21 participating teams in the 6-month infant brain dataset (iSeg-2017) and required only 10–30% of the parameters compared to similar deep learning-based methods.