基于联合对抗训练的深度学习表面缺陷检测方法

Union-adversarial Training for Deep Learning based on Surface Defect Detection

  • 摘要: 表面缺陷检测是确保产品质量的重要途径。深度学习在表面缺陷检测中已经得到了大量应用,并为实现自动化的表面缺陷检测提供有效的技术途径。然而,深度学习模型容易受到对抗攻击的威胁,进而严重影响其准确性,甚至造成模型失效。为提升深度学习模型对对抗攻击算法的防御能力,基于对抗训练提出新的基于联合对抗训练的深度学习模型。对抗攻击算法包括单步攻击算法和迭代攻击算法,为提升所提联合对抗深度学习模型的防御能力,在对深度学习模型训练时即同时添加不同对抗攻击算法生成的对抗样本,同时增强模型对单步对抗攻击算法和迭代对抗攻击算法的防御能力。为验证所提算法的有效性,所提算法以CIFAR 10和磁瓦表面缺陷数据集为基础进行验证。试验结果表明,所提的联合对抗训练深度学习模型能显著提升对单步对抗攻击与迭代对抗攻击的鲁棒性,并优于传统方法。

     

    Abstract: Surface defect detection(SDD) is the crucial way to ensure product quality. Deep learning(DL) has been extensively adopted in industry to support the automatic surface defect detection. However, DL model is vulnerable to the menace of adversarial attack, which would seriously affect the accuracy of DL model or even cause the failure of the DL model. To overcome the drawback, a new union adversarial training(UNT) based DL model is proposed to promote its defense ability on adversarial attack. As adversarial attack includes the single-step adversarial attack and iterative adversarial attack, various adversarial samples are generated by adversarial attack methods are used for training the DL models to enhance the defense ability against both the single-step and iterative adversarial attack algorithms. The experiments are conducted on the CIFAR 10 dataset and magnetic-tile surface detect detection dataset. The results show that the proposed UNT method for DL based SDD can significantly promote the robustness of DL model to the single-step adversarial attack and iterative adversarial attack, which is superior to traditional methods.

     

/

返回文章
返回