融合小波变换卷积和知识蒸馏的PCB缺陷检测模型
DOI:
CSTR:
作者:
作者单位:

兰州交通大学电子与信息工程学院兰州730070

作者简介:

通讯作者:

中图分类号:

TP391TH862

基金项目:

青海理工大学“昆仑英才”人才引进科研项目(W2023-QLGKLYCZX-034)资助


Integrating wavelet transform convolution and knowledge distillation for efficient PCB defect detection
Author:
Affiliation:

School of Electronic and Information Engineering, Lanzhou Jiaotong University, Lanzhou 730070, China

Fund Project:

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

    针对印制电路板(PCB)缺陷检测任务中存在的缺陷形态复杂、背景干扰强以及工业现场对实时性与轻量化部署的严苛需求,提出了一种融合小波变换与知识蒸馏的高效轻量化检测模型——KDYOLOv8。首先,设计Star-YOLO主干网络,利用星形操作将输入特征映射至高维非线性空间,在大幅降低计算冗余的同时增强对复杂缺陷模式的特征提取能力。其次,引入小波变换卷积(WTConv),通过多分辨率分析技术解耦图像的高频缺陷细节与低频背景纹理,在不显著增加参数量的前提下有效抑制噪声干扰并扩展感受野。同时,构建增强型多尺度双向特征金字塔网络(EMBFPN),利用双向信息流交互机制强化深浅层特征融合,解决深层网络中微小缺陷特征稀释的问题。此外,采用通道级知识蒸馏策略(CWD),引导轻量化模型学习教师网络的通道注意力分布,补偿模型压缩带来的精度损失。实验结果表明,在公开PCB缺陷数据集上,KDYOLOv8的平均精度均值(mAP)达到97.1%,模型体积仅2.9 MB,推理速度高达 117.3 fps;相较于基线YOLOv8n,在体积缩减52.5%的情况下依然保持了高精度。在跨数据集泛化实验中,模型对“鼠咬”和“短路”等细微缺陷的检测精度分别提升了1.9% 和1.6%。有效平衡了检测速度、精度与资源消耗,能够为算力受限环境下的工业部署提供有力支持。

    Abstract:

    To address the challenges of complex defect morphology, strong background interference, and the strict requirements for real-time performance and lightweight deployment in printed circuit board (PCB) defect detection, an efficient lightweight detection model named KDYOLOv8 is proposed in this article, which integrates wavelet transform and knowledge distillation. Firstly, a Star-YOLO backbone network is designed, utilizing star operation to map input features into a high-dimensional non-linear space, thereby enhancing feature extraction capabilities for complex defect patterns while significantly reducing computational redundancy. Secondly, the wavelet transform convolution (WTConv) is introduced to decouple high-frequency defect details from low-frequency background textures through multi-resolution analysis, effectively suppressing noise interference and expanding the receptive field without significantly increasing parameters. Meanwhile, an EMBFPN enhanced multi-scale bi-directional feature pyramid network is constructed, employing a bi-directional information flow interaction mechanism to strengthen the fusion of deep and shallow features, addressing the problem of small defect feature dilution in deep networks. Furthermore, a channel-wise knowledge distillation (CWD) strategy is adopted to guide the lightweight model in learning the channel attention distribution of the teacher network, compensating for accuracy loss caused by model compression. Experimental results show that, on a public PCB defect dataset, KDYOLOv8 achieves a mean average precision (mAP) of 97.1%, with a model size of only 2.9 MB and an inference speed of 117.3 fps. Compared with the baseline YOLOv8n, it maintains high accuracy while reducing the volume by 52.5%. In cross-dataset generalization experiments, the detection accuracy for subtle defects such as “mouse bite” and “short” improved by 1.9% and 1.6%, respectively. This study effectively balances detection speed, accuracy, and resource consumption, providing strong support for industrial deployment in resource-constrained environments.

    参考文献
    相似文献
    引证文献
引用本文

刘春娟,赵浩然,张明璇,闫浩文,吴小所.融合小波变换卷积和知识蒸馏的PCB缺陷检测模型[J].仪器仪表学报,2025,46(12):87-99

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-03-02
  • 出版日期:
文章二维码