CN112199543B - 一种基于图像检索模型的对抗样本生成方法 - Google Patents
一种基于图像检索模型的对抗样本生成方法 Download PDFInfo
- Publication number
- CN112199543B CN112199543B CN202011098487.3A CN202011098487A CN112199543B CN 112199543 B CN112199543 B CN 112199543B CN 202011098487 A CN202011098487 A CN 202011098487A CN 112199543 B CN112199543 B CN 112199543B
- Authority
- CN
- China
- Prior art keywords
- image
- sample
- image retrieval
- retrieval
- training
- 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.)
- Active
Links
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F16/00—Information retrieval; Database structures therefor; File system structures therefor
- G06F16/50—Information retrieval; Database structures therefor; File system structures therefor of still image data
- G06F16/58—Retrieval characterised by using metadata, e.g. metadata not derived from the content or metadata generated manually
- G06F16/583—Retrieval characterised by using metadata, e.g. metadata not derived from the content or metadata generated manually using metadata automatically derived from the content
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F16/00—Information retrieval; Database structures therefor; File system structures therefor
- G06F16/50—Information retrieval; Database structures therefor; File system structures therefor of still image data
- G06F16/55—Clustering; Classification
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/22—Matching criteria, e.g. proximity measures
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/23—Clustering techniques
- G06F18/232—Non-hierarchical techniques
- G06F18/2321—Non-hierarchical techniques using statistics or function optimisation, e.g. modelling of probability density functions
-
- 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/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- 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
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Data Mining & Analysis (AREA)
- Physics & Mathematics (AREA)
- General Physics & Mathematics (AREA)
- General Engineering & Computer Science (AREA)
- Life Sciences & Earth Sciences (AREA)
- Artificial Intelligence (AREA)
- Evolutionary Computation (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Evolutionary Biology (AREA)
- Bioinformatics & Computational Biology (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Molecular Biology (AREA)
- Biomedical Technology (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- General Health & Medical Sciences (AREA)
- Health & Medical Sciences (AREA)
- Computing Systems (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Library & Information Science (AREA)
- Databases & Information Systems (AREA)
- Probability & Statistics with Applications (AREA)
- Information Retrieval, Db Structures And Fs Structures Therefor (AREA)
Abstract
本发明属于深度学习中的图像检索模型技术领域,具体涉及一种基于图像检索模型的对抗样本生成方法。本发明首先基于深度图像特征构建图像检索模型,然后通过破坏序列近邻关系生成对抗扰动,降低图像检索性能,最后以集成的方式生成通用对抗扰动,提高对抗样本的迁移性。本发明通过使用图像检索模型提取到的特征向量,将图像特征之间的余弦相似度作为衡量匹配程度的得分,按降序排列显示检索到的参考图像,实现图像检索功能。本发明通过集成多个模型生成通用对抗扰动,生成具有更高迁移性的对抗样本。本发明提出的对抗样本生成方法能够生成使图像检索性能明显下降的对抗样本,并且提高了对抗样本迁移性。
Description
技术领域
本发明属于深度学习中的图像检索模型技术领域,具体涉及一种基于图像检索模型的对抗样本生成方法。
背景技术
随着深度学习在现实场景下的实际部署,人们开始关注人工智能系统的安全性问题。有研究发现,深度神经网络容易受到微小扰动的干扰,即在纯净样本上添加不易被人眼察觉的扰动后,模型会以高置信度给出错误的判断,从而对于受到干扰的样本产生错误的输出,这种样本称为对抗样本。
当前关于对抗样本的研究,大部分工作仍集中于图像分类的攻击,对于图像检索的对抗样本生成算法研究相对有限,并且由于图像分类与图像检索任务在数据标注格式、目标、输入尺度、输出和优化方法等方面的不同,适用于图像分类的对抗样本生成方法无法直接应用于图像检索。
当今现有的基于图像检索的对抗样本生成方法主要是通用对抗扰动方法。通用对抗扰动可以直接迁移至其他模型上,且生成对抗样本也能取得一定的攻击效果,但迁移攻击的效果仍不够好,针对对抗样本迁移性的提升仍是需要进一步研究的工作。而本发明提出的基于集成的方法可以在使图像检索性能明显下降的基础上提高通用对抗样本的迁移性。
发明内容
本发明的目的在于提供一种基于图像检索模型的对抗样本生成方法。
本发明的目的通过如下技术方案来实现:包括以下步骤:
步骤1:使用retrieval-SfM-120k作为训练数据集,将retrieval-SfM数据集的图像组织成元组形式(q,m(q),N(q))输入网络;其中,q表示查询样本;m(q)表示一个匹配查询样本的正样本;N(q)表示一组不匹配查询样本的负样本;由这些元组构成训练网络的图像对,共有|N(q)|+1对图像对;根据数据集的ground truth文件,已经标定了查询样本的集合,以及每个查询样本对应的正样本,通过读取ground truth文件即可确定每个元组的q与m(q);负样本需要在查询样本所属簇之外的其他簇中单独计算选取,选取与查询样本具有最高相似度的非配对图像;
步骤2:使用构建的6个图像检索模型AlexNet-MAC、AlexNet-GeM、VGG16-MAC、VGG16-GeM、ResNet50-MAC、ResNet50-GeM在retrieval-SfM-120k数据集进行训练;每个训练元组、由1个查询样本、1个正样本和5个负样本组成,构成6个图像对;样本输入进神经网络前,对尺寸进行缩放,使尺寸小于362×362;设置对比损失函数为:
对于AlexNet,τ=0.7;对于VGG16,τ=0.75;对于ResNet50,τ=0.85;
步骤3:使用在ImageNet数据集上获得的均值和方差对图像进行归一化处理,均值与标准差都有三个数值,分别对应图像的三个通道;由于预训练的CNN模型已具备优秀的特征提取能力,因此损失函数将会很快收敛,训练周期设置为20,批次大小为5;基于VGG16和ResNet50的图像检索模型采用Adam优化器进行优化,设置初始学习率为l0=10-6,并对学习率进行指数衰减,对第i个epoch衰减参数为exp(-0.1i),参数衰减设置为5×10-4;基于AlexNet的图像检索模型采用SGD优化器进行优化,初始学习率设置为l0=10-3,动量为0.9,其他参数与VGG16和ResNet50保持一致;
步骤4:使用retrieval-SfM-30k数据集的验证集在图像检索模型上生成通用对抗扰动;采用特征聚类预处理方法对retrieval-SfM-120k的验证集中图像进行聚类,保存512个聚类中心,在每个训练周期与查询图像计算相似度来选取最近与最远图像,构成破坏图像特征之间的成对关系;将检索模型设置为评估模式,使用检索模型进行测试,只起到提取特征向量的作用;训练周期设置为100,并使用早停法在训练过程进行检查;当超过5个训练周期没有取得更好的攻击效果后,训练过程提前结束;批次大小设置为1,即每次传入1张查询图像提取特征向量,与512个聚类中心进行余弦相似度的计算;优化器采用带有动量的SGD,由公式
进行优化,动量参数momentum设置为0.9,限制扰动大小的参数ε设置为10;由于读取图像数据后进行了归一化,表示图像的张量的每一个元素[0,1]范围;在优化扰动时,令ε=ε/255,在归一化条件下限制扰动的大小;初始学习率与该参数有关,设置为ε/10,并采取与训练图像检索模型相同的指数衰减方案调整学习率。
本发明的有益效果在于:
本发明首先基于深度图像特征构建图像检索模型,然后通过破坏序列近邻关系生成对抗扰动,降低图像检索性能,最后以集成的方式生成通用对抗扰动,提高对抗样本的迁移性。本发明通过使用图像检索模型提取到的特征向量,将图像特征之间的余弦相似度作为衡量匹配程度的得分,按降序排列显示检索到的参考图像,实现图像检索功能。在生成通用对抗扰动时,本发明采用基于优化的方法去生成一个广义的扰动噪声,该扰动噪声对任何输入图像都具有通用性。本发明通过集成多个模型生成通用对抗扰动,生成具有更高迁移性的对抗样本。本发明提出的对抗样本生成方法能够生成使图像检索性能明显下降的对抗样本,并且提高了对抗样本迁移性。
附图说明
图1是基于深度图像特征的图像检索模型示意图。
图2是基于图像检索的对抗样本生成扰动示意图。
图3是针对图像检索的通用对抗扰动生成过程mDR曲线图。
图4是单一模型与集成方法生成通用对抗扰动对比图。
图5是对抗样本攻击前后的部分检索结果可视化展示图。
图6是对比损失函数计算示意图。
具体实施方式
下面结合附图对本发明做进一步描述。
本发明的目的是基于深度学习技术生成针对图像检索任务的对抗样本算法,并提高了通用扰动的迁移性。本发明通过集成多个模型生成通用对抗扰动,生成具有更高迁移性的对抗样本。经实验验证,本发明提出的对抗样本生成方法能够生成使图像检索性能明显下降的对抗样本,并且提高了对抗样本迁移性。
本发明首先基于深度图像特征构建图像检索模型,然后通过破坏序列近邻关系生成对抗扰动,降低图像检索性能,最后以集成的方式生成通用对抗扰动,提高对抗样本的迁移性。
在构建图像检索模型时,本发明基于预训练的卷积神经网络进行微调,将其修改为全卷积网络结构,结合特殊的特征池化层构建图像特征描述子,构成图像检索模型,得到表示能力优秀的紧凑图像特征表示,如图1所示。本发明通过使用图像检索模型提取到的特征向量,将图像特征之间的余弦相似度作为衡量匹配程度的得分,按降序排列显示检索到的参考图像,实现图像检索功能。
在生成通用对抗扰动时,本发明采用基于优化的方法去生成一个广义的扰动噪声,生成扰动的示意图如图2所示。该扰动噪声对任何输入图像都具有通用性,即添加至任何纯净样本生成的对抗样本都能够有效干扰图像检索的性能。本发明通过使用对抗样本的特征向量进行检索,以Learning-to-Rank方法中的成对近邻关系破坏特征向量的相似性,优化扰动使图像检索模型无法有效提取对图像特征,导致检索性能下降。
在提升对抗样本迁移性时,本发明采用基于集成的方法对对抗样本生成方法做进一步改进,集成若干个模型生成一个通用扰动,攻击留出的黑盒模型,提升扰动对其他模型的迁移能力。
步骤1.设置训练数据集。使用retrieval-SfM-120k作为训练数据集,将retrieval-SfM数据集的图像组织成元组形式(q,m(q),N(q))输入网络,其中q表示查询样本,m(q)表示一个匹配查询样本的正样本,N(q)表示一组不匹配查询样本的负样本,由这些元组构成训练网络的图像对,共有|N(q)|+1对图像对。根据数据集的ground truth文件,已经标定了查询样本的集合,以及每个查询样本对应的正样本,通过读取ground truth文件即可确定每个元组的q与m(q)。负样本需要在查询样本所属簇之外的其他簇中单独计算选取。选取过程遵循难样本(hard example)原则,选取与查询样本具有最高相似度的非配对图像。
步骤2.使用构建的6个图像检索模型AlexNet-MAC、AlexNet-GeM、VGG16-MAC、VGG16-GeM、ResNet50-MAC、ResNet50-GeM在retrieval-SfM-120k数据集进行训练。具体设置为:每个训练元组中由1个查询样本、1个正样本和5个负样本组成,构成6个图像对。样本输入进神经网络前,对尺寸进行缩放,使尺寸小于362×362。对于AlexNet,设置对比损失函数
的参数τ=0.7,对于VGG16,τ=0.75,对于ResNet50,τ=0.85。对比损失函数的计算过程如图6所示。
步骤3.使用在ImageNet数据集上获得的均值和方差对图像进行归一化处理。均值mean=[0.485,0.456,0.406],标准差std=[0.229,0.224,0.225],均值与标准差都有三个数值,分别对应图像的三个通道。而且由于预训练的CNN模型已具备优秀的特征提取能力,因此损失函数将会很快收敛,训练周期(epoch)设置为20,批次大小(batch size)为5,基于VGG16和ResNet50的图像检索模型采用Adam优化器进行优化,设置初始学习率为l0=10-6,并对学习率进行指数衰减,对第i个epoch衰减参数为exp(-0.1i),参数衰减设置为5×10-4;基于AlexNet的图像检索模型采用SGD优化器进行优化,初始学习率设置为l0=10-3,动量为0.9,其他参数与VGG16和ResNet50保持一致。
步骤4.使用retrieval-SfM-30k数据集的验证集在图像检索模型上生成通用对抗扰动。共有6403张图像,其中有1691张查询图像。首先采用特征聚类预处理方法对retrieval-SfM-120k的验证集6403张图像进行聚类,保存512个聚类中心,在每个训练周期与查询图像计算相似度来选取最近与最远图像,构成破坏图像特征之间的成对关系。
检索模型设置为评估(eval)模式,相当于使用检索模型进行测试,只起到提取特征向量的作用。训练周期设置为100,并使用早停法(early stopping)在训练过程进行检查,当超过5个训练周期没有取得更好的攻击效果后,训练过程提前结束;批次大小设置为1,即每次传入1张查询图像提取特征向量,与512个聚类中心进行余弦相似度的计算;优化器采用带有动量的SGD,由公式
进行优化,动量参数momentum设置为0.9,限制扰动大小的参数ε设置为10。由于读取图像数据后进行了归一化,表示图像的张量的每一个元素[0,1]范围,因此,在优化扰动时,令ε=ε/255,在归一化条件下限制扰动的大小;初始学习率与该参数有关,设置为ε/10,约等于0.00392157,并采取与训练图像检索模型相同的指数衰减方案调整学习率。
与现有技术相比,本发明的有益效果是:
1.本发明提出的对抗样本方法是基于图像检索任务的。
当前关于对抗样本的研究,大部分工作仍集中于图像分类的攻击,对于图像检索的对抗样本生成算法研究相对有限,并且由于图像分类与图像检索任务在数据标注格式、目标、输入尺度、输出和优化方法等方面的不同,适用于图像分类的对抗样本生成方法无法直接应用于图像检索。本发明首先构造了高精度的图像检索模型,然后通过添加扰动降低图像检索的性能。
本发明构建的图像检索模型在Oxford5k和ROxford5k数据集的测试结果如表1所示,在Paris6k和RParis6k数据集的测试结果如表2所示,其中A-MAC、A-GeM、V-MAC、V-GeM、R-MAC、R-GeM分别为6个图像检索模型的缩写,E、M、H为ROxford5k和RParis6k数据集上的三种难度的评估模式。
表1图像检索模型Oxford5k/ROxford5k检索性能
表2图像检索模型Paris6k/RParis6k检索性能
测试结果直观地反映了图像检索模型的特征提取能力,通过余弦相似度进行检索,6个模型在4个数据集上都取得了优秀的检索性能,包括mAP和mP@10。其中,基于AlexNet的图像检索模型性能相比基于VGG16和ResNet的图像检索模型较低,这是因为其所采用的AlexNet前置网络结构相对后者较为简单,卷积神经网络结构提取图像特征的能力相对较弱导致的;在E(Easy)条件下,6个检索模型在ROxford5k和RParis6k数据集上都取得了较高的mAP,随难度增大,mAP有所下降,属于正常现象。
2.本发明提出的对抗样本生成算法明显的降低了图像检索模型的性能。
对于图像检索问题,需要使用检索序列的mAP来衡量检索性能受到的影响。因此,针对图像检索mAP以及mP@10指标的变化,使用平均下降率(mean Dropping Rate,mDR)作为评估攻击效果的指标。该指标通过计算检索性能的降低比例来衡量攻击的性能,数值越大,攻击的效果越好。
通过破坏成对近邻关系生成针对图像检索的通用对抗扰动,生成扰动过程中的测试mDR变化曲线如图3所示。可以观察到,随着迭代次数增加,扰动的攻击效果也逐渐提高。由于采用了早停法,当超过5个训练周期没有取得更高mDR时,训练过程提前终止,选择当前最佳扰动噪声作为生成的结果,由图3可以发现,除了ResNet50-GeM外,在其他模型上生成通用对抗扰动时都触发了提前终止的情况。6个模型的最佳扰动噪声点在图像中用“bestnoise”标出,其中ResNet50-GeM模型完成了100次迭代。
除了通过破坏成对近邻关系生成针对图像检索的通用对抗扰动外,本发明还使用了另外两种噪声进行实验,第一种噪声为简单的高斯噪声(Gaussian noise),第二种噪声是在以聚类中心ID作为伪标签(pseudo-label)构建的分类器上生成的。
表3和表4展示了正常检索和使用高斯噪声、分类攻击生成的扰动、破坏近邻关系的算法1生成的扰动进行攻击后的检索结果。表中O代表正常检索的初始结果(originalresults),G代表使用高斯噪声进行攻击的检索性能,C(classifier)代表使用分类攻击方法生成的扰动攻击后的检索性能,P代表使用破坏成对(pair-wise)关系生成扰动进行攻击的检索性能。
表3 Oxford5k/ROxford5k对抗样本攻击效果
表4 Paris6k/RParis6k对抗样本攻击效果
计算使用随机噪声、分类攻击扰动和破坏成对近邻关系的扰动攻击的mDR,结果如表5所示。
表5对抗样本攻击mDR
对图像添加的高斯噪声的大小被限制在与通用对抗扰动相同的[-ε,ε]区间。显然,对6个提取不同深度图像特征的图像检索模型,高斯噪声并没有影响到图像检索模型的性能,模型的mAP下降极其微小,甚至部分结果出现了微小的上升。使用分类攻击方法生成的扰动,能够对图像检索模型产生较弱的攻击效果,取得的mDR相对较低,仅在VGG16-MAC模型上取得了最高的mDR,在其他模型上的攻击效果远低于针对检索生成的扰动。针对图像检索破坏成对近邻关系生成的扰动使模型检索性能明显下降,mDR都达到了45%以上,特别是在ResNet50-GeM模型上mDR高达66.67%,这表明图像检索系统提取图像特征的能力受到了严重的干扰。以上数据表明,破坏成对近邻关系的通用对抗扰动生成算法能对图像检索产生强大的攻击。
3.本发明使用集成方法提高了对抗样本的迁移性。
①直接迁移
通过破坏成对近邻关系,在6个模型上生成扰动的直接迁移攻击效果如表6所示,评估指标为mDR。表中每一行数据为在该行首模型上生成的通用对抗扰动迁移至每一列模型进行对抗攻击取得的mDR。
表6直接迁移攻击效果
②集成方法迁移
在使用集成方法生成的扰动执行迁移攻击时,由于相同前置网络的图像检索模型提取图像特征的能力相近,迁移攻击时效果较好,因此每次选择1类前置网络结构的模型作为留出模型(如AlexNet-MAC和AlexNet-GeM),即黑盒迁移攻击的目标,集成另外4个具有不同前置网络结构的模型生成通用对抗扰动,结果如表7所示。
表7集成方法迁移攻击效果
表中“-”表示集成时留出使用该前置网络结构的模型,使用另外4个模型进行集成,比如“-AlexNet”表示使用VGG16-MAC、VGG16-GeM、ResNet50-MAC、ResNet50-GeM进行集成,AlexNet-MAC和AlexNet-GeM为迁移攻击的目标。
留出AlexNet进行集成时,扰动对另外4个模型的攻击mDR相比使用其中单一模型生成扰动的攻击效果降低,而迁移至AlexNet检索模型上进行攻击时,与表6中不同前置网络的模型迁移效果比较,迁移至AlexNet-MAC取得了最高的攻击mDR,为27.16%,而迁移至AlexNet-GeM时仅高于使用VGG16-MAC和ResNet50-MAC进行迁移攻击的效果;留出VGG16进行集成生成的扰动,在VGG16-MAC的迁移效果高于所有单一模型直接迁移的效果,在VGG16-GeM上仅低于ResNet50-GeM直接迁移的效果;留出ResNet50进行集成的扰动,迁移至ResNet50的两个检索模型上均取得最高mDR,但提升有限。
总体来看,集成方法生成的扰动在相比单一模型的白盒攻击性能有所降低的情况下,仍能取得黑盒迁移攻击性能的提升,证明了集成方法能够有效提高对抗样本的迁移性。而且,表6和表7的攻击结果表明ResNet50-GeM检索模型既具有最优秀的检索性能,在该模型上生成的扰动在多数情况下也能取得最高的迁移性能,而集成方法生成的扰动尽管不能确保取得最高迁移效果,但相比多数单一模型的迁移性能有所提升。
4.攻击效果可视化验证
图4将在使用单一模型和使用集成方法生成的通用对抗扰动可视化。第一、二行图像分别为使用MAC和GeM特征池化方法的模型生成的通用对抗扰动,第三行为使用集成方法生成的扰动。前两行使用单一模型生成的扰动由横向观察,不同前置网络的模型生成的扰动差异明显,由纵向观察,前置网络相同而特征池化不同的模型生成的扰动比较相似。显然,相同前置网络结构的模型非常相似,提取的特征向量差异并不大,因此在相同前置网络的模型上生成的对抗样本具有接近的攻击效果,这与表6呈现的攻击效果一致。
图5展示了AlexNet-GeM、VGG16-GeM、ResNet50-GeM检索模型在ROxford5k和RParis6k数据集上正常检索和使用对抗样本检索的检索结果。图中红色竖线将两个数据集分开,每个数据及均选择一个样本作为样例,两个数据集的第一列均为查询图像。奇数行是使用纯净样本的检索结果,偶数行是使用在相应检索模型上使用对抗样本进行检索的结果。
本发明构建了高精度的图像检索模型,通过生成对抗样本明显降低了图像检索的性能,并以集成的形式提高了对抗样本的迁移性,最后对攻击效果进行了可视化展示。
以上所述仅为本发明的优选实施例而已,并不用于限制本发明,对于本领域的技术人员来说,本发明可以有各种更改和变化。凡在本发明的精神和原则之内,所作的任何修改、等同替换、改进等,均应包含在本发明的保护范围之内。
Claims (1)
1.一种基于图像检索模型的对抗样本生成方法,其特征在于,包括以下步骤:
步骤1:使用retrieval-SfM-120k作为训练数据集,将retrieval-SfM-120k数据集的图像组织成元组形式(q,m(q),N(q))输入网络;其中,q表示查询样本;m(q)表示一个匹配查询样本的正样本;N(q)表示一组不匹配查询样本的负样本;由这些元组构成训练网络的图像对,共有|N(q)|+1对图像对;根据数据集的ground truth文件,已经标定了查询样本的集合,以及每个查询样本对应的正样本,通过读取ground truth文件即可确定每个元组的q与m(q);负样本需要在查询样本所属簇之外的其他簇中单独计算选取,选取与查询样本具有最高相似度的非配对图像;
步骤2:使用构建的6个图像检索模型AlexNet-MAC、AlexNet-GeM、VGG16-MAC、VGG16-GeM、ResNet50-MAC、ResNet50-GeM在retrieval-SfM-120k数据集进行训练;每个训练元组、由1个查询样本、1个正样本和5个负样本组成,构成6个图像对;样本输入进神经网络前,对尺寸进行缩放,使尺寸小于362×362;设置对比损失函数为:
对于AlexNet,τ=0.7;对于VGG16,τ=0.75;对于ResNet50,τ=0.85;
步骤3:使用在ImageNet数据集上获得的均值和方差对图像进行归一化处理,均值与标准差都有三个数值,分别对应图像的三个通道;由于预训练的CNN模型已具备优秀的特征提取能力,因此损失函数将会很快收敛,训练周期设置为20,批次大小为5;基于VGG16和ResNet50的图像检索模型采用Adam优化器进行优化,设置初始学习率为l0=10-6,并对学习率进行指数衰减,对第i个epoch衰减参数为exp(-0.1i),参数衰减设置为5×10-4;基于AlexNet的图像检索模型采用SGD优化器进行优化,初始学习率设置为l0=10-3,动量为0.9,其他参数与VGG16和ResNet50保持一致;
步骤4:使用retrieval-SfM-30k数据集的验证集在图像检索模型上生成通用对抗扰动;采用特征聚类预处理方法对retrieval-SfM-120k的验证集中图像进行聚类,保存512个聚类中心,在每个训练周期与查询图像计算相似度来选取最近与最远图像,构成破坏图像特征之间的成对关系;将检索模型设置为评估模式,使用检索模型进行测试,只起到提取特征向量的作用;训练周期设置为100,并使用早停法在训练过程进行检查;当超过5个训练周期没有取得更好的攻击效果后,训练过程提前结束;批次大小设置为1,即每次传入1张查询图像提取特征向量,与512个聚类中心进行余弦相似度的计算;优化器采用带有动量的SGD,由公式
进行优化,动量参数momentum设置为0.9,限制扰动大小的参数ε设置为10;由于读取图像数据后进行了归一化,表示图像的张量的每一个元素[0,1]范围;在优化扰动时,令ε=ε/255,在归一化条件下限制扰动的大小;初始学习率与该参数有关,设置为ε/10,并采取与训练图像检索模型相同的指数衰减方案调整学习率。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202011098487.3A CN112199543B (zh) | 2020-10-14 | 2020-10-14 | 一种基于图像检索模型的对抗样本生成方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202011098487.3A CN112199543B (zh) | 2020-10-14 | 2020-10-14 | 一种基于图像检索模型的对抗样本生成方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN112199543A CN112199543A (zh) | 2021-01-08 |
CN112199543B true CN112199543B (zh) | 2022-10-28 |
Family
ID=74008668
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202011098487.3A Active CN112199543B (zh) | 2020-10-14 | 2020-10-14 | 一种基于图像检索模型的对抗样本生成方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN112199543B (zh) |
Families Citing this family (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113255526B (zh) * | 2021-05-28 | 2021-09-21 | 华中科技大学 | 基于动量的对人群计数模型的对抗样本生成方法及系统 |
CN114169392A (zh) * | 2021-10-29 | 2022-03-11 | 阿里巴巴(中国)有限公司 | 模型训练方法及装置、任务处理方法、存储介质和处理器 |
CN114444690B (zh) * | 2022-01-27 | 2024-06-07 | 厦门大学 | 一种基于任务扩增的迁移攻击方法 |
CN115098732B (zh) * | 2022-08-11 | 2022-11-11 | 腾讯科技(深圳)有限公司 | 数据处理方法及相关装置 |
Citations (8)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109165309A (zh) * | 2018-08-06 | 2019-01-08 | 北京邮电大学 | 负例训练样本采集方法、装置及模型训练方法、装置 |
CN109960737A (zh) * | 2019-03-15 | 2019-07-02 | 西安电子科技大学 | 半监督深度对抗自编码哈希学习的遥感影像内容检索方法 |
CN110321451A (zh) * | 2019-04-25 | 2019-10-11 | 吉林大学 | 基于分布熵增益损失函数的图像检索算法 |
CN110321957A (zh) * | 2019-07-05 | 2019-10-11 | 重庆大学 | 融合三元组损失和生成对抗网络的多标签图像检索方法 |
CN110647645A (zh) * | 2019-08-06 | 2020-01-03 | 厦门大学 | 一种基于通用扰动的攻击图像检索方法 |
CN111008294A (zh) * | 2018-10-08 | 2020-04-14 | 阿里巴巴集团控股有限公司 | 交通图像处理、图像检索方法及装置 |
CN111310802A (zh) * | 2020-01-20 | 2020-06-19 | 星汉智能科技股份有限公司 | 一种基于生成对抗网络的对抗攻击防御训练方法 |
CN111368123A (zh) * | 2020-02-17 | 2020-07-03 | 同济大学 | 一种基于跨模态向导网络的三维模型草图检索方法 |
Family Cites Families (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US9767381B2 (en) * | 2015-09-22 | 2017-09-19 | Xerox Corporation | Similarity-based detection of prominent objects using deep CNN pooling layers as features |
US10831818B2 (en) * | 2018-10-31 | 2020-11-10 | Adobe Inc. | Digital image search training using aggregated digital images |
-
2020
- 2020-10-14 CN CN202011098487.3A patent/CN112199543B/zh active Active
Patent Citations (8)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109165309A (zh) * | 2018-08-06 | 2019-01-08 | 北京邮电大学 | 负例训练样本采集方法、装置及模型训练方法、装置 |
CN111008294A (zh) * | 2018-10-08 | 2020-04-14 | 阿里巴巴集团控股有限公司 | 交通图像处理、图像检索方法及装置 |
CN109960737A (zh) * | 2019-03-15 | 2019-07-02 | 西安电子科技大学 | 半监督深度对抗自编码哈希学习的遥感影像内容检索方法 |
CN110321451A (zh) * | 2019-04-25 | 2019-10-11 | 吉林大学 | 基于分布熵增益损失函数的图像检索算法 |
CN110321957A (zh) * | 2019-07-05 | 2019-10-11 | 重庆大学 | 融合三元组损失和生成对抗网络的多标签图像检索方法 |
CN110647645A (zh) * | 2019-08-06 | 2020-01-03 | 厦门大学 | 一种基于通用扰动的攻击图像检索方法 |
CN111310802A (zh) * | 2020-01-20 | 2020-06-19 | 星汉智能科技股份有限公司 | 一种基于生成对抗网络的对抗攻击防御训练方法 |
CN111368123A (zh) * | 2020-02-17 | 2020-07-03 | 同济大学 | 一种基于跨模态向导网络的三维模型草图检索方法 |
Non-Patent Citations (4)
Title |
---|
Synthesizing Chest X-Ray Pathology for Training Deep Convolutional Neural Networks;Hojjat Salehinejad等;《IEEE Transactions on Medical Imaging》;20181114;第1197-1206页 * |
Vulnerability Analysis on Noise-Injection Based Hardware Attack on Deep Neural Networks;Wenye Liu等;《2019 Asian Hardware Oriented Security and Trust Symposium (AsianHOST)》;20200224;第1-6页 * |
图像检索中的特征学习和索引技术研究;刘若愚;《中国博士学位论文全文数据库 信息科技辑》;20200115;I138-135 * |
复杂天气条件下水天场景图像增强技术研究;沈亮;《中国优秀硕士学位论文全文数据库 工程科技Ⅱ辑》;20200315;C036-145 * |
Also Published As
Publication number | Publication date |
---|---|
CN112199543A (zh) | 2021-01-08 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN112199543B (zh) | 一种基于图像检索模型的对抗样本生成方法 | |
Deng et al. | Marginal loss for deep face recognition | |
Hermans et al. | In defense of the triplet loss for person re-identification | |
CN110008842A (zh) | 一种基于深度多损失融合模型的行人重识别方法 | |
CN109800317B (zh) | 一种基于图像场景图谱对齐的图像查询回答方法 | |
CN109255340A (zh) | 一种融合多种改进vgg网络的人脸识别方法 | |
CN108921019A (zh) | 一种基于GEI和TripletLoss-DenseNet的步态识别方法 | |
CN110990580A (zh) | 知识图谱的构建方法、构建装置、计算机设备及存储介质 | |
CN114469120B (zh) | 一种基于相似度阈值迁移的多尺度Dtw-BiLstm-Gan心电信号生成方法 | |
CN116055175A (zh) | 联合对称不确定性和超参数优化神经网络的入侵检测方法 | |
CN113704758A (zh) | 一种黑盒攻击对抗样本生成方法及系统 | |
CN114880484A (zh) | 一种基于向量映射的卫星通信频轨资源图谱构建方法 | |
CN113297574B (zh) | 基于强化学习奖励机制的激活函数自适应变化模型窃取防御方法 | |
CN113987203A (zh) | 一种基于仿射变换与偏置建模的知识图谱推理方法与系统 | |
CN117236330B (zh) | 一种基于互信息和对抗神经网络的增强主题多样性方法 | |
CN116303976B (zh) | 基于网络安全知识图谱的渗透测试问答方法、系统及介质 | |
CN117727464A (zh) | 基于医疗多视图疾病预测模型的训练方法及设备 | |
CN113947706B (zh) | 一种基于生成对抗网络的图像多样性增强方法与系统 | |
Xiong et al. | Image Aesthetics Assessment Via Learnable Queries | |
CN113656833A (zh) | 基于进化计算的在垂直联邦架构下的隐私窃取防御方法 | |
CN114417713A (zh) | 一种来袭多目标的编群方法 | |
CN111062484B (zh) | 基于多任务学习的数据集选取方法及装置 | |
CN118193853B (zh) | 一种随机群组的兴趣点推荐方法、系统、设备和存储介质 | |
Du et al. | An effective approach for sonar image recognition with improved efficientdet and ensemble learning | |
Zhang et al. | Pyramid Pixel Context Adaption Network for Medical Image Classification With Supervised Contrastive Learning |
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 | ||
GR01 | Patent grant | ||
GR01 | Patent grant |