# MetaMedSeg: Volumetric Meta-learning for Few-Shot Organ Segmentation

arXiv: https://arxiv.org/abs/2109.09734 · HTML: https://mlsanity.com/paper/2109.09734/

## Key facts

- Cited by 6 papers in the map (data through 14 Oct 2026).
- Builds on 25 of its 30 references that are in the map.
- First posted on arXiv 18 Sep 2021.

## Builds on (most cited first, 25 of 25)

- [U-Net: Convolutional Networks for Biomedical Image Segmentation](https://mlsanity.com/paper/1505.04597/) (2015)
- [Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks](https://mlsanity.com/paper/1703.03400/) (2017)
- [Prototypical Networks for Few-shot Learning](https://mlsanity.com/paper/1703.05175/) (2017)
- [V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation](https://mlsanity.com/paper/1606.04797/) (2016)
- [On First-Order Meta-Learning Algorithms](https://mlsanity.com/paper/1803.02999/) (2018)
- [Generalised Dice overlap as a deep learning loss function for highly unbalanced segmentations](https://mlsanity.com/paper/1707.03237/) (2017)
- [PANet: Few-Shot Image Semantic Segmentation with Prototype Alignment](https://mlsanity.com/paper/1908.06391/) (2019)
- [A large annotated medical image dataset for the development and evaluation of segmentation algorithms](https://mlsanity.com/paper/1902.09063/) (2019)
- [Prior Guided Feature Enrichment Network for Few-Shot Segmentation](https://mlsanity.com/paper/2008.01449/) (2020)
- [Prototype Mixture Models for Few-shot Semantic Segmentation](https://mlsanity.com/paper/2008.03898/) (2020)
- [A Novel Focal Tversky loss function with improved Attention U-Net for lesion segmentation](https://mlsanity.com/paper/1810.07842/) (2018)
- [Few-Shot Segmentation Without Meta-Learning: A Good Transductive Inference Is All You Need?](https://mlsanity.com/paper/2012.06166/) (2020)
- [Self-Supervision with Superpixels: Training Few-shot Medical Image Segmentation without Annotation](https://mlsanity.com/paper/2007.09886/) (2020)
- [CRNet: Cross-Reference Networks for Few-Shot Segmentation](https://mlsanity.com/paper/2003.10658/) (2020)
- [TaskNorm: Rethinking Batch Normalization for Meta-Learning](https://mlsanity.com/paper/2003.03284/) (2020)
- [Inverse Distance Aggregation for Federated Learning with Non-IID Data](https://mlsanity.com/paper/2008.07665/) (2020)
- [On the Texture Bias for Few-Shot CNN Segmentation](https://mlsanity.com/paper/2003.04052/) (2020)
- [Few-shot 3D Multi-modal Medical Image Segmentation using Generative Adversarial Learning](https://mlsanity.com/paper/1810.12241/) (2018)
- [Differentiable Meta-learning Model for Few-shot Semantic Segmentation](https://mlsanity.com/paper/1911.10371/) (2019)
- [SimPropNet: Improved Similarity Propagation for Few-shot Image Segmentation](https://mlsanity.com/paper/2004.15014/) (2020)
- [Is Support Set Diversity Necessary for Meta-Learning?](https://mlsanity.com/paper/2011.14048/) (2020)
- [Few-Shot Microscopy Image Cell Segmentation](https://mlsanity.com/paper/2007.01671/) (2020)
- [COVID-19 detection from scarce chest x-ray image data using few-shot deep learning approach](https://mlsanity.com/paper/2102.06285/) (2021)
- [Instance-Level Meta Normalization](https://mlsanity.com/paper/1904.03516/) (2019)
- [Few-Shot Unsupervised Continual Learning through Meta-Examples](https://mlsanity.com/paper/2009.08107/) (2020)

## Built on by (most cited first, 6 of 6)

- [Self-Prompting Large Vision Models for Few-Shot Medical Image Segmentation](https://mlsanity.com/paper/2308.07624/) (2023)
- [A Systematic Review of Few-Shot Learning in Medical Imaging](https://mlsanity.com/paper/2309.11433/) (2023)
- [Self-Sampling Meta SAM: Enhancing Few-shot Medical Image Segmentation with Meta-Learning](https://mlsanity.com/paper/2308.16466/) (2023)
- [Knowing What to Label for Few Shot Microscopy Image Cell Segmentation](https://mlsanity.com/paper/2211.10244/) (2022)
- [Conformable Convolution for Topologically Aware Learning of Complex Anatomical Structures](https://mlsanity.com/paper/2412.20608/) (2024)
- [Locally Private Online Quantile Regression: Estimation and Inference](https://mlsanity.com/paper/2607.05312/) (2026)

## Abstract (arXiv)

The lack of sufficient annotated image data is a common issue in medical image segmentation. For some organs and densities, the annotation may be scarce, leading to poor model training convergence, while other organs have plenty of annotated data. In this work, we present MetaMedSeg, a gradient-based meta-learning algorithm that redefines the meta-learning task for the volumetric medical data with the goal to capture the variety between the slices. We also explore different weighting schemes for gradients aggregation, arguing that different tasks might have different complexity, and hence, contribute differently to the initialization. We propose an importance-aware weighting scheme to train our model. In the experiments, we present an evaluation of the medical decathlon dataset by extracting 2D slices from CT and MRI volumes of different organs and performing semantic segmentation. The results show that our proposed volumetric task definition leads to up to 30% improvement in terms of IoU compared to related baselines. The proposed update rule is also shown to improve the performance for complex scenarios where the data distribution of the target organ is very different from the source organs.
