用强化学习选关键样本,提升少样本医学图像分割效果
Active few-shot segmentation by reinforcing data selection

- 用强化学习联合优化支持集,考虑样本间互补性
- 在盆腔MRI数据上优于随机选择和现有最优方法
- 适合需要高效标注的医疗图像分割场景
少样本学习使医学图像分割模型仅用少量标注样本即可适应新任务。然而,适应性能高度依赖支持集的选择。有效的支持集应涵盖目标域内的相关变化,并提供互补信息。现有主动数据选择方法多独立评估样本,未显式建模样本间交互。本文提出一种强化学习框架,用于少样本医学图像分割中的支持集选择,使支持集能联合优化而非独立评分。给定未标注候选图像池,智能体直接预测最大化下游分割性能的支持集。在跨机构盆腔MRI数据集上的实验表明,该方法优于随机选择和当前最优方法。研究结果强调了支持集互补性对有效适应的重要性,并展示了强化学习在优化适应集方面的潜力。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。