Efficient few-shot medical image segmentation via self-supervised variational autoencoder. 2025

Yanjie Zhou, and Feng Zhou, and Fengjun Xi, and Yong Liu, and Yun Peng, and David E Carlson, and Liyun Tu
School of Artificial Intelligence, Beijing University of Posts and Telecommunications, Beijing, 100876, China.

Few-shot medical image segmentation typically uses a joint model for registration and segmentation. The registration model aligns a labeled atlas with unlabeled images to form initial masks, which are then refined by the segmentation model. However, inevitable spatial misalignments during registration can lead to inaccuracies and diminished segmentation quality. To address this, we developed EFS-MedSeg, an end-to-end model using two labeled atlases and few unlabeled images, enhanced by data augmentation and self-supervised learning. Initially, EFS-MedSeg applies a 3D random regional switch strategy to augment atlases, thereby enhancing supervision in segmentation tasks. This not only introduces variability to the training data but also enhances the model's ability to generalize and prevents overfitting, resulting in natural and smooth label boundaries. Following this, we use a variational autoencoder for a weighted reconstruction task, focusing the model's attention on areas with lower Dice scores to ensure accurate segmentation that conforms to the atlas image's shape and structural appearance. Moreover, we introduce a self-contrastive module aimed at improving feature extraction, guided by anatomical structure priors, thus enhancing the model's convergence and segmentation accuracy. Results on multi-modal medical image datasets show that EFS-MedSeg achieves performance comparable to fully-supervised methods. Moreover, it consistently surpasses the second-best method in Dice score by 1.4%, 9.1%, and 1.1% on the OASIS, BCV, and BCH datasets, respectively, highlighting its robustness and adaptability across diverse datasets. The source code will be made publicly available at: https://github.com/NoviceFodder/EFS-MedSeg.

UI MeSH Term Description Entries

Related Publications

Yanjie Zhou, and Feng Zhou, and Fengjun Xi, and Yong Liu, and Yun Peng, and David E Carlson, and Liyun Tu
July 2022, IEEE transactions on medical imaging,
Yanjie Zhou, and Feng Zhou, and Fengjun Xi, and Yong Liu, and Yun Peng, and David E Carlson, and Liyun Tu
January 2023, IEEE transactions on neural networks and learning systems,
Yanjie Zhou, and Feng Zhou, and Fengjun Xi, and Yong Liu, and Yun Peng, and David E Carlson, and Liyun Tu
June 2025, Artificial intelligence in medicine,
Yanjie Zhou, and Feng Zhou, and Fengjun Xi, and Yong Liu, and Yun Peng, and David E Carlson, and Liyun Tu
August 2023, Neural networks : the official journal of the International Neural Network Society,
Yanjie Zhou, and Feng Zhou, and Fengjun Xi, and Yong Liu, and Yun Peng, and David E Carlson, and Liyun Tu
January 2020, Frontiers in computational neuroscience,
Yanjie Zhou, and Feng Zhou, and Fengjun Xi, and Yong Liu, and Yun Peng, and David E Carlson, and Liyun Tu
May 2022, Medical image analysis,
Yanjie Zhou, and Feng Zhou, and Fengjun Xi, and Yong Liu, and Yun Peng, and David E Carlson, and Liyun Tu
October 2025, IEEE transactions on neural networks and learning systems,
Yanjie Zhou, and Feng Zhou, and Fengjun Xi, and Yong Liu, and Yun Peng, and David E Carlson, and Liyun Tu
July 2025, Medical physics,
Yanjie Zhou, and Feng Zhou, and Fengjun Xi, and Yong Liu, and Yun Peng, and David E Carlson, and Liyun Tu
October 2021, IEEE transactions on medical imaging,
Yanjie Zhou, and Feng Zhou, and Fengjun Xi, and Yong Liu, and Yun Peng, and David E Carlson, and Liyun Tu
December 2025, IEEE journal of biomedical and health informatics,
Copied contents to your clipboard!