用条件扩散模型做医学影像分类,无需标注也能解释结果并量化不确定度。
Conditional Diffusion Models are Medical Image Classifiers that Provide Explainability and Uncertainty for Free
- 通过多数投票机制提升扩散模型分类性能
- 在CheXpert和ISIC数据集上达到主流判别模型水平
- 天然可解释且能输出预测置信度,适合临床应用
判别式分类器已成为深度学习在医学影像中的基础工具,擅长学习复杂数据分布的可分特征。然而,这些模型常需精心设计、增强与训练技巧以确保安全可靠部署。近期,扩散模型在二维图像生成中成为主流,其分类能力可通过比较不同条件输入生成图像的重建误差实现。本文首次探索了条件扩散模型在2D医学图像分类中的潜力。首先提出一种新的多数投票方案,显著提升医学扩散分类器性能;其次,在CheXpert与ISIC皮肤癌数据集上的大量实验表明,基础模型与从零训练的扩散模型在无需显式监督的情况下,性能可媲美当前最优判别模型。此外,我们证明扩散分类器具有内在可解释性,能有效量化预测不确定性,从而在安全敏感的临床场景中提升可信度与可靠性。
原文摘要 · Abstract (English)
Discriminative classifiers have become a foundational tool in deep learning for medical imaging, excelling at learning separable features of complex data distributions. However, these models often need careful design, augmentation, and training techniques to ensure safe and reliable deployment. Recently, diffusion models have become synonymous with generative modeling in 2D. These models showcase robustness across a range of tasks including natural image classification, where classification is performed by comparing reconstruction errors across images generated for each possible conditioning input. This work presents the first exploration of the potential of class conditional diffusion models for 2D medical image classification. First, we develop a novel majority voting scheme shown to improve the performance of medical diffusion classifiers. Next, extensive experiments on the CheXpert and ISIC Melanoma skin cancer datasets demonstrate that foundation and trained-from-scratch diffusion models achieve competitive performance against SOTA discriminative classifiers without the need for explicit supervision. In addition, we show that diffusion classifiers are intrinsically explainable, and can be used to quantify the uncertainty of their predictions, increasing their trustworthiness and reliability in safety-critical, clinical contexts. Further information is available on our project page: https://faverogian.github.io/med-diffusion-classifier.github.io/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。