Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Diverse Normal Prototypes-Guided Contrastive Reconstruction for Medical Anomaly Detection

About

Anomaly detection in medical images is challenging due to limited annotations and the domain gap. Existing reconstruction-based methods often rely on frozen pre-trained encoders, restricting adaptation to domain-specific patterns and degrading localization accuracy. Meanwhile, prototype-based learning offers interpretable representations but commonly suffers from prototype collapse, where a few prototypes dominate training and reduce diversity. To address these issues, we propose DNP-ConFormer, a unified framework that integrates a trainable encoder with prototype-guided reconstruction and a Diversity-Aware Alignment Loss. A momentum encoder enables stable domain-adaptive representation learning, while a lightweight Prototype Extractor discovers informative normal prototypes and injects them into the decoder via attention to guide reconstruction. The proposed alignment objective further encourages balanced feature-to-prototype assignments, effectively mitigating prototype collapse. Extensive experiments on multiple medical imaging benchmarks demonstrate improved representation quality and anomaly localization compared with prior methods. Visualization and prototype assignment analyses further validate the effectiveness and interpretability of our approach. The code is available at https://github.com/liluhu0/DNP-ConFormer.

Luhu Li, Bin Liu, Bowen Lin, Zihan Shen, Chengwei Wang, Shujun Fu• 2025

Related benchmarks

TaskDatasetResultRank
Anomaly DetectionBrainMIR (in-house)
AUC92.52
7
Anomaly LocalizationBrainMIR (in-house)
AUC98.53
7
Medical Anomaly DetectionAPTOS 2019
AUC97.92
7
Medical Anomaly DetectionOCT 2017
AUC99.83
7
Medical Anomaly DetectionISIC 2018
AUC0.9173
7
Showing 5 of 5 rows

Other info

Follow for update