Region-based U-net for accelerated training and enhanced precision in deep brain segmentation

Mengyu Li,Magnus Magnusson,Thilo van Eimeren,Lotta M. Ellingsen
2024-03-14
Abstract:Segmentation of brain structures on MRI is the primary step for further quantitative analysis of brain diseases. Manual segmentation is still considered the gold standard in terms of accuracy; however, such data is extremely time-consuming to generate. This paper presents a deep learning-based segmentation approach for 12 deep-brain structures, utilizing multiple region-based U-Nets. The brain is divided into three focal regions of interest that encompass the brainstem, the ventricular system, and the striatum. Next, three region-based U-nets are run in parallel to parcellate these larger structures into their respective four substructures. This approach not only greatly reduces the training and processing times but also significantly enhances the segmentation accuracy, compared to segmenting the entire MRI image at once. Our approach achieves remarkable accuracy with an average Dice Similarity Coefficient (DSC) of 0.901 and 95% Hausdorff Distance (HD95) of 1.155 mm. The method was compared with state-of-the-art segmentation approaches, demonstrating a high level of accuracy and robustness of the proposed method.
Image and Video Processing,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to perform fast and high - precision segmentation of 12 deep brain structures in brain MRI images. Although traditional manual segmentation is regarded as the gold standard, it is very time - consuming; while the existing automatic segmentation methods can accelerate the processing time, there is still room for improvement in terms of accuracy and processing efficiency. For this reason, the paper proposes a multi - region - based 3D convolutional neural network (CNN) segmentation algorithm, especially for deep brain structures related to imaging biomarkers in Parkinson - plus syndromes. Specifically, the method in the paper divides the brain into three main regions - the brainstem region, the ventricular system region and the striatum region, and designs a U - Net model for each region respectively to achieve more efficient training and higher segmentation accuracy. This method not only reduces the training and processing time, but also improves the segmentation accuracy, especially when processing the entire MRI image. The experimental results show that this method performs excellently in both the average Dice similarity coefficient (DSC) and the 95% Hausdorff distance (HD95), reaching 0.901 and 1.155 mm respectively, significantly outperforming the existing patch - based methods and other state - of - the - art segmentation methods.