Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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

About

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

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

Related benchmarks

TaskDatasetResultRank
Anomaly DetectionVisA--
199
Anomaly SegmentationMVTec-AD (test)--
85
Anomaly SegmentationRESC
AUC99.07
74
Anomaly ClassificationLiverCT
AUC85.9
72
Anomaly ClassificationRESC
AUC (%)97.25
68
Anomaly DetectionMPDD
Clean AUROC0.709
62
Anomaly DetectionSDD
AUC0.825
57
Anomaly ClassificationOCT 17
AUC99.66
54
WSI ClassificationBRACS (test)--
54
Anomaly DetectionBraTS 2021
Clean AUROC97.7
50
Showing 10 of 89 rows
...

Other info

Code

Follow for update