提出CAT框架,提升金属表面缺陷检测在复杂场景下的精度与泛化能力。
Contrastive Augmented Transformer with Domain-specific Enhancement for Robust Multi-scenario Metal Surface Defect Detection

- 采用分层Swin Transformer与重设计的特征金字塔融合多尺度纹理与语义
- 在KolektorSDD2上达到99.54%像素级AUROC,显著优于现有方法
- 适用于多种工业场景,尤其适合标注数据少、噪声大的真实生产环境
金属表面缺陷检测对工业制造中的产品质量至关重要。然而,该任务面临标注数据有限、细微多尺度缺陷难识别以及跨场景泛化能力差等挑战。为此,本文提出一种新型对比增强变换器(CAT)框架,以实现鲁棒的缺陷检测。CAT采用分层Swin Transformer主干网络,并重构特征金字塔网络,有效融合低层纹理与高层语义信息,精确建模细微且多尺度的缺陷模式。为增强模型在真实噪声条件下的鲁棒性,提出一种领域特定的液滴增强算法;同时,在对比损失中引入困难负样本挖掘策略,强化模型在模糊缺陷区域的判别能力。在KolektorSDD2数据集上的实验结果表明,CAT达到99.54%的像素级AUROC,优于现有方法。此外,CAT在三个未见数据集(KSDD1、MTD用于瓷砖缺陷、MSDD用于轨道表面缺陷)上均展现出优异的泛化与鲁棒性,具备广泛工业部署潜力。
原文摘要 · Abstract (English)
Metal surface defect detection is critical for maintaining product quality in industrial manufacturing. However, it faces significant challenges, including limited annotated data, difficulty in identifying subtle multi-scale defects, and poor generalization across diverse scenarios. To address these issues, this paper proposes a novel Contrastive Augmented Transformer (CAT) framework for robust defect detection. CAT employs a hierarchical Swin Transformer backbone and redesigns the feature pyramid network to effectively fuse low-level textures with high-level semantics, enabling precise modeling of subtle and multi-scale defect patterns. To enhance robustness under real-world noise conditions, we propose a domain-specific droplet augmentation algorithm. Furthermore, we incorporate a hard negative mining strategy into the contrastive loss to strengthen the model's discrimination ability in ambiguous defect regions. Experimental results on the KolektorSDD2 dataset demonstrate that CAT achieves a pixel-level AUROC of 99.54%, outperforming existing methods. In addition, CAT exhibits superior generalization and robustness on three unseen datasets, including KSDD1, MTD for tile defects, and MSDD for rail surface defects, demonstrating its potential for wide-scale industrial deployment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。