基于BERT模型的文本对抗样本生成方法  

Text adversarial example generation method based on BERT model

作　　者：李宇航 杨玉丽[1] 马垚 于丹[1] 陈永乐[1] LI Yuhang;YANG Yuli;MA Yao;YU Dan;CHEN Yongle(College of Computer Science and Technology(College of Data Science),Taiyuan University of Technology,Taiyuan Shanxi 030600,China)

机构地区：[1]太原理工大学计算机科学与技术学院(大数据学院),太原030600

出　　处：《计算机应用》2023年第10期3093-3098,共6页journal of Computer Applications

基　　金：山西省基础研究计划项目(20210302123131,20210302124395)。

摘　　要：针对现有对抗样本生成方法需要大量访问目标模型,导致攻击效果较差的问题,提出了基于BERT(Bidirectional Encoder Representations from Transformers)模型的文本对抗样本生成方法(TAEGM)。首先采用注意力机制,在不访问目标模型的情况下,定位显著影响分类结果的关键单词;其次通过BERT模型对关键单词进行单词级扰动,从而生成候选样本;最后对候选样本进行聚类,并从对分类结果影响更大的簇中选择对抗样本。在Yelp Reviews、AG News和IMDB Review数据集上的实验结果表明,相较于攻击成功率(SR)次优的对抗样本生成方法CLARE(ContextuaLized AdversaRial Example generation model),TAEGM在保证对抗攻击SR的前提下,对目标模型的访问次数(QC)平均减少了62.3%,时间平均减少了68.6%。在此基础之上,进一步的实验结果验证了TAEGM生成的对抗样本不仅具有很好的迁移性,还可以通过对抗训练提升模型的鲁棒性。Aiming at the problem that the existing adversarial example generation methods require a lot of queries to the target model,which leads to poor attack effects,a Text Adversarial Examples Generation Method based on BERT(Bidirectional Encoder Representations from Transformers)model(TAEGM)was proposed.Firstly,the attention mechanism was adopted to locate the keywords that significantly influence the classification results without query of the target model.Secondly,word-level perturbation of keywords was performed by BERT model to generate candidate adversarial examples.Finally,the candidate examples were clustered,and the adversarial examples were selected from the clusters that have more influence on the classification results.Experimental results on Yelp Reviews,AG News,and IMDB Review datasets show that compared to the suboptimal adversarial example generation method CLARE(ContextuaLized AdversaRial Example generation model)on Success Rate(SR),TAEGM can reduce the Query Counts(QC)to the target model by 62.3%and time consumption by 68.6%averagely while ensuring the SR of adversarial attacks.Based on the above,further experimental results verify that the adversarial examples generated by TAEGM not only have good transferability,but also improve the robustness of the model through adversarial training.

关 键 词：对抗样本 注意力机制 BERT 对抗攻击 聚类算法 

分 类 号：TP309[自动化与计算机技术—计算机系统结构]