融合自注意力与缺陷凸显的缺陷检测方法

作     者：乐华钢 王文武 朱磊 朱思宇 Yue Huagang;Wang Wenwu;Zhu Lei;Zhu Siyu

作者机构：武汉科技大学信息科学与工程学院武汉430081 

基　　金：国家自然科学基金(62173262)项目资助 

出 版 物：《电子测量与仪器学报》 (Journal of Electronic Measurement and Instrumentation)

年 卷 期：2023年第37卷第9期

页      码：85-92页

摘      要：针对无监督缺陷检测中重建网络在抑制异常重建的同时无法保留正常区域细节信息的问题,提出了一种融合自注意力与缺陷凸显的缺陷检测方法。首先,在重建网络中引入离散小波变换(DWT)进行下采样,并使用离散小波逆变换(IDWT)进行上采样。相较于传统重建网络,这种方法能减少细节信息的丢失,并对特征进行频率分解。同时,在跳连接中加入自注意力模块对特征重新编码,使其重点关注正常区域的细节信息。此外,设计了一个缺陷区域凸显模块,利用正常样本特征构建特征库,将从测试图像提取的特征与特征库中特征对比得到异常图,将异常图与重建差值图相结合来改善缺陷定位结果。在工业缺陷检测数据集MVTec AD上进行测试,在图像级AUROC上取得了99.3%的结果,同时在像素级AUROC上取得了98.3%的结果,在无监督缺陷检测中具有较高的检测精度和鲁棒性。

主 题 词：缺陷检测 重建网络 自注意力 缺陷凸显 

学科分类：0711[理学-心理学类] 07[理学] 08[工学] 080203[080203] 080401[080401] 0804[工学-材料学] 080402[080402] 0802[工学-机械学] 

核心收录：

D　O　I：10.13382/j.jemi.B2306646

馆 藏 号：203124534...