arXiv:2604.19240cs.AI2026-04

用扩散模型生成缺陷数据,结合师生网络实现精准无监督检测

Industrial Surface Defect Detection via Diffusion Generation and Asymmetric Student-Teacher Network

论文配图:Industrial Surface Defect Detection via Diffusion Generation and Asymmetric Student-Teacher Network
图 1 · 摘自论文原文
  • 用扩散模型从正常样本生成逼真缺陷数据,解决样本稀缺问题
  • 师生网络架构使异常区域差异被放大,定位精度达98.3%像素级AUROC
  • 无需真实缺陷样本,适合工业场景中细微缺陷的高精度检测

工业表面缺陷检测常面临缺陷样本少、长尾分布严重及复杂背景下的微小缺陷定位困难等问题。本文提出一种无监督缺陷检测方法,结合去噪扩散概率模型(DDPM)与非对称师生网络架构。在数据层面,仅用正常样本训练DDPM,通过恒定方差高斯扰动和Perlin噪声掩码生成高保真且物理一致的缺陷样本及像素级标注,缓解数据稀缺问题。在模型层面,构建非对称双流网络:教师网络提供稳定的正常特征表示,学生网络重建正常模式并放大正常与异常区域的差异。采用余弦相似性损失与像素级分割监督联合优化策略,实现微小缺陷的精确定位。在MVTecAD数据集上的实验表明,该方法图像级和像素级AUROC分别达到98.4%和98.3%,显著优于现有无监督及主流深度学习方法。所提方法无需大量真实缺陷样本,可实现准确可靠的工业缺陷检测与定位。

原文摘要 · Abstract (English)

Industrial surface defect detection often suffers from limited defect samples, severe long-tailed distributions, and difficulties in accurately localizing subtle defects under complex backgrounds. To address these challenges, this paper proposes an unsupervised defect detection method that integrates a Denoising Diffusion Probabilistic Model (DDPM) with an asymmetric teacher-student architecture. First, at the data level, the DDPM is trained solely on normal samples. By introducing constant-variance Gaussian perturbations and Perlin noise-based masks, high-fidelity and physically consistent defect samples along with pixel-level annotations are generated, effectively alleviating the data scarcity problem. Second, at the model level, an asymmetric dual-stream network is constructed. The teacher network provides stable representations of normal features, while the student network reconstructs normal patterns and amplifies discrepancies between normal and anomalous regions. Finally, a joint optimization strategy combining cosine similarity loss and pixel-wise segmentation supervision is adopted to achieve precise localization of subtle defects. Experimental results on the MVTecAD dataset show that the proposed method achieves 98.4\% image-level AUROC and 98.3\% pixel-level AUROC, significantly outperforming existing unsupervised and mainstream deep learning methods. The proposed approach does not require large amounts of real defect samples and enables accurate and robust industrial defect detection and localization. \keywords{Industrial defect detection \and diffusion models \and data generation \and teacher-student architecture \and pixel-level localization}

缺陷检测扩散模型无监督学习像素定位

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