NA-MIC Project Weeks
Back to Projects List
Multi-organ Segmentation in Fetal Mice
Key Investigators
- Murat Maga (Seattle Children’s Research Institute, UW)
- Sara Rolfe (Seattle Children’s Research Institute, UW)
- Andres Diaz-Pinto (Kings College, Nvidia)
Project Description
Objective
- We would like to assess whether MONAILabel can segment internal organs of fetal mice in similar quality to the deformable atlas-based registration.
- Evaluate pros and cons of both approach.
Approach and Plan
- 3D Fetal mice data comes from International Mouse Phenotyping Consortium. VOlumes are approximately 210x250x400 voxels.
- E15 Fetal Mice atlas comes from. There are a total of 50 unique labels (plus background).
- We used R implementation of ANTs to transfer atlas labels to individual mice using a two-pass deformable registration (initial low-res with SyN + MI, and high-resolution used SyN + CC). It takes approximately 4h to register one sample using 24 threads.
- Reviewed the transfered labels on subject space (all resultant templates considered good enough to be used as is, without manual editing
- Edited the MultiLabel version of DeepEdit to match our label names.
- Used 86 such labeled volumes for training data in MonaiLabel with default 0.8/0.2 split. We retained 14 additional samples with labels for inference tests.
Progress and Next Steps
- Initial model training for mouse embryo segmentation was done a spatial resolution of 128x128x128. Training carried out to 500 epochs. While successeful, there wasn’t sufficient anatomical detail due to low spatial resolution of model.
- Edited the spatial resolution to 192x192x192 (highest we can fit on Nvidia A6000, with 48GB of RAM). Used the initial model weights, and rerun the tranining for further 100 epochs. This improved the detail considerably. We also achievied higher training dice scores on this higher resolution modes.
- Currently running inference on unseen data. Inference takes approximately one-minute when monaiserver is local to the Slicer application.
- Average DICE coefficient of 0.81 between inferred reference labels registered using Syn + CC. Scores similar to other state-of-the-art methods for individual organ labels.
Illustrations
