1. DBSegment: Fast and robust segmentation of deep brain structures considering domain generalization
- Author
-
Mehri Baniasadi, Mikkel V. Petersen, Jorge Gonçalves, Andreas Horn, Vanja Vlasov, Frank Hertel, Andreas Husch, Baniasadi, Mehri [0000-0002-7709-0513], Apollo - University of Cambridge Repository, Fonds National de la Recherche - FnR [sponsor], Lundbeckfonden [sponsor], Emmy Noether stipend [sponsor], jascha fonden [sponsor], deutsche Forschungsgemeinschaft [sponsor], Berlin institute ion health [sponsor], 111 Project on computational intelligence and intelligent control [sponsor], and National institute of health [sponsor]
- Subjects
Radiological and Ultrasound Technology ,Multidisciplinary, general & others [C99] [Engineering, computing & technology] ,segmentation ,deep learning ,Reproducibility of Results ,Brain ,confounder ,Multidisciplinaire, généralités & autres [C99] [Ingénierie, informatique & technologie] ,Segmentation ,Deep Learning ,Neurology ,Image Processing, Computer-Assisted ,deep brain structures ,magnetic resonance imaging ,Humans ,Radiology, Nuclear Medicine and imaging ,Neurology (clinical) ,Neural Networks, Computer ,Anatomy ,Confounder ,MRI - Abstract
Segmenting deep brain structures from magnetic resonance images is important for patient diagnosis, surgical planning, and research. Most current state-of-the-art solutions follow a segmentation-by-registration approach, where subject magnetic resonance imaging (MRIs) are mapped to a template with well-defined segmentations. However, registration-based pipelines are time-consuming, thus, limiting their clinical use. This paper uses deep learning to provide a one-step, robust, and efficient deep brain segmentation solution directly in the native space. The method consists of a preprocessing step to conform all MRI images to the same orientation, followed by a convolutional neural network using the nnU-Net framework. We use a total of 14 datasets from both research and clinical collections. Of these, seven were used for training and validation and seven were retained for testing. We trained the network to segment 30 deep brain structures, as well as a brain mask, using labels generated from a registration-based approach. We evaluated the generalizability of the network by performing a leave-one-dataset-out cross-validation, and independent testing on unseen datasets. Furthermore, we assessed cross-domain transportability by evaluating the results separately on different domains. We achieved an average dice score similarity of 0.89 ± 0.04 on the test datasets when compared to the registration-based gold standard. On our test system, the computation time decreased from 43 min for a reference registration-based pipeline to 1.3 min. Our proposed method is fast, robust, and generalizes with high reliability. It can be extended to the segmentation of other brain structures. It is publicly available on GitHub, and as a pip package for convenient usage.
- Published
- 2023
- Full Text
- View/download PDF