Enhancing Skin Disease Diagnosis: Interpretable Visual Concept Discovery with SAM

TL;DR

This paper introduces a novel framework that leverages the Segment Anything Model (SAM) to generate interpretable visual concepts for skin disease diagnosis, addressing challenges of noisy clinical images and limited annotations.

Contribution

It proposes a cross-attentive fusion approach that combines SAM-generated visual concepts with global features for improved interpretability and diagnosis accuracy in skin lesion classification.

Findings

Effective lesion diagnosis demonstrated on two datasets.

Enhanced interpretability of skin disease predictions.

Improved performance over baseline models.

Abstract

Current AI-assisted skin image diagnosis has achieved dermatologist-level performance in classifying skin cancer, driven by rapid advancements in deep learning architectures. However, unlike traditional vision tasks, skin images in general present unique challenges due to the limited availability of well-annotated datasets, complex variations in conditions, and the necessity for detailed interpretations to ensure patient safety. Previous segmentation methods have sought to reduce image noise and enhance diagnostic performance, but these techniques require fine-grained, pixel-level ground truth masks for training. In contrast, with the rise of foundation models, the Segment Anything Model (SAM) has been introduced to facilitate promptable segmentation, enabling the automation of the segmentation process with simple yet effective prompts. Efforts applying SAM predominantly focus on dermatoscopy images, which present more easily identifiable lesion boundaries than clinical photos taken with smartphones. This limitation constrains the practicality of these approaches to real-world applications. To overcome the challenges posed by noisy clinical photos acquired via non-standardized protocols and to improve diagnostic accessibility, we propose a novel Cross-Attentive Fusion framework for interpretable skin lesion diagnosis. Our method leverages SAM to generate visual concepts for skin diseases using prompts, integrating local visual concepts with global image features to enhance model performance. Extensive evaluation on two skin disease datasets demonstrates our proposed method's effectiveness on lesion diagnosis and interpretability.