Few-shot anomaly detection via adaptive graph feature registration
摘要
Most existing anomaly detection methods adopt a “one-category-one-model” paradigm, requiring hundreds or even thousands of images per category. This is costly and impractical in industrial settings with limited samples, motivating few-shot anomaly detection (FSAD) to detect anomalies using only a few normal images per category. However, current state-of-the-art FSAD methods fail because direct image comparison overlooks feature-level dependencies, reducing precision. To address these challenges, we propose adaptive graph feature registration (AGFReg). First, we introduce a multi-scale maximum mean discrepancy (MMMD) loss function to measure feature distribution differences across multiple scales, improving robustness while preserving accuracy. Then, we design a graph feature registration module (GFRM) that combines graph convolution and channel attention to effectively capture relationships between features and their neighbors. Experimental results on the MVTec AD and MPDD benchmarks under various few-shot settings demonstrate that AGFReg surpasses existing FSAD methods, yielding an improvement of