arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.00126cs.CVeess.IV

D3R-Net:双域去噪重建网络用于鲁棒工业异常检测

D3R-Net: Dual-Domain Denoising Reconstruction Network for Robust Industrial Anomaly Detection

Aimech Technologies Corp.(Aimech技术公司);Covijn Ltd.(Covijn有限公司)

Dmytro Filatov, Valentyn Fedorov, Vira Filatova, Andrii Zelenchuk

AI总结:

D3R-Net通过双域去噪重建框架结合自监督修复任务和频率感知正则化,提升工业异常检测的鲁棒性和精度。

摘要:

无监督异常检测(UAD)是现代制造自动化视觉检测的关键组成部分。基于重建的方法因其基本的架构设计和快速处理数据而受到青睐,但它们在高频细节上会产生过度平滑的结果。因此,细微的缺陷部分被重建而不是突出显示,这限制了分割的准确性。我们在此基础上提出D3R-Net,一种双域去噪重建框架,将自监督的'修复'任务与频率感知正则化相结合。在训练过程中,网络接收合成的受污染正常图像,并被要求重建干净的目标,这防止了简单的身份映射,并推动模型学习无缺陷纹理的流形。除了空间均方误差外,我们还采用快速傅里叶变换(FFT)幅度损失,以在频域内促进一致性。实现还允许可选的结构相似性(SSIM)项,我们在消融实验中进行了研究。在MVTec AD榛子基准测试中,D3R-Net结合FFT损失在空间-only基线之上提高了定位一致性:PRO AUC从0.603增加到0.687,而图像级ROC AUC保持稳健。在十五个MVTec类别上评估时,FFT变体将平均像素ROC AUC从0.733提高到0.751,PRO AUC从0.417提高到0.468,与MSE-only基线相比,大约在单个GPU上以20 FPS的速度运行。该网络从头开始训练,并使用轻量级卷积自编码器主干,提供了一种替代重预训练特征嵌入方法的实用方案。

英文摘要:

Unsupervised anomaly detection (UAD) is a key ingredient of automated visual inspection in modern manufacturing. The reconstruction-based methods appeal because they have basic architectural design and they process data quickly but they produce oversmoothed results for high-frequency details. As a result, subtle defects are partially reconstructed rather than highlighted, which limits segmentation accuracy. We build on this line of work and introduce D3R-Net, a Dual-Domain Denoising Reconstruction framework that couples a self-supervised 'healing' task with frequency-aware regularization. During training, the network receives synthetically corrupted normal images and is asked to reconstruct the clean targets, which prevents trivial identity mapping and pushes the model to learn the manifold of defect-free textures. In addition to the spatial mean squared error, we employ a Fast Fourier Transform (FFT) magnitude loss that encourages consistency in the frequency domain. The implementation also allows an optional structural similarity (SSIM) term, which we study in an ablation. On the MVTec AD Hazelnut benchmark, D3R-Net with the FFT loss improves localization consistency over a spatial-only baseline: PRO AUC increases from 0.603 to 0.687, while image-level ROC AUC remains robust. Evaluated across fifteen MVTec categories, the FFT variant raises the average pixel ROC AUC from 0.733 to 0.751 and PRO AUC from 0.417 to 0.468 compared to the MSE-only baseline, at roughly 20 FPS on a single GPU. The network is trained from scratch and uses a lightweight convolutional autoencoder backbone, providing a practical alternative to heavy pre-trained feature embedding methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑