ivadomed / model_seg_mouse-sc_wm-gm_t1

White and grey matter segmentation on T1-weighted exvivo mouse spinal cord
MIT License
2 stars 1 forks source link

Consider using ensemble/bagging for higher performance on prediction #27

Closed jcohenadad closed 1 year ago

jcohenadad commented 1 year ago

The idea would be to train several models, with randomized train/validation split, and then aggregate the various inferences.

Possibly relevant: https://colab.research.google.com/github/Project-MONAI/tutorials/blob/main/modules/cross_validation_models_ensemble.ipynb#scrollTo=ZWSGnqA12FgX

jcohenadad commented 1 year ago

With https://github.com/ivadomed/model_seg_mouse-sc_wm-gm_t1/commit/9c60213ed37a3617113856aa4b8bfdf4b7e2a708, I get better results with only 2 models (39-40, green) vs. 4 models (39-42, blue):

anim

Maybe an issue with the majority voting code?

Here are each individual predictions (one per model):

anim

Looking at the individual predictions, it does make sense that the 39-40 would produce better predictions in this case.

In light of these results, it might be more interesting to create another aggregation method that would take the max across segmentations (as opposed to majority vote).

jcohenadad commented 1 year ago

Five best models: 39, 40, 41, 42, 44

image