无需人工反馈,通过原型引导扩散模型提升医学图像生成的生物合理性。
IMPROVE: Improving Medical Plausibility without Reliance on HumanValidation -- An Enhanced Prototype-Guided Diffusion Framework
- 基于原型引导的扩散框架,自动捕捉医学图像特征
- 在骨髓和HAM10000数据集上显著提升医学准确性
- 避免耗时的人工评估,适合医疗数据增强场景
生成模型在合成医学图像方面表现优异,广泛应用于罕见病数据集增强、长尾分布扩充及机器学习算法扩展。尽管传统指标(如FID、精确率、召回率)显示生成图像质量良好,但这些指标无法衡量图像的医学或生物学合理性。此前研究依赖人类专家反馈并通过强化学习从人类反馈(RLHF)提升合理性,但该过程成本高且耗时。本文提出IMPROVE:一种无需人类验证的原型引导扩散框架,通过引入原型机制增强生成图像的生物学一致性。在骨髓和HAM10000数据集上的实验表明,该方法显著提升医学准确性,且完全不依赖人工反馈。
原文摘要 · Abstract (English)
Generative models have proven to be very effective in generating synthetic medical images and find applications in downstream tasks such as enhancing rare disease datasets, long-tailed dataset augmentation, and scaling machine learning algorithms. For medical applications, the synthetically generated medical images by such models are still reasonable in quality when evaluated based on traditional metrics such as FID score, precision, and recall. However, these metrics fail to capture the medical/biological plausibility of the generated images. Human expert feedback has been used to get biological plausibility which demonstrates that these generated images have very low plausibility. Recently, the research community has further integrated this human feedback through Reinforcement Learning from Human Feedback(RLHF), which generates more medically plausible images. However, incorporating human feedback is a costly and slow process. In this work, we propose a novel approach to improve the medical plausibility of generated images without the need for human feedback. We introduce IMPROVE:Improving Medical Plausibility without Reliance on Human Validation - An Enhanced Prototype-Guided Diffusion Framework, a prototype-guided diffusion process for medical image generation and show that it substantially enhances the biological plausibility of the generated medical images without the need for any human feedback. We perform experiments on Bone Marrow and HAM10000 datasets and show that medical accuracy can be substantially increased without human feedback.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。