论文标题
用于异常检测和定位的原型残留网络
Prototypical Residual Networks for Anomaly Detection and Localization
论文作者
论文摘要
异常检测和本地化被广泛用于工业制造业的效率和有效性。异常很少见,很难收集和监督模型,可以轻松地与这些异常样本异常相处,从而产生不令人满意的性能。另一方面,异常通常是微妙的,难以辨别的,并且具有各种外观,因此很难检测到异常,更不用说定位异常区域了。为了解决这些问题,我们提出了一个称为典型残差网络(PRN)的框架,该框架在异常和正常模式之间学习具有不同尺度和大小的残留物,以准确地重建异常区域的分割图。 PRN主要由两个部分组成:多尺度原型,这些原型明确表示异常的残留特征与正常模式;一种多尺寸的自我发挥机制,可以实现可变大小的异常学习。此外,我们提出了各种异常产生策略,这些策略既认为看到和看不见的外观差异都会扩大和多样化异常。关于具有挑战性且广泛使用的MVTEC AD基准的广泛实验表明,PRN优于当前最新的无监督和监督方法。我们进一步在另外三个数据集上报告了SOTA结果,以证明PRN的有效性和概括性。
Anomaly detection and localization are widely used in industrial manufacturing for its efficiency and effectiveness. Anomalies are rare and hard to collect and supervised models easily over-fit to these seen anomalies with a handful of abnormal samples, producing unsatisfactory performance. On the other hand, anomalies are typically subtle, hard to discern, and of various appearance, making it difficult to detect anomalies and let alone locate anomalous regions. To address these issues, we propose a framework called Prototypical Residual Network (PRN), which learns feature residuals of varying scales and sizes between anomalous and normal patterns to accurately reconstruct the segmentation maps of anomalous regions. PRN mainly consists of two parts: multi-scale prototypes that explicitly represent the residual features of anomalies to normal patterns; a multisize self-attention mechanism that enables variable-sized anomalous feature learning. Besides, we present a variety of anomaly generation strategies that consider both seen and unseen appearance variance to enlarge and diversify anomalies. Extensive experiments on the challenging and widely used MVTec AD benchmark show that PRN outperforms current state-of-the-art unsupervised and supervised methods. We further report SOTA results on three additional datasets to demonstrate the effectiveness and generalizability of PRN.