arXiv:2605.03221cs.CV2026-05

用扩散模型生成皮肤病图像,提升罕见病分类准确率

Synthetic Data Generation for Long-Tail Medical Image Classification: A Case Study in Skin Lesions

论文配图:Synthetic Data Generation for Long-Tail Medical Image Classification: A Case Study in Skin Lesions
图 1 · 摘自论文原文
  • 基于新型修复式扩散模型生成合成医学图像
  • 在样本最少类别上提升超28%分类性能
  • 适合医疗影像少样本场景的模型增强

长尾类分布广泛存在于多类医学数据集中,对深度学习模型构成挑战,尤其在罕见病类别上表现较差。现有方法如特殊架构、重加权损失函数和手工数据增强仅带来有限改进且难以扩展。本文提出一种面向医学长尾分类的扩散模型驱动合成数据增强流程,包含新颖的修复式扩散模型与异常检测后筛选机制,确保生成样本多样性、真实性和临床意义。在最大的不平衡医学影像基准ISIC2019皮肤病变分类数据集上评估,整体性能显著提升,样本最少类别上性能提升超过28%,验证了扩散生成在缓解长尾偏差与增强分类鲁棒性方面的有效性。

原文摘要 · Abstract (English)

Long-tailed class distributions are pervasive in multi-class medical datasets and pose significant challenges for deep learning models which typically underperform on tail classes with limited samples. This limitation is particularly problematic in medical applications, where rare classes often correspond to severe or high-risk diseases and therefore require high diagnostic accuracy. Existing solutions-including specialized architectures, rebalanced loss functions, and handcrafted data augmentation-offer only marginal improvements and struggle to scale due to their limited and largely deterministic variability. To address these challenges, we introduce a diffusion-model-driven synthetic data augmentation pipeline tailored for medical long-tailed classification. Our approach features a novel inpainting diffusion model combined with an Out-of-Distribution (OOD) post-selection mechanism to ensure diverse, realistic, and clinically meaningful synthetic samples. Evaluated on the ISIC2019 skin lesion classification dataset, one of the largest and most imbalanced medical imaging benchmarks, our method yields substantial improvements in overall performance, with particularly pronounced gains on tail classes with more than $28\%$ improvement on the class with the fewest samples. These results demonstrate the effectiveness of diffusion-based augmentation in mitigating long-tail imbalance and enhancing medical classification robustness.

医学图像扩散模型长尾学习

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