arXiv:2602.00126cs.CVeess.IV2026-02

通过频域正则化提升工业缺陷检测的细节重建精度。

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

  • 用自监督修复任务+频域一致性损失,防止过度平滑。
  • 在MVTec Hazelnut上定位准确率(PRO AUC)提升至0.687。
  • 轻量级结构,单卡20 FPS,适合实际工业部署。

无监督异常检测是现代制造中自动化视觉检测的关键。基于重建的方法因其架构简单、处理速度快而受到青睐,但常产生过度平滑的结果,导致高频细节(如细微缺陷)被部分重建而非凸显,限制了分割精度。本文提出D3R-Net,一种双域去噪重建框架,将自监督‘修复’任务与频域感知正则化结合。训练时,网络接收合成损坏的正常图像,需重建出干净目标,从而避免恒等映射,学习无缺陷纹理流形。除空间均方误差外,引入快速傅里叶变换(FFT)幅值损失,增强频域一致性。可选结构相似性(SSIM)项用于消融研究。在MVTec AD Hazelnut基准上,使用FFT损失的D3R-Net使定位一致性提升:PRO AUC从0.603增至0.687,图像级ROC AUC保持稳定。在15个MVTec类别上,像素级ROC AUC平均从0.733升至0.751,PRO AUC从0.417升至0.468,单卡约20 FPS。模型从头训练,采用轻量卷积自动编码器主干,为重型预训练特征嵌入方法提供了实用替代方案。

原文摘要 · Abstract (English)

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.

缺陷检测频域正则自监督轻量模型

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