跳到主要内容
AI News HubLIVE
站内改写1 分钟阅读

长尾医学图像分类的合成数据生成:皮肤病变案例研究

文章摘要

论文提出了一种基于扩散模型的数据增强管道,用于解决医学图像分类中的长尾分布问题。该方法结合修复扩散模型和分布外后选择机制,生成多样化、真实且临床有意义的合成样本。在ISIC2019皮肤病变数据集上,尾部类别性能提升超过28%。

来源arXiv Computer Vision作者: Jiaxiang Jiang, Mahesh Subedar, Omesh Tickoo
长尾医学图像分类的合成数据生成:皮肤病变案例研究
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

长尾分布是医学图像分类中一个长期存在的难题。在多类医学数据集中,某些疾病类别(尤其是罕见但严重的疾病)的样本数量往往远少于常见类别,导致深度学习模型在“尾部”类别上性能显著下降。这一问题在临床应用中尤为关键,因为这些罕见类别往往对应高风险疾病,如某些类型的皮肤癌,需要极高的诊断准确率。

现有的应对方法包括设计专用的网络架构、采用重平衡的损失函数(如焦点损失)以及手工设计的数据增强策略。然而,这些方法只能带来有限的性能提升,且由于它们生成的变异有限且大多具有确定性,难以扩展到更复杂的实际场景。

针对上述挑战,来自英特尔实验室的Jiaxiang Jiang、Mahesh Subedar和Omesh Tickoo在2026年5月提交的一篇论文中,提出了一种基于扩散模型的合成数据增强管道,专门用于医学长尾分类任务。该方法的核心是一个新颖的图像修复扩散模型(inpainting diffusion model),它能够根据给定的病变区域生成与真实图像高度相似的合成图像。此外,该管道还引入了一个分布外(Out-of-Distribution, OOD)后选择机制,用于筛选生成的样本,确保其多样性和临床相关性。

研究人员在ISIC2019皮肤病变分类数据集上进行了实验验证。ISIC2019是最大的、最不平衡的医学影像基准之一,包含大量的皮肤镜图像。实验结果表明,所提出的方法在整体分类性能上取得了显著提升,尤其是在样本最少的尾部类别上,准确率提高了超过28%。这一成果充分证明了基于扩散模型的合成数据增强在缓解长尾不平衡问题上的巨大潜力,为提高医学图像分类模型的鲁棒性提供了新的思路。

该方法的成功不仅适用于皮肤病变分类,还有望推广到其他医学影像领域,如视网膜病变、肺部疾病等,为应对医疗数据中普遍存在的长尾分布问题提供了有效的解决方案。论文作者来自英特尔实验室,展示了工业界在推动医疗AI发展方面的积极努力。

展开要点与分析

文章情报

工程师进阶

要点

  • 长尾分布是医学多类数据集中的普遍问题,导致深度学习模型对稀有类别性能不佳。
  • 现有解决方案(特殊架构、重平衡损失、手工数据增强)改进有限且难以扩展。
  • 新方法采用修复扩散模型与分布外后选择机制生成合成数据。
  • 在ISIC2019数据集上,样本最少的类别性能提升超过28%。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。