Towards Robust In-Context Learning for Medical Image Segmentation via Data Synthesis

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Hu, Jiesi, Yang, Yanwu, Ye, Zhiyu, Ye, Chenfei, Peng, Hanyang, Cao, Jianfeng, Ma, Ting
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916966967541760
author Hu, Jiesi
Yang, Yanwu
Ye, Zhiyu
Ye, Chenfei
Peng, Hanyang
Cao, Jianfeng
Ma, Ting
author_facet Hu, Jiesi
Yang, Yanwu
Ye, Zhiyu
Ye, Chenfei
Peng, Hanyang
Cao, Jianfeng
Ma, Ting
contents The rise of In-Context Learning (ICL) for universal medical image segmentation has introduced an unprecedented demand for large-scale, diverse datasets for training, exacerbating the long-standing problem of data scarcity. While data synthesis offers a promising solution, existing methods often fail to simultaneously achieve both high data diversity and a domain distribution suitable for medical data. To bridge this gap, we propose \textbf{SynthICL}, a novel data synthesis framework built upon domain randomization. SynthICL ensures realism by leveraging anatomical priors from real-world datasets, generates diverse anatomical structures to cover a broad data distribution, and explicitly models inter-subject variations to create data cohorts suitable for ICL. Extensive experiments on four held-out datasets validate our framework's effectiveness, showing that models trained with our data achieve performance gains of up to 63\% in average Dice and substantially enhanced generalization to unseen anatomical domains. Our work helps mitigate the data bottleneck for ICL-based segmentation, paving the way for robust models. Our code and the generated dataset are publicly available at https://github.com/jiesihu/Neuroverse3D.
format Preprint
id arxiv_https___arxiv_org_abs_2509_19711
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Towards Robust In-Context Learning for Medical Image Segmentation via Data Synthesis
Hu, Jiesi
Yang, Yanwu
Ye, Zhiyu
Ye, Chenfei
Peng, Hanyang
Cao, Jianfeng
Ma, Ting
Computer Vision and Pattern Recognition
The rise of In-Context Learning (ICL) for universal medical image segmentation has introduced an unprecedented demand for large-scale, diverse datasets for training, exacerbating the long-standing problem of data scarcity. While data synthesis offers a promising solution, existing methods often fail to simultaneously achieve both high data diversity and a domain distribution suitable for medical data. To bridge this gap, we propose \textbf{SynthICL}, a novel data synthesis framework built upon domain randomization. SynthICL ensures realism by leveraging anatomical priors from real-world datasets, generates diverse anatomical structures to cover a broad data distribution, and explicitly models inter-subject variations to create data cohorts suitable for ICL. Extensive experiments on four held-out datasets validate our framework's effectiveness, showing that models trained with our data achieve performance gains of up to 63\% in average Dice and substantially enhanced generalization to unseen anatomical domains. Our work helps mitigate the data bottleneck for ICL-based segmentation, paving the way for robust models. Our code and the generated dataset are publicly available at https://github.com/jiesihu/Neuroverse3D.
title Towards Robust In-Context Learning for Medical Image Segmentation via Data Synthesis
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2509.19711