arXiv:2411.05609cs.CVcs.LG2024-11

用两步法提升皮肤病变诊断的可解释性与可信度

A Two-Step Concept-Based Approach for Enhanced Interpretability and Trust in Skin Lesion Diagnosis

  • 先用视觉语言模型自动提取临床概念,再由大语言模型生成诊断
  • 仅需少量标注数据,且在三个数据集上表现优于传统方法
  • 支持人工干预修正概念,提升透明度,适合临床部署

深度学习系统在临床应用中的主要障碍是标注数据稀缺以及缺乏可解释性和信任感。概念瓶颈模型(CBMs)通过约束疾病预测基于人类可理解的概念,提供了内在可解释性,但需大量标注且新增概念需重训。本文提出一种新型两步法:利用预训练视觉语言模型(VLM)自动预测临床概念,再通过现成大语言模型(LLM)根据预测概念生成诊断。该方法支持测试时人工干预修正概念,提升诊断准确性和决策透明度。在三个皮肤病变数据集上验证,无需训练、仅用少量标注样本即超越传统CBMs和先进可解释方法。代码已开源。

原文摘要 · Abstract (English)

The main challenges hindering the adoption of deep learning-based systems in clinical settings are the scarcity of annotated data and the lack of interpretability and trust in these systems. Concept Bottleneck Models (CBMs) offer inherent interpretability by constraining the final disease prediction on a set of human-understandable concepts. However, this inherent interpretability comes at the cost of greater annotation burden. Additionally, adding new concepts requires retraining the entire system. In this work, we introduce a novel two-step methodology that addresses both of these challenges. By simulating the two stages of a CBM, we utilize a pretrained Vision Language Model (VLM) to automatically predict clinical concepts, and an off-the-shelf Large Language Model (LLM) to generate disease diagnoses based on the predicted concepts. Furthermore, our approach supports test-time human intervention, enabling corrections to predicted concepts, which improves final diagnoses and enhances transparency in decision-making. We validate our approach on three skin lesion datasets, demonstrating that it outperforms traditional CBMs and state-of-the-art explainable methods, all without requiring any training and utilizing only a few annotated examples. The code is available at https://github.com/CristianoPatricio/2-step-concept-based-skin-diagnosis.

可解释性皮肤诊断多模态大模型应用

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。