arXiv:2603.22874cs.CV2026-03中稿 · Engineering Applic…被引 21

用模板聚合特征,精准区分正常与异常,提升工业质检效率。

Template-Based Feature Aggregation Network for Industrial Anomaly Detection

  • 以固定模板图像聚合多层特征,过滤异常信息
  • 在MVTecAD等数据集上达到顶尖检测精度
  • 结构简单高效,适合工业实时质检场景

工业异常检测对保障产品质量至关重要。现有基于特征重构的方法虽表现优异,但存在捷径学习问题,可能导致异常特征被错误重建。为此,本文提出一种基于模板的特征聚合网络(TFA-Net),通过预训练卷积神经网络提取模板图像与输入图像的多层级特征。不同于直接重构输入特征,TFA-Net将输入特征聚合至模板特征,有效滤除与正常模板特征相似度低的异常部分。随后利用已融合正常特征的模板特征,细化并重构出完整特征图。最终通过对比输入与重构特征差异定位缺陷区域。此外,引入随机掩码策略进一步提升模型整体检测性能。所提模板特征聚合机制带来有意义的重构任务,TFA-Net结构简洁高效,在多个真实工业数据集上实现领先检测效果,并满足工业场景实时性要求。代码已开源。

原文摘要 · Abstract (English)

Industrial anomaly detection plays a crucial role in ensuring product quality control. Therefore, proposing an effective anomaly detection model is of great significance. While existing feature-reconstruction methods have demonstrated excellent performance, they face challenges with shortcut learning, which can lead to undesirable reconstruction of anomalous features. To address this concern, we present a novel feature-reconstruction model called the \textbf{T}emplate-based \textbf{F}eature \textbf{A}ggregation \textbf{Net}work (TFA-Net) for anomaly detection via template-based feature aggregation. Specifically, TFA-Net first extracts multiple hierarchical features from a pre-trained convolutional neural network for a fixed template image and an input image. Instead of directly reconstructing input features, TFA-Net aggregates them onto the template features, effectively filtering out anomalous features that exhibit low similarity to normal template features. Next, TFA-Net utilizes the template features that have already fused normal features in the input features to refine feature details and obtain the reconstructed feature map. Finally, the defective regions can be located by comparing the differences between the input and reconstructed features. Additionally, a random masking strategy for input features is employed to enhance the overall inspection performance of the model. Our template-based feature aggregation schema yields a nontrivial and meaningful feature reconstruction task. The simple, yet efficient, TFA-Net exhibits state-of-the-art detection performance on various real-world industrial datasets. Additionally, it fulfills the real-time demands of industrial scenarios, rendering it highly suitable for practical applications in the industry. Code is available at https://github.com/luow23/TFA-Net.

工业质检特征聚合异常检测

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