Abstract

Few-shot learning enables medical image segmentation models to adapt to new tasks using only a small number of labelled examples. However, adaptation performance depends strongly on which examples are selected for the support set. Effective support sets should capture relevant variation within the target domain and be informative for adaptation, with constituent samples providing complementary information. Despite this, existing active data selection approaches largely prioritise samples individually and do not explicitly account for interactions between examples. In this work, we propose a reinforcement learning framework for support-set selection in few-shot medical image segmentation, enabling support sets to be optimised jointly rather than through independent sample scoring. Given a pool of unlabelled candidate images, an agent directly predicts a support set that maximises downstream segmentation performance. Experiments on a cross-institutional pelvic MRI dataset demonstrate improvements over random selection and current state-of-the-art methods. Our findings highlight the importance of support-set complementarity for effective adaptation and demonstrate the potential of reinforcement learning for optimising adaptation sets. Code: https://github.com/timfkl/active-few-shot-learning

Links to Paper and Supplementary Materials

Main Paper (Open Access Version): https://papers.miccai.org/miccai-2026-sat/paper/EMA_008.pdf

SharedIt Link: Not yet available

SpringerLink (DOI): Not yet available

Supplementary Material: Not Submitted

Link to Open Review

Open Review Page: Not Available

BibTex

@InProceedings{ZhaChe_Active_MICCAISAT2026,
        author = { Zhao, Chenlan AND Wong, Benny AND Lundberg, Timothy F. AND Elsayed, Ahmed M. AND Aljarkas, Abdallah AND Aljamaan, Hamad A. AND Karam, Lynn AND Yang, Qianye AND Hu, Yipeng AND Villette, Claire C. AND Saeed, Shaheer U.},
        title = { { Active few-shot segmentation by reinforcing data selection } },
        booktitle = {Medical Image Computing and Computer Assisted Intervention -- MICCAI 2026 Workshops and Challenges},
        year = {2026},
        publisher = {Springer Nature Switzerland},
        volume = {LNCS 17265},
        month = {pending},
        page = {pending}
}


back to top