XCoOp: Explainable Prompt Learning for Computer-Aided Diagnosis via Concept-guided Context Optimization

Yequan Bie,Luyang Luo,Zhixuan Chen,Hao Chen
2024-03-14
Abstract:Utilizing potent representations of the large vision-language models (VLMs) to accomplish various downstream tasks has attracted increasing attention. Within this research field, soft prompt learning has become a representative approach for efficiently adapting VLMs such as CLIP, to tasks like image classification. However, most existing prompt learning methods learn text tokens that are unexplainable, which cannot satisfy the stringent interpretability requirements of Explainable Artificial Intelligence (XAI) in high-stakes scenarios like healthcare. To address this issue, we propose a novel explainable prompt learning framework that leverages medical knowledge by aligning the semantics of images, learnable prompts, and clinical concept-driven prompts at multiple granularities. Moreover, our framework addresses the lack of valuable concept annotations by eliciting knowledge from large language models and offers both visual and textual explanations for the prompts. Extensive experiments and explainability analyses conducted on various datasets, with and without concept labels, demonstrate that our method simultaneously achieves superior diagnostic performance, flexibility, and interpretability, shedding light on the effectiveness of foundation models in facilitating XAI. The code will be made publically available.
Computer Vision and Pattern Recognition,Artificial Intelligence
What problem does this paper attempt to address?
The paper aims to address the following issues: In computer-aided diagnosis (CAD), leveraging the powerful representation capabilities of large-scale vision-language models (VLMs) to accomplish various downstream tasks has garnered increasing attention. Soft prompt learning, as an effective method to adapt VLMs, such as CLIP, has achieved significant results in tasks like image classification. However, existing prompt learning methods typically learn uninterpretable text tokens, which fail to meet the stringent requirements for explainable artificial intelligence (XAI) in high-risk scenarios such as healthcare. To address this issue, the authors propose a novel explainable prompt learning framework, XCoOp, which enhances the interpretability of prompts by aligning the semantics of images, learnable prompts, and clinically driven prompts at multiple granularity levels, thereby leveraging medical knowledge. Additionally, this framework compensates for the lack of valuable clinical concept annotations by extracting knowledge from large language models and provides both visual and textual explanations for the learned prompts. Extensive experiments and interpretability analyses demonstrate that this method achieves superior diagnostic performance, flexibility, and interpretability across various datasets, regardless of the presence of concept labels, showcasing the effectiveness of foundational models in promoting XAI.