用弱监督方法实现高精度工业缺陷分割,无需大量标注数据。
Region-Aware CAM: High-Resolution Weakly-Supervised Defect Segmentation via Salient Region Perception

- 通过区域感知的CAM和梯度过滤技术提升缺陷定位精度。
- 在多个工业数据集上达到优于现有弱监督方法的分割性能。
- 适合标注成本高的工业质检场景,尤其适用于小样本缺陷检测。
表面缺陷检测在工业质量检验中至关重要。近年来人工智能的进步显著提升了检测自动化水平,但传统语义分割与目标检测模型严重依赖大规模标注数据,与实际缺陷检测需求相悖。本文提出一种新型弱监督语义分割框架,包含两个核心组件:区域感知类激活图(CAM)与伪标签训练。为克服现有CAM方法分辨率低、细节保留不足的问题,提出滤波引导反向传播(FGBP),通过过滤梯度幅值识别缺陷相关区域;在此基础上构建区域感知加权模块以增强空间精度。最后通过伪标签分割实现模型性能的迭代优化。在多个工业缺陷数据集上的实验表明,该方法显著优于现有弱监督方法,有效弥合弱监督学习与高精度缺陷分割之间的差距,为资源受限的工业场景提供可行解决方案。
原文摘要 · Abstract (English)
Surface defect detection plays a critical role in industrial quality inspection. Recent advances in artificial intelligence have significantly enhanced the automation level of detection processes. However, conventional semantic segmentation and object detection models heavily rely on large-scale annotated datasets, which conflicts with the practical requirements of defect detection tasks. This paper proposes a novel weakly supervised semantic segmentation framework comprising two key components: a region-aware class activation map (CAM) and pseudo-label training. To address the limitations of existing CAM methods, especially low-resolution thermal maps, and insufficient detail preservation, we introduce filtering-guided backpropagation (FGBP), which refines target regions by filtering gradient magnitudes to identify areas with higher relevance to defects. Building upon this, we further develop a region-aware weighted module to enhance spatial precision. Finally, pseudo-label segmentation is implemented to refine the model's performance iteratively. Comprehensive experiments on industrial defect datasets demonstrate the superiority of our method. The proposed framework effectively bridges the gap between weakly supervised learning and high-precision defect segmentation, offering a practical solution for resource-constrained industrial scenarios.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。