基于知识蒸馏与自适应特征融合的工业缺陷检测方法
DOI:
作者:
作者单位:

西安邮电大学

作者简介:

通讯作者:

基金项目:

陕西省科学家与工程师创新团队项目(No.2023KXJ-091)


Knowledge Distillation with Selective Fusion Module for Industrial Defect Detection
Author:
Affiliation:

Xi’an University of Posts and Telecommunications

Fund Project:

Project of Innovation Team of Scientists and Engineers in Shaanxi Province(No.2023KXJ-091)

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
    摘要:

    本文针对工业生产中异常样本稀缺以及缺陷尺度变化剧烈问题,提出了一种基于知识蒸馏与自适应多尺度特征融合的无监督异常检测方法(STU-Seg)。该方法设计了一种异构的师生网络架构,将学生网络重构为U-Net结构并引入跳跃连接,通过复用浅层高分辨率特征,提升了模型对微小缺陷几何边界的识别与还原能力。同时,本文引入了自适应多尺度特征融合(SFM)模块,能够根据图像内容的全局上下文动态调整不同层级特征的权重,增强了模型对多尺度缺陷的感知。在MVTec AD基准数据集上的实验结果表明,STU-Seg模型的图像级异常检测指标与当前先进的无监督模型持平的同时,在像素级平均精度(Pixel AP)和实例级平均精度(IAP)上分别达到了72.43%和72.52%,均优于现有的主流无监督模型。

    Abstract:

    Aiming at the issues of scarcity of anomalous samples and drastic variations in defect scales in industrial production, this paper proposes the STU-Seg method, an unsupervised anomaly detection method based on knowledge distillation and a Selective Fusion Module (SFM). This method designs a heterogeneous student-teacher network architecture, in which the student network is reconstructed into a U-Net structure with skip connections. By reusing shallow high-resolution features, the model improves its ability to identify and restore the geometric boundaries of tiny defects. Meanwhile, the introduced SFM dynamically adjusts the weights of features at different levels according to the global context of the image content, enhancing the model's perception of multi-scale defects. Experimental results on the MVTec AD benchmark dataset show that while the image-level anomaly detection metrics of the STU-Seg model are on par with current state-of-the-art unsupervised models, it achieves 72.43% and 72.52% in Pixel-level Average Precision (Pixel AP) and Instance-level Average Precision (IAP) respectively, both outperforming existing mainstream unsupervised models.

    参考文献
    相似文献
    引证文献
引用本文
分享
文章指标
  • 点击次数:
  • 下载次数:
历史
  • 收稿日期:2025-11-28
  • 最后修改日期:2026-04-06
  • 录用日期:2026-05-18
  • 在线发布日期: