[发明专利]一种基于交通标志目标检测器改进的对抗样本生成方法

说明书

本发明公开了一种基于交通标志目标检测器改进的对抗样本生成方法，先通过随机RGB图像生成算法生成一个初始对抗样本；再通过目标检测器对原始训练样本进行训练，获取各原始训练样本上交通标志的检测框；然后对初始对抗样本进行一系列数据增强操作附加在原始训练样本上形成对抗训练样本，计算交通标志仍然被检测到的置信度；最后通过置信度来衡量对抗样本性能的损失函数。目前由于大多数的对抗方法主要的对抗目标针是大目标物体，而对小目标物体的对抗效果较差，本发明融合了一个新的损失函数来进一步优化对抗目标以应对该问题，该损失函数通过最小化目标检测器所预测的视觉距离和真实的视觉距离之间的偏差，从而来提高对抗样本的鲁棒性。

技术领域

本发明涉及一种基于交通标志目标检测器改进的对抗样本生成方法，属于计算机图形处理技术和人工智能技术。

背景技术

随着深度学习领域的不断发展以及计算机性能的快速提高，计算机视觉领域的发展取得了巨大的成功。而在计算机视觉领域中，卷积神经网络更是其主要的代表技术之一，并且已经在图像识别、定位、视频跟踪和视频分割等领域得到了广泛的应用。

CNN在图像上的端到端的深度学习处理方法已经在广泛的计算机视觉任务中取得了良好的效果。通过多层的网络结构，神经网络能够让输入模块从网络底层的过滤器学习到具有上下文信息的高级特征。为了做到这一点，一个典型的CNN通常包含数以百万计的学习参数。虽然这种方法能够得到非常精确的模型，但其解释性显著降低，并且已经被证明容易受到特定对抗干扰的欺骗。这种缺陷对于非安全系统的应用程序来说影响相对较小，但是在安全系统中可能会造成严重的实际问题。比如在无人驾驶系统中交通标志检测模型中的一个弱点可能被用来恶意导致车载识别摄像头检测失效。对抗性攻击也可用于破坏欺诈检测、恶意软件检测或误导自主导航系统等。Thys等人最近的一项研究进一步证实了这些结果。该研究结果显示，通过一种对抗块生成算法生成一个具有对抗性的“样本”，能够导致主流的目标检测器对人的检测失效或者分类错误，但是该算法针对交通标志检测性能较差。

发明内容

发明目的：为了克服现有技术中存在的不足，本发明提供一种基于交通标志目标检测器改进的对抗样本生成方法，通过最小化检测器所看到的感知距离和真实距离之间的偏差，从而来提高对抗样本的鲁棒性。

技术方案：为实现上述目的，本发明采用的技术方案为：

一种基于交通标志目标检测器改进的对抗样本生成方法，包括如下步骤：

步骤1：通过随机RGB图像生成算法生成一个初始对抗样本；

步骤2：选取图像作为原始训练样本，将图像上的交通标志作为对抗目标；所有原始训练样本构成原始训练样本集；

步骤3：基于Tiny-YOLO目标检测器，通过非极大值抑制算法对原始训练样本进行交通标志检测，得到各原始训练样本上交通标志的检测框；

步骤4：对对抗样本进行数据增强操作后附加在各原始训练样本的检测框中心位置形成敌对训练样本；

步骤5：基于Tiny-YOLO目标检测器，通过非极大值抑制算法对敌对训练样本进行交通标志检测，计算相对于原始训练样本交通标志在敌对训练样本中仍被检测到的置信度；通过置信度衡量对抗样本性能的损失函数；

步骤6：通过网络的反向传播，使用Adam优化器(自适应矩估计优化器)改变对抗样本的像素值，生成新的对抗样本；

步骤7：重复执行步骤4～6，直至对抗样本性能的损失函数收敛或达到迭代次数，最终获得一个具有高鲁棒性、高攻击性的对抗样本。

优选的，所述步骤5中，通过置信度衡量对抗样本性能的损失函数，该损失函数通过下式计算：

L＝αL_nps+βL_tv+L_obj+L_det