SubspaceAD: Training-Free Few-Shot Anomaly Detection via Subspace Modeling
AI 摘要
SubspaceAD是一种无需训练的少样本异常检测方法,基于预训练模型特征和子空间建模。
主要贡献
- 提出无需训练的少样本异常检测方法SubspaceAD
- 利用预训练DINOv2提取特征并进行PCA降维
- 在MVTec-AD和VisA数据集上达到SOTA性能
方法论
利用DINOv2提取图像块特征,通过PCA估计正常变化的低维子空间,并基于重构残差进行异常检测。
原文摘要
Detecting visual anomalies in industrial inspection often requires training with only a few normal images per category. Recent few-shot methods achieve strong results employing foundation-model features, but typically rely on memory banks, auxiliary datasets, or multi-modal tuning of vision-language models. We therefore question whether such complexity is necessary given the feature representations of vision foundation models. To answer this question, we introduce SubspaceAD, a training-free method, that operates in two simple stages. First, patch-level features are extracted from a small set of normal images by a frozen DINOv2 backbone. Second, a Principal Component Analysis (PCA) model is fit to these features to estimate the low-dimensional subspace of normal variations. At inference, anomalies are detected via the reconstruction residual with respect to this subspace, producing interpretable and statistically grounded anomaly scores. Despite its simplicity, SubspaceAD achieves state-of-the-art performance across one-shot and few-shot settings without training, prompt tuning, or memory banks. In the one-shot anomaly detection setting, SubspaceAD achieves image-level and pixel-level AUROC of 98.0% and 97.6% on the MVTec-AD dataset, and 93.3% and 98.3% on the VisA dataset, respectively, surpassing prior state-of-the-art results. Code and demo are available at https://github.com/CLendering/SubspaceAD.