DBSegment: Fast and robust segmentation of deep brain structures considering domain generalization

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.

A Review of Causality for Learning Algorithms in Medical Image Analysis

“…On a similar note develop a causal inference-based method to search associations in genomics data from the UK Biobank. Additionally, (Baniasadi et al, 2022) employ causal analysis to explain the performance of their brain structure segmentation network. Similarly (Singla et al, 2021) employ mediation analysis to identify the units and parameters of radiological reports that influence their classifier's outcomes; this method is applied on chest X-rays.…”

Untitled

“…On a similar note develop a causal inference-based method to search associations in genomics data from the UK Biobank. Additionally, (Baniasadi et al, 2022) employ causal analysis to explain the performance of their brain structure segmentation network. Similarly (Singla et al, 2021) employ mediation analysis to identify the units and parameters of radiological reports that influence their classifier's outcomes; this method is applied on chest X-rays.…”