用轻量级特征重建法同时检测工业图像的结构与逻辑异常
Revisiting Deep Feature Reconstruction for Logical and Structural Industrial Anomaly Detection
- 基于深度特征重建,降低内存与计算开销
- 在五个数据集上超越8种先进方法,实现两类异常精准定位
- 适合需要兼顾效率与多类型异常检测的工业场景
工业异常检测对质量控制和预测性维护至关重要,但受限于训练数据少、异常类型多样及外部因素导致的外观变化。现有方法通常通过深度预训练网络提取图像块的多尺度特征来检测结构异常(如凹痕、划痕),但面临显著的内存与计算负担。此外,传统基于块的方法难以捕捉空间关系,无法有效识别逻辑异常(如缺失或多余部件)。本文聚焦深度特征重建(DFR),提出一种高效检测结构异常的方法,并进一步构建统一框架ULSAD,可同时处理结构与逻辑异常。通过优化训练目标提升结构异常检测性能,引入基于注意力的全局自编码器损失机制以应对逻辑异常。在五个基准数据集上的实证评估显示,ULSAD在两类异常检测与定位上均优于八种先进方法。消融实验验证了各模块对性能提升的关键贡献。代码已开源。
原文摘要 · Abstract (English)
Industrial anomaly detection is crucial for quality control and predictive maintenance, but it presents challenges due to limited training data, diverse anomaly types, and external factors that alter object appearances. Existing methods commonly detect structural anomalies, such as dents and scratches, by leveraging multi-scale features from image patches extracted through deep pre-trained networks. However, significant memory and computational demands often limit their practical application. Additionally, detecting logical anomalies-such as images with missing or excess elements-requires an understanding of spatial relationships that traditional patch-based methods fail to capture. In this work, we address these limitations by focusing on Deep Feature Reconstruction (DFR), a memory- and compute-efficient approach for detecting structural anomalies. We further enhance DFR into a unified framework, called ULSAD, which is capable of detecting both structural and logical anomalies. Specifically, we refine the DFR training objective to improve performance in structural anomaly detection, while introducing an attention-based loss mechanism using a global autoencoder-like network to handle logical anomaly detection. Our empirical evaluation across five benchmark datasets demonstrates the performance of ULSAD in detecting and localizing both structural and logical anomalies, outperforming eight state-of-the-art methods. An extensive ablation study further highlights the contribution of each component to the overall performance improvement. Our code is available at https://github.com/sukanyapatra1997/ULSAD-2024.git
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。