Scale-Aware Contrastive Reverse Distillation for Unsupervised Medical Anomaly Detection
By: Chunlei Li , Yilei Shi , Jingliang Hu and more
Potential Business Impact:
Finds hidden problems in medical pictures.
Unsupervised anomaly detection using deep learning has garnered significant research attention due to its broad applicability, particularly in medical imaging where labeled anomalous data are scarce. While earlier approaches leverage generative models like autoencoders and generative adversarial networks (GANs), they often fall short due to overgeneralization. Recent methods explore various strategies, including memory banks, normalizing flows, self-supervised learning, and knowledge distillation, to enhance discrimination. Among these, knowledge distillation, particularly reverse distillation, has shown promise. Following this paradigm, we propose a novel scale-aware contrastive reverse distillation model that addresses two key limitations of existing reverse distillation methods: insufficient feature discriminability and inability to handle anomaly scale variations. Specifically, we introduce a contrastive student-teacher learning approach to derive more discriminative representations by generating and exploring out-of-normal distributions. Further, we design a scale adaptation mechanism to softly weight contrastive distillation losses at different scales to account for the scale variation issue. Extensive experiments on benchmark datasets demonstrate state-of-the-art performance, validating the efficacy of the proposed method. Code is available at https://github.com/MedAITech/SCRD4AD.
Similar Papers
DNP-Guided Contrastive Reconstruction with a Reverse Distillation Transformer for Medical Anomaly Detection
CV and Pattern Recognition
Finds hidden problems in medical pictures.
Scale-wise Distillation of Diffusion Models
CV and Pattern Recognition
Makes AI art creation much faster.
Uncertainty-Aware Multi-Expert Knowledge Distillation for Imbalanced Disease Grading
CV and Pattern Recognition
Helps doctors grade disease pictures better.