arXiv:2412.12532cs.CVcs.AI2024-12被引 7

用DDPM生成医学图像,显著提升小样本和不平衡数据的分类准确率。

Addressing Small and Imbalanced Medical Image Datasets Using Generative Models: A Comparative Study of DDPM and PGGANs with Random and Greedy K Sampling

  • 采用DDPM和PGGAN生成合成图像,结合随机与贪心采样构建小规模数据集。
  • DDPM生成图像更真实(FID更低),使分类准确率最高提升6%。
  • 适合处理医疗数据稀缺、类别不均衡问题的研究者参考。

医疗图像分类模型的开发常受限于隐私顾虑和特定疾病的样本稀缺,导致数据集小且类别不平衡。为解决此问题,本研究探索使用去噪扩散概率模型(DDPM)和渐进生长生成对抗网络(PGGAN)进行数据增强。研究设计框架评估这两种生成模型在四种模型上的表现:自定义CNN、未训练VGG16、预训练VGG16和预训练ResNet50。通过随机采样和贪心K采样构建小而不平衡的数据集,利用弗雷切特初始距离(FID)评估合成图像质量,并通过分类指标对比原始数据集。结果表明,DDPM持续生成更真实的图像(FID更低),在所有模型和数据集上均显著优于PGGAN;将DDPM生成图像融入原数据集后,准确率最高提升6%,尤其在类别不平衡场景下增强了模型鲁棒性与稳定性。随机采样表现出更高稳定性,贪心K采样虽提供多样性但伴随更高的FID得分。研究证实,DDPM能有效增强小而不平衡的医学图像数据集,通过扩大规模和平衡分布提升模型性能。

原文摘要 · Abstract (English)

The development of accurate medical image classification models is often constrained by privacy concerns and data scarcity for certain conditions, leading to small and imbalanced datasets. To address these limitations, this study explores the use of generative models, such as Denoising Diffusion Probabilistic Models (DDPM) and Progressive Growing Generative Adversarial Networks (PGGANs), for dataset augmentation. The research introduces a framework to assess the impact of synthetic images generated by DDPM and PGGANs on the performance of four models: a custom CNN, Untrained VGG16, Pretrained VGG16, and Pretrained ResNet50. Experiments were conducted using Random Sampling and Greedy K Sampling to create small, imbalanced datasets. The synthetic images were evaluated using Frechet Inception Distance (FID) and compared to original datasets through classification metrics. The results show that DDPM consistently generated more realistic images with lower FID scores and significantly outperformed PGGANs in improving classification metrics across all models and datasets. Incorporating DDPM-generated images into the original datasets increased accuracy by up to 6%, enhancing model robustness and stability, particularly in imbalanced scenarios. Random Sampling demonstrated superior stability, while Greedy K Sampling offered diversity at the cost of higher FID scores. This study highlights the efficacy of DDPM in augmenting small, imbalanced medical image datasets, improving model performance by balancing the dataset and expanding its size.

医学图像数据增强生成模型小样本学习

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