arXiv:2506.13097cs.CV2025-06

通过增强原型与约束机制,提升多类无监督异常检测的准确率。

Pro-AD: Learning Comprehensive Prototypes with Prototype-based Constraint for Multi-class Unsupervised Anomaly Detection

  • 用更多可学习原型捕获更全面的正常特征信息。
  • 动态双向解码器融合特征聚合与重建,提升重建质量。
  • 引入原型约束防止异常被误重建,适合工业缺陷检测场景。

基于原型的无监督异常检测方法通常使用有限数量的可学习原型,难以充分捕捉正常样本的语义信息,导致重建效果不佳。而增加原型数量可能引发异常通过注意力机制被良好重建,即“软身份映射”问题。本文提出Pro-AD,首先引入扩展的可学习原型集以增强语义容量;其次设计动态双向解码器,将正常信息聚合与目标特征重建过程融合,使原型能从图像特征多层级中聚合更完整的正常信息,同时重建过程可动态利用这些原型并保留上下文信息。此外,为防止异常被过度重建,提出原型约束机制,施加于解码器的目标特征重建阶段,进一步提升性能。在多个挑战性基准数据集上的实验表明,Pro-AD达到当前最优水平,展现出卓越的鲁棒性与实用性。

原文摘要 · Abstract (English)

Prototype-based reconstruction methods for unsupervised anomaly detection utilize a limited set of learnable prototypes which only aggregates insufficient normal information, resulting in undesirable reconstruction. However, increasing the number of prototypes may lead to anomalies being well reconstructed through the attention mechanism, which we refer to as the "Soft Identity Mapping" problem. In this paper, we propose Pro-AD to address these issues and fully utilize the prototypes to boost the performance of anomaly detection. Specifically, we first introduce an expanded set of learnable prototypes to provide sufficient capacity for semantic information. Then we employ a Dynamic Bidirectional Decoder which integrates the process of the normal information aggregation and the target feature reconstruction via prototypes, with the aim of allowing the prototypes to aggregate more comprehensive normal semantic information from different levels of the image features and the target feature reconstruction to not only utilize its contextual information but also dynamically leverage the learned comprehensive prototypes. Additionally, to prevent the anomalies from being well reconstructed using sufficient semantic information through the attention mechanism, Pro-AD introduces a Prototype-based Constraint that applied within the target feature reconstruction process of the decoder, which further improves the performance of our approach. Extensive experiments on multiple challenging benchmarks demonstrate that our Pro-AD achieve state-of-the-art performance, highlighting its superior robustness and practical effectiveness for Multi-class Unsupervised Anomaly Detection task.

异常检测原型学习无监督图像重建

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。