arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.09520cs.CVcs.AI
收藏

增强皮肤病诊断:基于SAM的可解释视觉概念发现

Enhancing Skin Disease Diagnosis: Interpretable Visual Concept Discovery with SAM

Tulane University(杜兰大学)

Xin Hu, Janet Wang, Jihun Hamm, Rie R Yotsu, Zhengming Ding

AI总结:

针对临床皮肤照片噪声大、标注少、需可解释诊断的问题,提出基于SAM的交叉注意力融合框架,融合局部视觉概念与全局特征,在两个皮肤病数据集上实现了更好的诊断性能与可解释性。

摘要:

在深度学习架构快速发展的推动下,当前的AI辅助皮肤图像诊断在皮肤癌分类方面已达到皮肤科医生级别的性能。然而,与传统视觉任务不同,由于标注良好的数据集数量有限、病情复杂多变,且为保障患者安全需要提供详细的解释,皮肤图像通常面临独特的挑战。以往的分割方法试图减少图像噪声并提升诊断性能,但这类技术需要细粒度的像素级真值掩码用于训练。相比之下,随着基础模型的兴起,Segment Anything Model(SAM,分割一切模型)被提出以支持可提示分割,能够通过简单且有效的提示实现分割流程的自动化。目前应用SAM的研究主要聚焦于皮肤镜图像,这类图像的病变边界比智能手机拍摄的临床照片更易识别,这一限制制约了此类方法在实际场景中的实用性。为解决非标准化流程采集的含噪临床照片带来的挑战、提升诊断可及性,我们提出了一种用于可解释皮肤病变诊断的新型交叉注意力融合框架。我们的方法利用SAM通过提示生成皮肤病的视觉概念,将局部视觉概念与全局图像特征相融合以提升模型性能。在两个皮肤病数据集上的广泛评估证明了我们提出的方法在病变诊断和可解释性方面的有效性。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑