TACE让医学影像AI解释更精准,只改肿瘤区域不扰动整体结构。
TACE: Tumor-Aware Counterfactual Explanations
- 仅修改病灶区域,保持器官整体结构不变以提升解释可靠性
- 乳腺癌分类准确率提升10.69%,脑瘤提升98.02%
- 生成速度快且解释更忠实,适合临床可信AI需求
深度学习在医学影像中的应用显著提升了诊断的准确性与效率。然而,这些AI模型常被视为“黑箱”,其缺乏透明性引发了临床应用中的信任问题。可解释人工智能(XAI)旨在通过开发可理解的AI决策方法来缓解这一担忧。本文提出肿瘤感知反事实解释(TACE)框架,用于生成医学图像中更可靠的反事实解释。与现有方法不同,TACE专注于修改肿瘤特异性特征,而不改变整体器官结构,从而确保反事实解释的忠实性。我们通过在生成过程中引入额外步骤,仅对感兴趣区域(ROI)进行修改,使其余器官保持不变,从而生成更可信的反事实样本。我们在乳腺钼靶和脑部MRI数据集上进行了评估,结果表明,TACE在反事实生成质量、忠实度和速度方面均显著优于现有最先进方法。更忠实的解释带来分类成功率显著提升:乳腺癌提高10.69%,脑肿瘤提高98.02%。代码已开源于https://github.com/ispamm/TACE。
原文摘要 · Abstract (English)
The application of deep learning in medical imaging has significantly advanced diagnostic capabilities, enhancing both accuracy and efficiency. Despite these benefits, the lack of transparency in these AI models, often termed "black boxes," raises concerns about their reliability in clinical settings. Explainable AI (XAI) aims to mitigate these concerns by developing methods that make AI decisions understandable and trustworthy. In this study, we propose Tumor Aware Counterfactual Explanations (TACE), a framework designed to generate reliable counterfactual explanations for medical images. Unlike existing methods, TACE focuses on modifying tumor-specific features without altering the overall organ structure, ensuring the faithfulness of the counterfactuals. We achieve this by including an additional step in the generation process which allows to modify only the region of interest (ROI), thus yielding more reliable counterfactuals as the rest of the organ remains unchanged. We evaluate our method on mammography images and brain MRI. We find that our method far exceeds existing state-of-the-art techniques in quality, faithfulness, and generation speed of counterfactuals. Indeed, more faithful explanations lead to a significant improvement in classification success rates, with a 10.69% increase for breast cancer and a 98.02% increase for brain tumors. The code of our work is available at https://github.com/ispamm/TACE.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。