arXiv:2605.16469eess.IVcs.CV2026-05

针对罕见病影像数据少的问题,用子类先验优化生成路径,提升稀有病种的生成质量。

Flow Matching with Optimized Subclass Priors for Medical Image Augmentation

  • 按子类型划分疾病,用混合高斯模型在隐空间分组,构建更精细的生成条件。
  • 为每个子类学习定制起点分布,缩短生成路径,减少类内差异,提升多样性。
  • 适合医学影像增强,尤其对罕见病检测任务有显著帮助。

罕见病在医学影像诊断中构成主要挑战,但临床数据集严重缺乏此类样本,导致分类器在最需可靠检测的场景下失效。生成式数据增强可弥补尾部类别覆盖不足,但粗粒度疾病标签将多种亚型和采集条件混为一谈,使生成器偏向主导亚模态,且共享高斯源迫使稀有子群体经历过长的传输路径。本文提出一种离线策略,在两个层面引入信息性先验:首先,通过生成模型隐空间中的高斯混合建模,将每个粗粒度标签划分为连贯的子模态;其次,学习子类条件下的源分布,按子模态重新中心化与缩放起始分布,缩短生成轨迹并降低类内分散度。为防止退化解,施加显式几何控制,适度集中归一化位移方向至可学习原型,同时截断路径长度异常值。在长尾胸部X光(MIMIC-LT、NIH-LT)和CT切片(CT-RATE)基准上,该方法持续提升尾部类别的生成保真度与多样性(FID、IRS),并显著提高下游任务的平衡准确率与宏平均F1值,优于非增强基线,跨模态表现稳定。

原文摘要 · Abstract (English)

Rare diseases dominate the diagnostic challenge in medical imaging yet are severely underrepresented in clinical datasets, causing classifiers to fail on exactly the conditions where reliable detection matters most. Generative augmentation can supply the missing tail-class coverage, but coarse disease labels aggregate diverse subtypes and acquisition settings into multi-modal conditionals that bias generators toward dominant submodes, while a shared Gaussian source forces rare subpopulations through disproportionately long transport paths. We propose an offline strategy that introduces informative priors at two levels: first, we partition each coarse label into coherent submodes via Gaussian mixture modeling in the generative model's latent space; second, we learn subclass-conditioned source distributions that re-center and re-scale the starting distribution per submode, shortening trajectories and reducing within-subclass dispersion. To prevent degenerate solutions we impose explicit geometric control, moderately concentrating normalized displacement directions around learnable prototypes while capping path-length outliers. On long-tailed chest X-ray (MIMIC-LT, NIH-LT) and CT slice (CT-RATE) benchmarks the proposed method consistently improves tail-class generation fidelity and diversity (FID, IRS) and is a promising augmentation strategy that reliably improves downstream balanced accuracy and macro-F1 over a non-augmented baseline across modalities.

医学图像数据增强生成模型长尾分布

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