CN115019102A - 一种对抗样本生成模型的构建方法及应用 - Google Patents
一种对抗样本生成模型的构建方法及应用 Download PDFInfo
- Publication number
- CN115019102A CN115019102A CN202210693348.8A CN202210693348A CN115019102A CN 115019102 A CN115019102 A CN 115019102A CN 202210693348 A CN202210693348 A CN 202210693348A CN 115019102 A CN115019102 A CN 115019102A
- Authority
- CN
- China
- Prior art keywords
- image
- disturbance
- sample
- target
- singular value
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/764—Arrangements for image or video recognition or understanding using pattern recognition or machine learning using classification, e.g. of video objects
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F17/00—Digital computing or data processing equipment or methods, specially adapted for specific functions
- G06F17/10—Complex mathematical operations
- G06F17/16—Matrix or vector computation, e.g. matrix-matrix or matrix-vector multiplication, matrix factorization
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/74—Image or video pattern matching; Proximity measures in feature spaces
- G06V10/761—Proximity, similarity or dissimilarity measures
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/77—Processing image or video features in feature spaces; using data integration or data reduction, e.g. principal component analysis [PCA] or independent component analysis [ICA] or self-organising maps [SOM]; Blind source separation
- G06V10/774—Generating sets of training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/82—Arrangements for image or video recognition or understanding using pattern recognition or machine learning using neural networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V2201/00—Indexing scheme relating to image or video recognition or understanding
- G06V2201/07—Target detection
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- General Physics & Mathematics (AREA)
- Software Systems (AREA)
- Computing Systems (AREA)
- Evolutionary Computation (AREA)
- General Health & Medical Sciences (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Artificial Intelligence (AREA)
- Health & Medical Sciences (AREA)
- Databases & Information Systems (AREA)
- Mathematical Physics (AREA)
- Medical Informatics (AREA)
- Multimedia (AREA)
- Data Mining & Analysis (AREA)
- Computational Mathematics (AREA)
- Mathematical Optimization (AREA)
- Mathematical Analysis (AREA)
- General Engineering & Computer Science (AREA)
- Pure & Applied Mathematics (AREA)
- Algebra (AREA)
- Life Sciences & Earth Sciences (AREA)
- Biomedical Technology (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- Molecular Biology (AREA)
- Image Analysis (AREA)
Abstract
本发明公开了一种对抗样本生成模型的构建方法及应用,属于人工智能安全领域,所构建的模型包括级联的扰动模块和奖励计算模块;通过最大化图像样本集中的所有图像样本所对应的奖励的累积值,对对抗样本生成模型进行训练;当对抗样本生成模型训练完成后,扰动模块输出的扰动后的图像即为基于图像样本生成的对抗样本;本发明构建了一个基于强化学习的对抗样本生成模型,通过在对抗样本生成模型的训练过程中不断优化扰动值,使得查询朝向所期望的目标进行,规避了大量的重复查询,计算效率较高。除此之外,本发明所构建的模型并不依赖目标模型的梯度来生成对抗样本,实用性更强,适用范围更广。
Description
技术领域
本发明属于人工智能安全领域,更具体地,涉及一种对抗样本生成模型的构建方法及应用。
背景技术
随着人工智能的大规模应用,人工智能的安全问题逐渐成为限制人工智能应用规模的瓶颈,而人工智能安全中的对抗样本问题是保障人工智能安全中的关键问题,充分并高效的实现对抗样本的生成是实现人工智能安全的前提与要义。
为了更高效地生成神经网络中的对抗样本,需要通过一些算法对输入样本进行修改。其中,基于梯度的方法是一种有效的对抗样本生成算法,通过目标模型的梯度方向,指导对抗样本的生成过程,直到生成模型能够输出造成目标模型产生错误分类但与输入视觉相似的样本,即对抗样本。这种生成对抗样本的方法存在以下几个方面的问题,第一,由于基于梯度的方法需要沿梯度方向迭代修改样本以到达逃逸目的,因此,生成对抗样本时存在大量的重复查询,计算时间较长,计算效率较低;第二,该方法需要预先知晓目标模型的梯度,对模型梯度的依赖性较强,而现实应用中,目标模型的梯度是不可获取的;第三,该方法忽略了不同类别的图像在对抗样本的发现过程中存在不同约束数值的情况,对抗样本范数约束的差异性过小,精确度较低。
发明内容
针对现有技术的以上缺陷或改进需求,本发明提供了一种对抗样本生成模型的构建方法及应用,用以解决现有技术存在的计算效率较低的技术问题。
为了实现上述目的,第一方面,本发明提供了一种对抗样本生成模型的构建方法,包括以下步骤:
S1、搭建对抗样本生成模型;其中,对抗样本生成模型包括级联的扰动模块和奖励计算模块;
扰动模块用于对图像样本进行扰动,得到扰动后的图像;
奖励计算模块用于分别将扰动前和扰动后的图像输入到神经网络中进行目标检测,得到扰动前和扰动后的图像中的各目标类别及其对应的分类置信度;并计算扰动前和扰动后图像中各目标类别和对应分类置信度的变化,以及当前扰动前后和上一次扰动前后图像像素差异的比值,得到图像样本所对应的奖励;
S2、将图像样本集输入到对抗样本生成模型中,通过最大化图像样本集中的所有图像样本所对应的奖励的累积值,对对抗样本生成模型进行训练;当对抗样本生成模型训练完成后,扰动模块输出的扰动后的图像即为基于图像样本生成的对抗样本;
其中,图像样本集包括若干不同类别的图像样本及其对应的真实类别。
进一步优选地,上述扰动模块包括级联的奇异值分解模块、强化学习模型和整合模块;
奇异值分解模块用于对图像样本进行奇异值分解,得到图像样本的左奇异矩阵U、奇异值矩阵Σ和右奇异矩阵V;
强化学习模型用于获取奇异值矩阵Σ的扰动值;
整合模块用于基于奇异值矩阵Σ的扰动值对奇异值矩阵Σ进行扰动,得到扰动后的奇异值矩阵Σ';计算UΣ'VT,得到扰动后的图像。
进一步优选地,上述奇异值分解模块用于对图像样本按照通道进行奇异值分解,得到图像样本不同通道下的左奇异矩阵、奇异值矩阵和右奇异矩阵;
上述强化学习模型用于获取图像样本不同通道下的奇异值矩阵的扰动值;
其中,P为图像样本的通道数量;Ui为图像样本第i个通道下的左奇异矩阵;Σi'为图像样本第i个通道下扰动后的奇异值矩阵;Vi为图像样本第i个通道下的右奇异矩阵。
进一步优选地,上述扰动模块包括级联的目标提取模块、奇异值分解模块、强化学习模型和整合模块;
目标提取模块用于提取图像样本中的目标区域,得到目标样本;
奇异值分解模块用于对目标样本进行奇异值分解,得到目标样本的左奇异矩阵Ut、奇异值矩阵Σt和右奇异矩阵Vt;
强化学习模型用于获取奇异值矩阵Σt的扰动值;
整合模块用于基于奇异值矩阵Σt的扰动值对奇异值矩阵Σt进行扰动,得到扰动后的奇异值矩阵Σ't;计算得到UtΣt'Vt T后,替换图像样本中的目标区域,得到扰动后的图像。
进一步优选地,上述奇异值分解模块用于对目标样本按照通道进行奇异值分解,得到目标样本不同通道下的左奇异矩阵、奇异值矩阵和右奇异矩阵;
上述强化学习模型用于获取目标样本不同通道下的奇异值矩阵的扰动值;
其中,P为目标样本的通道数量;Ut_i为目标样本第i个通道下的左奇异矩阵;Σt_i’为目标样本第i个通道下扰动后的奇异值矩阵;Vt_i为目标样本第i个通道下的右奇异矩阵。
进一步优选地,图像样本所对应的奖励包括:
r1=P1w1
r2=P2w2
其中,r1为当前扰动前和扰动后的图像中的目标类别不一致时的奖励;P1为当前扰动前和扰动后的图像中目标类别不一致的总数量;w1为当前扰动前和扰动后的图像中的目标类别不一致时的权重值;r2为当前扰动后的图像中存在未识别出类别的目标时的奖励;P2为当前扰动后的图像中所存在的未识别出类别的目标数量;w2为当前扰动后的图像中存在未识别出类别的目标时的权重值;r3为与当前扰动前相比,当前扰动后图像中目标的分类置信度下降时的奖励;K为图像样本中的目标数量;w3为与当前扰动前相比,当前扰动后图像中目标的分类置信度下降时的权重值;Confj为当前扰动后图像中第j个目标的分类置信度;Confj'为当前扰动前图像中第j个目标的分类置信度;r4为隐蔽性约束惩罚项;t为当前扰动所对应的扰动轮次;Xt-1为当前扰动前的图像;Xt为当前扰动后的图像;Xt-2为当前扰动前的上一次扰动前的图像;||·||p表示p范数。
进一步优选地,步骤S2包括:将图像样本集输入到对抗样本生成模型中,在满足图像样本的扰动值不超过其真实类别的扰动上界的条件下,通过最大化图像样本集中的所有图像样本所对应的奖励的累积值,对对抗样本生成模型进行训练。
进一步优选地,类别L的扰动上界的获取方法包括:对真实类别为类别L的图像的奇异值进行重复扰动,直至扰动前后该图像出现像素级差异,此时,计算当前扰动前后图像像素的差异值作为类别L的扰动上界。
进一步优选地,类别L的扰动上界的获取方法包括:分别对真实类别为类别L的若干图像分别执行如下操作:对图像的奇异值进行重复扰动,直至扰动前后该图像出现像素级差异,此时,计算当前扰动前后图像像素的差异值作为该图像的扰动上界;
对所得各图像的扰动上界进行汇总,得到类别L的扰动上界。
第二方面,本发明提供了一种对抗样本生成方法,包括:将图像输入到采用上述对抗样本生成模型的构建方法所构建的对抗样本生成模型中,得到该图像所对应的对抗样本。
第三方面,本发明提供了一种对抗样本生成系统,包括:存储器和处理器,所述存储器存储有计算机程序,所述处理器执行所述计算机程序时执行本发明第二方面所提供的对抗样本生成方法。
第四方面,本发明提供了一种对抗样本防御方法,包括:
分别计算当前输入图像与对抗样本集中各对抗样本的相似度,若存在相似度小于预设阈值的情况,则将当前输入图像判定为对抗样本,将最小相似度所对应的对抗样本的原始图像类别作为当前输入图像的类别进行输出;
其中,上述对抗样本集的获取方法,包括:采用上述对抗样本生成方法,生成不同类别的原始图像所对应的对抗样本。
第五方面,本发明提供了一种对抗样本防御系统,包括:存储器和处理器,所述存储器存储有计算机程序,所述处理器执行所述计算机程序时执行本发明第四方面所提供的对抗样本防御方法。
第六方面,本发明还提供了一种计算机可读存储介质,所述计算机可读存储介质包括存储的计算机程序,其中,在所述计算机程序被处理器运行时控制所述存储介质所在设备执行本发明第一方面所提供的对抗样本生成模型的构建方法、本发明第二方面所提供的对抗样本生成方法、本发明第四方面所提供的对抗样本防御方法中的一种或多种。
总体而言,通过本发明所构思的以上技术方案,能够取得以下有益效果:
1、本发明提供了一种对抗样本生成模型的构建方法,构建了一个基于强化学习的对抗样本生成模型,考虑到扰动前后分类性能以及像素差异的变化来计算图像样本所对应的奖励,并通过最大化图像样本集中的所有图像样本所对应的奖励的累积值,对对抗样本生成模型进行训练,通过在对抗样本生成模型的训练过程中不断优化扰动值,使得查询朝向所期望的目标进行,规避了大量的重复查询,计算时间较短,计算效率较高。
2、采用本发明所提供的对抗样本生成模型的构建方法所构建的对抗样本生成模型,并不需要预先知晓目标模型的梯度,能够高效地针对包括目标检测、图像识别等多种计算机视觉任务的神经网络生成对抗样本,适用于现实应用,实用性更强,适用范围更广。
3、本发明所提供的对抗样本生成模型的构建方法,通过对奇异值矩阵进行扰动来对实现对输入图像的扰动,基于奇异值分解来获取结构化的特征,能够生成隐蔽性更强的对抗样本,所生成的对抗样本的质量较高。
4、本发明所提供的对抗样本生成模型的构建方法,在对图像进行扰动时,首先通过目标提取模块提取图像样本中的目标区域,然后再对目标区域的奇异值矩阵进行针对性扰动,从而降低了生成对抗样本所需要的迭代次数,同时避免了图像整体奇异值变化对于部分小目标没有影响的情况。
5、本发明所提供的对抗样本生成模型的构建方法,考虑到不同类别的图像在对抗样本的发现过程中存在不同约束数值,优先获取了图像不同类别的扰动上界,并在满足图像样本的扰动值不超过其真实类别的扰动上界的条件下,通过最大化图像样本集中的所有图像样本所对应的奖励的累积值,对对抗样本生成模型进行训练,该方法能够过滤明显不是对抗样本的图像,增加了对抗样本生成模型的精确度,同时也提高了对抗样本生成模型的训练效率。
6、本发明提供了一种对抗样本防御方法,通过本发明所提供的基于上述对抗样本生成模型的对抗样本生成方法,生成不同类别的原始图像所对应的对抗样本,进而将当前输入图像与对抗样本集中各对抗样本进行相似度比对,从而确定当前输入图像是否为对抗样本,防御效率较高。
附图说明
图1为本发明实施例1提供的对抗样本生成模型构建方法的流程图。
具体实施方式
为了使本发明的目的、技术方案及优点更加清楚明白,以下结合附图及实施例,对本发明进行进一步详细说明。应当理解,此处所描述的具体实施例仅仅用以解释本发明,并不用于限定本发明。此外,下面所描述的本发明各个实施方式中所涉及到的技术特征只要彼此之间未构成冲突就可以相互组合。
实施例1、
一种对抗样本生成模型的构建方法,如图1所示,包括以下步骤:
S1、搭建对抗样本生成模型;其中,对抗样本生成模型包括级联的扰动模块和奖励计算模块;
扰动模块用于对图像样本进行扰动,得到扰动后的图像;
奖励计算模块用于分别将扰动前和扰动后的图像输入到神经网络(本实施例优选为卷积神经网络)中进行目标检测,得到扰动前和扰动后的图像中的各目标类别及其对应的分类置信度;并计算扰动前和扰动后图像中各目标类别和对应分类置信度的变化,以及当前扰动前后和上一次扰动前后图像像素差异的比值,得到图像样本所对应的奖励;
S2、将图像样本集输入到对抗样本生成模型中,通过最大化图像样本集中的所有图像样本所对应的奖励的累积值,对对抗样本生成模型进行训练;当对抗样本生成模型训练完成后,扰动模块输出的扰动后的图像即为基于图像样本生成的对抗样本;
其中,图像样本集包括若干不同类别的图像样本及其对应的真实类别。
通过上述方法所学习的优化扰动策略在对抗样本生成模型的训练过程中不断优化扰动值,使得查询朝向所期望的目标进行,规避了大量的重复查询,计算时间较短,计算效率较高;除此之外,对抗样本生成模型能够在无需目标模型梯度的前提下,高效地针对包括目标检测、图像识别等多种计算机视觉任务的神经网络生成对抗样本。
具体地,图像样本所对应的奖励包括:
r1=P1w1
r2=P2w2
其中,r1为当前扰动前和扰动后的图像中的目标类别不一致时的奖励;P1为当前扰动前和扰动后的图像中目标类别不一致的总数量;w1为当前扰动前和扰动后的图像中的目标类别不一致时的权重值;r2为当前扰动后的图像中存在未识别出类别的目标时的奖励;P2为当前扰动后的图像中所存在的未识别出类别的目标数量;w2为当前扰动后的图像中存在未识别出类别的目标时的权重值;r3为与当前扰动前相比,当前扰动后图像中目标的分类置信度下降时的奖励;K为图像样本中的目标数量;w3为与当前扰动前相比,当前扰动后图像中目标的分类置信度下降时的权重值;Confj为当前扰动后图像中第j个目标的分类置信度;Confj'为当前扰动前图像中第j个目标的分类置信度;r4为隐蔽性约束惩罚项;t为当前扰动所对应的扰动轮次;Xt-1为当前扰动前的图像;Xt为当前扰动后的图像;Xt-2为当前扰动前的上一次扰动前的图像;||·||p表示p范数;这里的p范数可以为0范数、2范数和无穷范数,优选为2范数。
需要说明的是,r1和r2分别是两种对抗样本的直接发现途径,即原始分类和目标分类不一致或目标分类不存在;r3为潜在的对抗样本发现途径,即当原始分类的置信度出现下降的情况;r4为对抗样本的隐蔽性约束惩罚项,即使用发现的对抗样本与原始样本的p范数作为负的奖励值。
需要说明的是,扰动模块基于奇异值分解来获取图像样本的结构化特征,通过对图像样本或者其上的目标样本的奇异值矩阵进行扰动来对实现对图像样本的扰动;其中,对抗样本生成模型的具体扰动目标,可以是完整的图像样本,也可以是图像样本中包含的目标部分(即目标样本)。
在一种可选实施方式一下,上述扰动模块包括级联的奇异值分解模块、强化学习模型和整合模块;其中,奇异值分解模块用于对图像样本进行奇异值分解,得到图像样本的左奇异矩阵U、奇异值矩阵Σ和右奇异矩阵V;强化学习模型用于获取奇异值矩阵Σ的扰动值;整合模块用于基于奇异值矩阵Σ的扰动值对奇异值矩阵Σ进行扰动,得到扰动后的奇异值矩阵Σ';计算UΣ'VT,得到扰动后的图像。优选地,在上述可选实施方式一下,在对图像样本进行奇异值分解与整合时,上述奇异值分解模块用于对图像样本按照通道进行奇异值分解,得到图像样本不同通道下的左奇异矩阵、奇异值矩阵和右奇异矩阵。上述强化学习模型用于获取图像样本不同通道下的奇异值矩阵的扰动值。上述整合模块用于基于图像样本不同通道下的奇异值矩阵的扰动值分别对对应通道下的奇异值矩阵进行扰动,得到不同通道下扰动后的奇异值矩阵;计算得到扰动后的图像;其中,P为图像样本的通道数量;Ui为图像样本第i个通道下的左奇异矩阵;Σi'为图像样本第i个通道下扰动后的奇异值矩阵;Vi为图像样本第i个通道下的右奇异矩阵。
在另一种可选实施方式二下,为了利于加强目标物体的针对性扰动,降低生成对抗样本所需要的迭代次数,以及避免图像整体奇异值变化对于部分小目标没有影响的情况,上述扰动模块包括级联的目标提取模块、奇异值分解模块、强化学习模型和整合模块;其中,目标提取模块用于提取图像样本中的目标区域,得到目标样本;奇异值分解模块用于对目标样本进行奇异值分解,得到目标样本的左奇异矩阵Ut、奇异值矩阵Σt和右奇异矩阵Vt;强化学习模型用于获取奇异值矩阵Σt的扰动值;整合模块用于基于奇异值矩阵Σt的扰动值对奇异值矩阵Σt进行扰动,得到扰动后的奇异值矩阵Σ't;计算得到UtΣt'Vt T后,替换将图像样本中的目标区域,得到扰动后的图像。需要说明的是,上述目标提取模块可以通过已知标签或着目标检测器来标定目标样本的大小和位置,以提取图像样本中的目标区域。具体地,在给定的标签的情况下,目标样本的位置和大小通过标签标定的位置和大小获取;而在没有给定标签的情况下,目标样本的大小和位置可以通过现有的预训练好的目标检测器对图像样本进行检测获取。优选地,在上述可选实施方式二下,在对图像样本进行奇异值分解与整合时,上述奇异值分解模块用于对图像样本按照通道进行奇异值分解,得到图像样本不同通道下的左奇异矩阵、奇异值矩阵和右奇异矩阵。上述奇异值分解模块用于对目标样本按照通道进行奇异值分解,得到目标样本不同通道下的左奇异矩阵、奇异值矩阵和右奇异矩阵;上述强化学习模型用于获取目标样本不同通道下的奇异值矩阵的扰动值;上述整合模块用于基于目标样本不同通道下的奇异值矩阵的扰动值分别对对应通道下的奇异值矩阵进行扰动,得到不同通道下扰动后的奇异值矩阵;计算得到后,替换将目标样本中的目标区域,得到扰动后的图像;其中,P为目标样本的通道数量;Ut_i为目标样本第i个通道下的左奇异矩阵;Σt_i’为目标样本第i个通道下扰动后的奇异值矩阵;Vt_i为目标样本第i个通道下的右奇异矩阵。
优选地,为了过滤明显不是对抗样本的图像,增加对抗样本生成模型的精确度,提高对抗样本生成模型的训练效率;上述步骤S2包括:将图像样本集输入到对抗样本生成模型中,在满足图像样本的扰动值不超过其真实类别的扰动上界的条件下,通过最大化图像样本集中的所有图像样本所对应的奖励的累积值,对对抗样本生成模型进行训练。
具体地,在一种可选实施方式下,类别L的扰动上界的获取方法包括:对真实类别为类别L的图像的奇异值进行重复扰动,直至扰动前后该图像出现像素级差异,此时,计算当前扰动前后图像像素的差异值作为类别L的扰动上界。
在另一种可选实施方式下,类别L的扰动上界的获取方法包括:分别对真实类别为类别L的若干图像分别执行如下操作:对图像的奇异值进行重复扰动,直至扰动前后该图像出现像素级差异,此时,计算当前扰动前后图像像素的差异值作为该图像的扰动上界;对所得各图像的扰动上界进行汇总,得到类别L的扰动上界。具体地,可以将所得各图像的扰动上界的平均值作为类别L的扰动上界,还可以先对所得各图像的扰动上界使用格鲁布斯检验法去除异常值后,再取平均值作为类别L的扰动上界。本具体实施方式在测量输入图像扰动边界时,测量了超过十万张图像,并通过对照实验和极值过滤的方法获取了恰当的扰动边界值。
在上述两种可选实施方式下,在扰动的过程中计算p范数,也就是输入图像与迭代扰动的图像中每个元素的差值,之后用一个循环来寻找扰动值的临界点,使得扰动后的图像的分类结果与原始图像的分类检测结果不一致。
需要说明的是,p范数的约束公式为||x-x'||p;其中,x和x'分别代表输入图像和扰动后生成的图像,p代表范数的计算方式,可以为0、2和无穷。在第二种可选实施方式下,不仅考虑了单次修改的约束,还考虑了持续修改的累积约束,范数越大,它与原始样本的视觉差距就越大,成为潜在对抗样本的威胁性也就越低。
为了进一步说明本发明所提供的对抗样本生成模型的训练过程,下面以三维图像(P取值为3)为例进行详述,构建强化学习智能体(即对抗样本生成模型),定义其环境为扰动后的图像像素矩阵与扰动后的奇异值矩阵的拼接,其动作为对输入图像三通道奇异值矩阵的扰动,通过智能体持续扰动输入图像奇异值矩阵。
强化学习智能体被用来产生扰动后的图像,其探索与发现的方式如下:
优化的扰动策略为P*,通过计算最大化奖励R的期望来对于给定的状态s在当前策略p下产生的优化后的动作。对于输入的三维图像x∈Rm×n×3,由于三维数据直接分解的方法的计算复杂度和时间复杂度较高,本实施例对三维图像按照通道分别进行SVD分解。具体地,将输入的三通道图像按照通道分别做奇异值分解,得到三通道各自的奇异值矩阵;对原始图像的第i个通道的图像xi,其大小为m×n;对其进行奇异值分解为:xi=UiΣiVi T,i=1,2,3;通过计算的三通道奇异值矩阵,得到Σi=[xivi (1)/ui (1),…,xivi (r)/ui (r)];其中,r为xi的秩,ui为图像第i个通道的左奇异向量(ui (r)为ui中的第r个值),vi为图像第i个通道的右奇异向量(vi (r)为vi中的第r个值);然后生成强化学习的动作A=[a1,a2,a3],其中,a1、a2和a3分别为三维图像不同通道所对应的奇异值矩阵的扰动值;按通道将奇异值矩阵与其对应的扰动值相加得到Σ'=[Σ1',Σ2',Σ3'];通过奇异值分解的逆向操作,得到新的图像即扰动后的状态。
上述强化学习智能体根据自身包含的值函数网络ν(·)和Q网络进行更新:
其中,λ是折扣因子,其取值通常为0.99,rt为t时刻智能体获得的奖励值,st+1为t+1时刻强化学习智能体的状态。值函数网络ν(·)的更新方法为:
值函数的含义是在时序步骤t+1、在状态s上执行动作a所获得的预期奖励,因此,用于优化策略的损失函数的计算方式为:
其中,H0是熵的阈值,通常取值为-2;α是温度超参数,用于优化策略的随机性。
本实施例通过依次更新策略网络PΦ和值函数网络QΛ来更新深度强化学习智能体。当深度强化学习智能体的迭代轮次超过某一阈值(本实施例中设置为1000)时,终止迭代终止。
实施例2、
一种对抗样本生成方法,包括:将图像输入到采用本发明实施例1所提供的对抗样本生成模型的构建方法所构建的对抗样本生成模型中,得到该图像所对应的对抗样本。
相关技术方案同实施例1,这里不做赘述。
实施例3、
一种对抗样本生成系统,包括:存储器和处理器,所述存储器存储有计算机程序,所述处理器执行所述计算机程序时执行本发明实施例2所提供的对抗样本生成方法。
相关技术方案同实施例2,这里不做赘述。
实施例4、
一种对抗样本防御方法,包括:
分别计算当前输入图像与对抗样本集中各对抗样本的相似度,若存在相似度小于预设阈值的情况,则将当前输入图像判定为对抗样本,并对当前输入图像进行标记并拒绝当前输入图像进入目标系统,同时将最小相似度所对应的对抗样本的原始图像类别作为当前输入图像的类别进行输出;否则,将当前输入图像输入到神经网络中进行目标检测;本实施例中,上述预设阈值取值为0.1;
其中,上述对抗样本集的获取方法,包括:采用本发明实施例2所提供的对抗样本生成方法,生成不同类别的原始图像所对应的对抗样本。
需要说明的是,计算当前输入图像与对抗样本集中各对抗样本的相似度的方法有多种,本实施例中,通过计算当前输入图像与对抗样本集中各对抗样本的余弦相似度来对二者之间的相似程度进行度量。具体地,余弦相似度的度量公式为:其中,x1和x2分别为当前输入图像和对抗样本集中的某一对抗样本。通过计算新的输入图像与已有对抗样本的余弦相似度实现对抗样本的防御。
相关技术方案同实施例2,这里不做赘述。
实施例5、
一种对抗样本防御系统,包括:存储器和处理器,所述存储器存储有计算机程序,所述处理器执行所述计算机程序时执行本发明实施例4所提供的对抗样本防御方法。
相关技术方案同实施例4,这里不做赘述。
实施例6、
本发明还提供了一种计算机可读存储介质,所述计算机可读存储介质包括存储的计算机程序,其中,在所述计算机程序被处理器运行时控制所述存储介质所在设备执行本发明实施例1所提供的对抗样本生成模型的构建方法、本发明实施例2所提供的对抗样本生成方法、本发明实施例4所提供的对抗样本防御方法中的一种或多种。
相关技术方案同实施例1、实施例2和实施例4,这里不做赘述。
综上本发明的实施例1-实施例6可知,本发明在测量图像类别的扰动边界时,通过对照实验和极值过滤的方法获取恰当的扰动边界值。在输入图像奇异值分解时,对三通道的输入分别做三次奇异值分解,并将获取的奇异值进行存储,有效增强了视觉隐蔽性强的对抗样本发现。在循环扰动图像奇异值矩阵时,应用深度强化学习训练得到一个优化的扰动策略,从而降低每发现一个对抗样本所需的查询次数。在对抗样本生成与防御时,通过设置合适的奖励函数以优化生成的过程,并通过奇异值矩阵的相似度比较方法来防御已发现的样本。理论和实验证明,与现有的DPatch、AutoAttacker、ZOO等方法相比,本发明所提供的对抗样本生成模型在对抗样本生成精度,样本迁移能力,查询次数等方面均有更优的表现。
本领域的技术人员容易理解,以上所述仅为本发明的较佳实施例而已,并不用以限制本发明,凡在本发明的精神和原则之内所作的任何修改、等同替换和改进等,均应包含在本发明的保护范围之内。
Claims (10)
1.一种对抗样本生成模型的构建方法,其特征在于,包括以下步骤:
S1、搭建对抗样本生成模型;所述对抗样本生成模型包括级联的扰动模块和奖励计算模块;
所述扰动模块用于对图像样本进行扰动,得到扰动后的图像;
所述奖励计算模块用于分别将扰动前和扰动后的图像输入到神经网络中进行目标检测,得到扰动前和扰动后的图像中的各目标类别及其对应的分类置信度;并计算扰动前和扰动后图像中各目标类别和对应分类置信度的变化,以及当前扰动前后和上一次扰动前后图像像素差异的比值,得到图像样本所对应的奖励;
S2、将图像样本集输入到所述对抗样本生成模型中,通过最大化所述图像样本集中的所有图像样本所对应的奖励的累积值,对所述对抗样本生成模型进行训练;当所述对抗样本生成模型训练完成后,所述扰动模块输出的扰动后的图像即为基于图像样本生成的对抗样本;
其中,所述图像样本集包括若干不同类别的图像样本及其对应的真实类别。
2.根据权利要求1所述的对抗样本生成模型的构建方法,其特征在于,所述扰动模块包括级联的奇异值分解模块、强化学习模型和整合模块;
所述奇异值分解模块用于对图像样本进行奇异值分解,得到图像样本的左奇异矩阵U、奇异值矩阵Σ和右奇异矩阵V;
所述强化学习模型用于获取所述奇异值矩阵Σ的扰动值;
所述整合模块用于基于所述奇异值矩阵Σ的扰动值对所述奇异值矩阵Σ进行扰动,得到扰动后的奇异值矩阵Σ';计算UΣ'VT,得到扰动后的图像。
4.根据权利要求1所述的对抗样本生成模型的构建方法,其特征在于,所述扰动模块包括级联的目标提取模块、奇异值分解模块、强化学习模型和整合模块;
所述目标提取模块用于提取图像样本中的目标区域,得到目标样本;
所述奇异值分解模块用于对所述目标样本进行奇异值分解,得到所述目标样本的左奇异矩阵Ut、奇异值矩阵Σt和右奇异矩阵Vt;
所述强化学习模型用于获取所述奇异值矩阵Σt的扰动值;
所述整合模块用于基于所述奇异值矩阵Σt的扰动值对所述奇异值矩阵Σt进行扰动,得到扰动后的奇异值矩阵Σ't;计算得到UtΣt'Vt T后,替换所述图像样本中的目标区域,得到扰动后的图像。
5.根据权利要求1-4任意一项所述的对抗样本生成模型的构建方法,其特征在于,所述图像样本所对应的奖励包括:
r1=P1w1
r2=P2w2
其中,r1为当前扰动前和扰动后的图像中的目标类别不一致时的奖励;P1为当前扰动前和扰动后的图像中目标类别不一致的总数量;w1为当前扰动前和扰动后的图像中的目标类别不一致时的权重值;r2为当前扰动后的图像中存在未识别出类别的目标时的奖励;P2为当前扰动后的图像中所存在的未识别出类别的目标数量;w2为当前扰动后的图像中存在未识别出类别的目标时的权重值;r3为与当前扰动前相比,当前扰动后图像中目标的分类置信度下降时的奖励;K为图像样本中的目标数量;w3为与当前扰动前相比,当前扰动后图像中目标的分类置信度下降时的权重值;Confj为当前扰动后图像中第j个目标的分类置信度;Confj'为当前扰动前图像中第j个目标的分类置信度;r4为隐蔽性约束惩罚项;t为当前扰动所对应的扰动轮次;Xt-1为当前扰动前的图像;Xt为当前扰动后的图像;Xt-2为当前扰动前的上一次扰动前的图像;||·||p表示p范数。
6.根据权利要求1-4任意一项所述的对抗样本生成模型的构建方法,其特征在于,所述步骤S2包括:将所述图像样本集输入到所述对抗样本生成模型中,在满足图像样本的扰动值不超过其真实类别的扰动上界的条件下,通过最大化所述图像样本集中的所有图像样本所对应的奖励的累积值,对所述对抗样本生成模型进行训练。
7.根据权利要求6所述的对抗样本生成模型的构建方法,其特征在于,类别L的扰动上界的获取方法包括:方式一和方式二;
所述方式一包括:对真实类别为类别L的图像的奇异值进行重复扰动,直至扰动前后该图像出现像素级差异,此时,计算当前扰动前后图像像素的差异值作为类别L的扰动上界;
所述方式二包括:分别对真实类别为类别L的若干图像分别执行如下操作:对图像的奇异值进行重复扰动,直至扰动前后该图像出现像素级差异,此时,计算当前扰动前后图像像素的差异值作为该图像的扰动上界;
对所得各图像的扰动上界进行汇总,得到类别L的扰动上界。
8.一种对抗样本生成方法,其特征在于,包括:将图像输入到采用权利要求1-7任意一项所述的对抗样本生成模型的构建方法所构建的对抗样本生成模型中,得到该图像所对应的对抗样本。
9.一种对抗样本防御方法,其特征在于,包括:分别计算当前输入图像与对抗样本集中各对抗样本的相似度,若存在相似度小于预设阈值的情况,则将当前输入图像判定为对抗样本,将最小相似度所对应的对抗样本的原始图像类别作为当前输入图像的类别进行输出;
其中,所述对抗样本集的获取方法,包括:采用权利要求8所述的对抗样本生成方法,生成不同类别的原始图像所对应的对抗样本。
10.一种计算机可读存储介质,所述计算机可读存储介质包括存储的计算机程序,其中,在所述计算机程序被处理器运行时控制所述存储介质所在设备执行权利要求1-7任意一项所述的对抗样本生成模型的构建方法、权利要求8所述的对抗样本生成方法、以及权利要求9所述的对抗样本防御方法中的一种或多种。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210693348.8A CN115019102A (zh) | 2022-06-17 | 2022-06-17 | 一种对抗样本生成模型的构建方法及应用 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210693348.8A CN115019102A (zh) | 2022-06-17 | 2022-06-17 | 一种对抗样本生成模型的构建方法及应用 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN115019102A true CN115019102A (zh) | 2022-09-06 |
Family
ID=83074828
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202210693348.8A Pending CN115019102A (zh) | 2022-06-17 | 2022-06-17 | 一种对抗样本生成模型的构建方法及应用 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN115019102A (zh) |
Cited By (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116486463A (zh) * | 2023-06-15 | 2023-07-25 | 北京瑞莱智慧科技有限公司 | 图像处理方法、相关装置及存储介质 |
CN117934819A (zh) * | 2024-03-20 | 2024-04-26 | 中铁第六勘察设计院集团有限公司 | 一种轨道缺陷检测系统鲁棒性提升方法 |
-
2022
- 2022-06-17 CN CN202210693348.8A patent/CN115019102A/zh active Pending
Cited By (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116486463A (zh) * | 2023-06-15 | 2023-07-25 | 北京瑞莱智慧科技有限公司 | 图像处理方法、相关装置及存储介质 |
CN116486463B (zh) * | 2023-06-15 | 2023-10-03 | 北京瑞莱智慧科技有限公司 | 图像处理方法、相关装置及存储介质 |
CN117934819A (zh) * | 2024-03-20 | 2024-04-26 | 中铁第六勘察设计院集团有限公司 | 一种轨道缺陷检测系统鲁棒性提升方法 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
Khemakhem et al. | Ice-beem: Identifiable conditional energy-based deep models based on nonlinear ica | |
Lin et al. | Towards optimal structured cnn pruning via generative adversarial learning | |
Gu et al. | Stack-captioning: Coarse-to-fine learning for image captioning | |
Ghosh et al. | Structured variational learning of Bayesian neural networks with horseshoe priors | |
Onken et al. | Discretize-optimize vs. optimize-discretize for time-series regression and continuous normalizing flows | |
CN111382868A (zh) | 神经网络结构搜索方法和神经网络结构搜索装置 | |
CN113220886A (zh) | 文本分类方法、文本分类模型训练方法及相关设备 | |
EP4287144A1 (en) | Video behavior recognition method and apparatus, and computer device and storage medium | |
CN112699941B (zh) | 植物病害严重程度图像分类方法、装置、设备和存储介质 | |
Kutyniok | The mathematics of artificial intelligence | |
CN113254927B (zh) | 一种基于网络防御的模型处理方法、装置及存储介质 | |
CN112766603A (zh) | 一种交通流量预测方法、系统、计算机设备及存储介质 | |
Perugachi-Diaz et al. | Invertible densenets with concatenated lipswish | |
CN115496144A (zh) | 配电网运行场景确定方法、装置、计算机设备和存储介质 | |
CN115019102A (zh) | 一种对抗样本生成模型的构建方法及应用 | |
CN113553918B (zh) | 一种基于脉冲主动学习的机打发票字符识别方法 | |
Chen et al. | Active multi-task representation learning | |
CN114494783A (zh) | 一种基于动态图神经网络的预训练方法 | |
CN117114053A (zh) | 基于结构搜索和知识蒸馏的卷积神经网络模型压缩方法和装置 | |
CN116824334A (zh) | 一种基于频域特征融合再构的模型后门攻击对抗方法 | |
Zha et al. | Invertible attention | |
CN116721327A (zh) | 一种基于泛化界的神经网络架构搜索方法 | |
Liu et al. | Flexi-compression: a flexible model compression method for autonomous driving | |
CN115630361A (zh) | 一种基于注意力蒸馏的联邦学习后门防御方法 | |
CN115834153A (zh) | 一种基于节点投票机制的图神经网络模型黑盒攻击装置及方法 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination |