让医学图像分类更可信,减少模型过度自信
Explainable Image Classification with Reduced Overconfidence for Tissue Characterisation
- 用多轮预测生成像素级解释分布,提升可解释性
- 首次引入风险估计,量化解释结果的可靠性
- 适合医疗影像分析、需要可信决策支持的场景
术中应用机器学习模型进行组织特征识别可辅助决策并引导安全肿瘤切除。现有图像分类模型常依赖像素归因方法实现可解释性,但深度学习模型的过度自信会传递至像素归因结果。本文提出首个将风险估计融入像素归因方法的新框架。该方法通过迭代使用分类模型与归因方法,生成归因图(PA maps)的体积,并首次构建像素级归因值分布。进一步通过期望值估计生成增强型归因图,并利用变异系数(CV)评估该归因图的像素级风险。在探针式共聚焦激光内窥镜(pCLE)数据和ImageNet上的实验表明,所提方法在可解释性上优于现有最先进水平。
原文摘要 · Abstract (English)
The deployment of Machine Learning models intraoperatively for tissue characterisation can assist decision making and guide safe tumour resections. For image classification models, pixel attribution methods are popular to infer explainability. However, overconfidence in deep learning model's predictions translates to overconfidence in pixel attribution. In this paper, we propose the first approach which incorporates risk estimation into a pixel attribution method for improved image classification explainability. The proposed method iteratively applies a classification model with a pixel attribution method to create a volume of PA maps. This volume is used for the first time, to generate a pixel-wise distribution of PA values. We introduce a method to generate an enhanced PA map by estimating the expectation values of the pixel-wise distributions. In addition, the coefficient of variation (CV) is used to estimate pixel-wise risk of this enhanced PA map. Hence, the proposed method not only provides an improved PA map but also produces an estimation of risk on the output PA values. Performance evaluation on probe-based Confocal Laser Endomicroscopy (pCLE) data and ImageNet verifies that our improved explainability method outperforms the state-of-the-art.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。