Comparing 3D, 2.5D, and 2D Approaches to Brain Image Segmentation
Avesta, A. E.; Hossain, S.; Lin, M.; Aboian, M.; Krumholz, H.; Aneja, S.
Show abstract
Deep-learning methods for auto-segmenting brain images either segment one slice of the image (2D), five consecutive slices of the image (2.5D), or an entire volume of the image (3D). Whether one approach is superior for auto-segmenting brain images is not known. We compared these three approaches (3D, 2.5D, and 2D) across three auto-segmentation models (capsule networks, UNets, and nnUNets) to segment brain structures. We used 3430 brain MRIs, acquired in a multi-institutional study, to train and test our models. We used the following performance metrics: segmentation accuracy, performance with limited training data, required computational memory, and computational speed during training and deployment. 3D, 2.5D, and 2D approaches respectively gave the highest to lowest Dice scores across all models. 3D models maintained higher Dice scores when the training set size was decreased from 3199 MRIs down to 60 MRIs. 3D models converged 20% to 40% faster during training and were 30% to 50% faster during deployment. However, 3D models require 20 times more computational memory compared to 2.5D or 2D models. This study showed that 3D models are more accurate, maintain better performance with limited training data, and are faster to train and deploy. However, 3D models require more computational memory compared to 2.5D or 2D models.
Matching journals
The top 8 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Automated Generation of Cerebral Blood Flow Maps Using Deep Learning and Multiple Delay Arterial Spin-Labelled MRI 96%
- Deep Learning Based Cardiac Cine Segmentation – Transfer Learning Application to 7T Ultrahigh-Field MRI 96%
- Prospective Motion Correction and Automatic Segmentation of Penetrating Arteries in Phase Contrast MRI at 7 T 95%
Similar papers in this journal
Similar papers in this journal
- Automated joint skull-stripping and segmentation with Multi-Task U-Net in large mouse brain MRI databases 96%
- A preliminary attempt to harmonize using physics-constrained deep neural networks for multisite and multiscanner MRI datasets (PhyCHarm) 96%
- Subcortical Segmentation of the Fetal Brain in 3D Ultrasound using Deep Learning 96%
Similar papers in this journal
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.