无需标签的异常检测框架,提升多类别工业质检准确性
BoRAD: Bootstrap your Own Representations for Multi-class Anomaly Detection

- 用可学习原型库施加双重正则化,抑制异常信息复制
- 在MVTec AD上达86.2% mAD,显著降低异常漏检
- 训练后推理无需原型查询,适合工业部署
基于重构的异常检测在工业质检中具有吸引力,但将其从类别特定训练扩展到通用设置面临挑战。单一模型需重建多样正常外观而不复制异常细节,这导致两种耦合失效模式:相同捷径(异常通过重构路径)和误重构(正常类别混淆)。我们提出BoRAD,一种无标签训练框架,将问题视为表示容量分配问题。BoRAD使用共享可学习原型库施加两个互补正则化:空间原型对齐压缩局部原型内变异以抑制异常复制,原型相对全局对齐保留原型间结构并增强对异常角度偏移的敏感性。原型库和预测头仅用于训练;推理仍为标准师生特征差异流程,无需类别标签、负样本对、记忆检索或原型查找。BoRAD在多项基准上表现优异,包括在MVTec AD上达86.2% mAD,VisA上80.7% mAD,Real-IAD上73.1% mAD。诊断分析显示异常泄漏减少、正常类别可分性提升、异常与正常得分分离更强。
原文摘要 · Abstract (English)
Reconstruction-based anomaly detection is attractive for industrial inspection, but scaling it from category-specific training to a one-for-all setting is challenging. A single model must reconstruct diverse normal appearances without copying abnormal details, which exposes two coupled failure modes: identical shortcut, where anomalies pass through the reconstruction path, and mis-reconstruction, where normal categories are confused with one another. We propose \textbf{BoRAD}, a label-free training framework that treats this as a representation-capacity allocation problem. BoRAD uses a shared learnable prototype bank to impose two complementary regularizers: spatial prototype alignment contracts local within-prototype variation to suppress anomaly copying, while prototype-relative global alignment preserves between-prototype structure and improves sensitivity to abnormal angular deviations. The prototype bank and prediction heads are used only during training; inference remains a standard teacher-student feature discrepancy pass, with no class labels, negative pairs, memory retrieval, or prototype lookup. BoRAD achieves competitive one-for-all anomaly detection performance, including 86.2\% mAD on MVTec AD, 80.7\% mAD on VisA and 73.1\% mAD on Real-IAD. Diagnostic analyses further show reduced anomaly leakage, improved normal-category separability, and stronger anomaly-normal score separation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。