Self-Configuring Capsule Networks for Brain Image Segmentation
Avesta, A. E.; Hossain, S.; Aboian, M.; Krumholz, H.; Aneja, S.
Show abstract
When an auto-segmentation model needs to be applied to a new segmentation task, multiple decisions should be made about the pre-processing steps and training hyperparameters. These decisions are cumbersome and require a high level of expertise. To remedy this problem, I developed self-configuring CapsNets (scCapsNets) that can scan the training data as well as the computational resources that are available, and then self-configure most of their design options. In this study, we developed a self-configuring capsule network that can configure its design options with minimal user input. We showed that our self-configuring capsule netwrok can segment brain tumor components, namely edema and enhancing core of brain tumors, with high accuracy. Out model outperforms UNet-based models in the absence of data augmentation, is faster to train, and is computationally more efficient compared to UNet-based 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 95%
- Prospective Motion Correction and Automatic Segmentation of Penetrating Arteries in Phase Contrast MRI at 7 T 95%
- Deep Learning Based Cardiac Cine Segmentation – Transfer Learning Application to 7T Ultrahigh-Field MRI 95%
Similar papers in this journal
Similar papers in this journal
- A preliminary attempt to harmonize using physics-constrained deep neural networks for multisite and multiscanner MRI datasets (PhyCHarm) 96%
- Automated joint skull-stripping and segmentation with Multi-Task U-Net in large mouse brain MRI databases 95%
- Subcortical Segmentation of the Fetal Brain in 3D Ultrasound using Deep Learning 95%
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.