arXiv:2606.25770cs.LGcs.CV2026-06中稿 · publication at the…

通过混合嵌入生成真实患者,解决医疗小样本学习难题。

Re-mixing Embeddings for Patient Augmentation in Data Scarce Multiple Instance Learning

论文配图:Re-mixing Embeddings for Patient Augmentation in Data Scarce Multiple Instance Learning
图 1 · 摘自论文原文
  • 用高斯混合模型在嵌入空间聚类,学习疾病特异性组合分布。
  • 在无健康样本数据集上生成患者,性能接近全量数据训练。
  • 适用于罕见病诊断与隐私保护,支持单细胞、流式等小样本任务。

数据稀缺是医学多实例学习(MIL)的主要瓶颈,尤其在罕见病或昂贵检测模态中。本文提出一种基于统计学的患者增强方法,直接在嵌入空间生成真实患者。通过将所有患者的实例嵌入进行聚类,使用高斯混合模型学习疾病特异性的“配方”——即实例在无监督簇中的统计分布。新患者通过从这些簇中采样嵌入并按配方重组生成。该方法无需各类样本示例,可离线重构患者。生成后患者通过不确定性量化筛选以提升MIL性能。我们在三种临床相关场景验证:(i) 跨数据集迁移,用外部队列统计生成缺失的“健康”类别;(ii) 低数据量情形,类别样本极有限;(iii) 小队列非图像任务,包括单细胞RNA测序和流式细胞术。实验表明,本方法在各项任务中均优于基线,常胜于其他袋混合策略。尤其在缺失类别场景下,性能接近全数据训练,展现出在罕见病诊断与隐私保护患者增强中的潜力。代码已开源。

原文摘要 · Abstract (English)

Data scarcity is a major bottleneck in medical Multiple Instance Learning (MIL), especially for rare diseases or expensive modalities. We introduce a statistically grounded patient augmentation approach that generates realistic patients directly in embedding space. Using Gaussian Mixture Models as a probabilistic clustering approach on pooled instance embeddings from all patients, our method learns disease-specific "recipes"-statistical distributions of instances across unsupervised clusters. New patients are then generated by sampling embeddings from clusters based on learned recipes. Unlike existing methods that require examples from all categories, our method can generate patients offline by re-mixing pooled embeddings. Generated patients are further selected based on uncertainty quantification to improve MIL performance. We evaluate our method across three clinically relevant scarcity scenarios: (i) cross-dataset transfer, where an entirely missing "healthy" class is generated using statistics from an external cohort; (ii) low-data regimes, where class sizes are extremely limited; and (iii) small-cohort non-image tasks, including single-cell RNA-seq and flow cytometry. Across all experiments, our method improves performance over baseline, often outperforming other bag-mixing strategies. Notably, in the missing-class scenario, a performance comparable to full-dataset training is achieved, demonstrating its potential for rare disease diagnostic and privacy-preserving patient augmentation. The code is available at https://github.com/marrlab/RECIPE

医疗AI小样本学习数据增强嵌入空间

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。