电气工程与系统科学 > 图像与视频处理
[提交于 2025年7月26日
]
标题: SkinDualGen:皮肤病变中同时图像-掩码生成的提示驱动扩散模型
标题: SkinDualGen: Prompt-Driven Diffusion for Simultaneous Image-Mask Generation in Skin Lesions
摘要: 医学图像分析在诸如皮肤病变等疾病的早期诊断中起着关键作用。 然而,数据的稀缺性和类别不平衡显著阻碍了深度学习模型的性能。 我们提出了一种新方法,利用预训练的Stable Diffusion-2.0模型生成高质量的合成皮肤病变图像和相应的分割掩码。 这种方法增强了分类和分割任务的训练数据集。 我们通过领域特定的低秩适应(LoRA)微调和多目标损失函数的联合优化来适应Stable Diffusion-2.0,使模型能够在单一步骤中同时生成基于文本描述的临床相关图像和分割掩码。 实验结果表明,生成的图像经过FID分数验证,其质量与真实图像非常接近。 结合真实和合成数据的混合数据集显著提高了分类和分割模型的性能,在准确率和F1分数上实现了8%到15%的显著提升,并在其他关键指标如Dice系数和IoU上也获得了额外的积极增长。 我们的方法提供了一个可扩展的解决方案,以应对医学成像数据的挑战,有助于提高罕见疾病诊断的准确性与可靠性。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.