arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.12570cs.CV

适应视觉语言模型用于医学图像中的可泛化异常检测

Adapting Visual-Language Models for Generalizable Anomaly Detection in Medical Images

Shanghai Jiao Tong University(上海交通大学);National University of Singapore(新加坡国立大学);Shanghai Artificial Intelligence Laboratory(上海人工智能实验室)

Chaoqin Huang, Aofan Jiang, Jinghao Feng, Ya Zhang, Xinchao Wang, Yanfeng Wang

AI总结:

提出一种轻量级多层次适应框架,通过残差适配器和像素级视觉-语言对齐损失,将CLIP模型适配到医学图像异常检测,在零样本和少样本设置下显著提升分类与分割性能。

摘要:

近年来,大规模视觉语言预训练模型的进展在自然图像领域的零样本/少样本异常检测中取得了显著成果。然而,自然图像与医学图像之间存在巨大的领域差异,限制了这些方法在医学异常检测中的有效性。本文提出了一种新颖的轻量级多层次适应与比较框架,以重新利用CLIP模型进行医学异常检测。我们的方法将多个残差适配器集成到预训练的视觉编码器中,实现了跨不同层级的视觉特征逐步增强。这种多层次适应由多层级、像素级的视觉-语言特征对齐损失函数引导,将模型的关注点从自然图像中的物体语义重新校准到医学图像中的异常识别。适应后的特征在多种医学数据类型上表现出更好的泛化能力,即使在零样本场景下,模型在训练时未见过某些医学模态和解剖区域,也能实现良好性能。我们在医学异常检测基准上的实验表明,我们的方法显著超越了当前最先进的模型,在零样本和少样本设置下,异常分类的平均AUC分别提升了6.24%和7.33%,异常分割的平均AUC分别提升了2.03%和2.37%。源代码可在以下网址获取:https://github.com/MediaBrain-SJTU/MVFA-AD

英文摘要:

Recent advancements in large-scale visual-language pre-trained models have led to significant progress in zero-/few-shot anomaly detection within natural image domains. However, the substantial domain divergence between natural and medical images limits the effectiveness of these methodologies in medical anomaly detection. This paper introduces a novel lightweight multi-level adaptation and comparison framework to repurpose the CLIP model for medical anomaly detection. Our approach integrates multiple residual adapters into the pre-trained visual encoder, enabling a stepwise enhancement of visual features across different levels. This multi-level adaptation is guided by multi-level, pixel-wise visual-language feature alignment loss functions, which recalibrate the model's focus from object semantics in natural imagery to anomaly identification in medical images. The adapted features exhibit improved generalization across various medical data types, even in zero-shot scenarios where the model encounters unseen medical modalities and anatomical regions during training. Our experiments on medical anomaly detection benchmarks demonstrate that our method significantly surpasses current state-of-the-art models, with an average AUC improvement of 6.24% and 7.33% for anomaly classification, 2.03% and 2.37% for anomaly segmentation, under the zero-shot and few-shot settings, respectively. Source code is available at: https://github.com/MediaBrain-SJTU/MVFA-AD

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

↑