arXiv:2605.17780cs.CV2026-05

用先验知识指导训练,让缺陷检测模型更准更可信。

Network Knowledge Prior Guided Learning for Data-Efficient Surface Defect Detection

论文配图:Network Knowledge Prior Guided Learning for Data-Efficient Surface Defect Detection
图 1 · 摘自论文原文
  • 用主模型的热力图作为先验知识,指导新模型训练。
  • 在多个数据集上提升准确率和平均精度,热力图更清晰。
  • 无需额外计算成本,适合工业质检场景使用。

基于深度学习的工业缺陷检测已成为主流方法,但其对数据依赖性强且缺乏可解释性,限制了实际应用中的性能与可信度。本文提出一种新型知识引导损失函数,将模型可解释性无缝融入训练过程,不增加推理开销。方法分为两阶段:首先训练一个主分类网络并生成其显著性图作为先验知识;其次构建多任务框架,主任务完成分类,辅助任务通过专用知识引导损失项强制最终模型与主模型的显著性图保持一致,从而有效正则化模型,促进鲁棒特征学习。在多个公开缺陷数据集上的实验表明,该方法能持续提升基线模型的准确率与平均精度(AP)。视觉分析显示,所提方法生成的显著性图更加集中且符合人类认知。本工作提供了一种简单高效的范式,弥合了模型性能与可解释性之间的差距,为工业质量检测中更可靠、高性能的视觉系统铺平道路。

原文摘要 · Abstract (English)

Deep learning-based methods have become the de facto standard for industrial defect detection. However, their data-hungry nature and inherent "black-box" characteristics often lead to performance bottlenecks and limited trustworthiness in real-world applications. To address these challenges, this paper proposes a novel knowledge-guided loss function that seamlessly integrates model interpretability into the training process without incurring any additional inference cost. Our method operates in two phases: first, a primary classification network is trained, and its explanations, in the form of saliency maps, are generated as prior knowledge. Second, a multi-task learning framework is established, where the main task performs classification, and an auxiliary task imposes consistency between the saliency maps of the final model and the primary model. This consistency is enforced by a dedicated knowledge-guided loss term, effectively acting as a powerful regularizer to steer the model towards robust feature representations. Extensive experiments on multiple public defect datasets demonstrate that our approach consistently enhances the performance of baseline models in terms of accuracy and AP. Moreover, visual analysis reveals that the proposed method yields more concentrated and human-intelligible saliency maps. This work presents a simple yet effective paradigm for bridging the gap between model performance and interpretability, paving the way for more reliable and high-performing vision systems in industrial quality inspection.

缺陷检测可解释性多任务学习

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