提出协作修复框架,解决工业多类异常检测中重建失效问题。
Collaborative Reconstruction and Repair for Multi-class Industrial Anomaly Detection
- 将重建改为修复,使模型区分正常与异常区域
- 通过随机掩码保留局部特征,提升定位精度
- 适合需要高精度异常定位的工业质检场景
工业异常检测是一项开放集任务,旨在识别偏离正常数据分布的未知异常模式。为避免为每类构建独立模型带来的内存开销与泛化能力不足问题,本文致力于开发统一的多类异常检测框架。然而在此设定下,传统基于重建的网络常出现身份映射问题,即无论输入是否异常,均直接复制输入特征,导致检测失败。为此,本文提出新型框架——协作重建与修复(CRR),将重建过程转化为修复过程:首先优化解码器,在重建正常样本的同时修复合成异常,使其对异常区域生成不同表征,而对正常区域生成与编码器输出相似的表征;其次引入特征级随机掩码,确保解码器输出包含充分局部信息;最后,通过合成异常掩码监督分割网络训练,降低编码器与解码器特征表示差异带来的检测误差,显著提升定位性能。在多个工业数据集上的大量实验表明,CRR有效缓解了身份映射问题,并在多类工业异常检测中达到当前最优性能。
原文摘要 · Abstract (English)
Industrial anomaly detection is a challenging open-set task that aims to identify unknown anomalous patterns deviating from normal data distribution. To avoid the significant memory consumption and limited generalizability brought by building separate models per class, we focus on developing a unified framework for multi-class anomaly detection. However, under this challenging setting, conventional reconstruction-based networks often suffer from an identity mapping problem, where they directly replicate input features regardless of whether they are normal or anomalous, resulting in detection failures. To address this issue, this study proposes a novel framework termed Collaborative Reconstruction and Repair (CRR), which transforms the reconstruction to repairation. First, we optimize the decoder to reconstruct normal samples while repairing synthesized anomalies. Consequently, it generates distinct representations for anomalous regions and similar representations for normal areas compared to the encoder's output. Second, we implement feature-level random masking to ensure that the representations from decoder contain sufficient local information. Finally, to minimize detection errors arising from the discrepancies between feature representations from the encoder and decoder, we train a segmentation network supervised by synthetic anomaly masks, thereby enhancing localization performance. Extensive experiments on industrial datasets that CRR effectively mitigates the identity mapping issue and achieves state-of-the-art performance in multi-class industrial anomaly detection.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。