节点文献

基于对抗训练的文本表示和分类算法

Text Representation and Classification Algorithm Based on Adversarial Training

  • 推荐 CAJ下载
  • PDF下载
  • 不支持迅雷等下载工具,请取消加速工具后下载。

【作者】 张晓辉于双元王全新徐保民

【Author】 ZHANG Xiao-hui;YU Shuang-yuan;WANG Quan-xin;XU Bao-min;School of Computer and Information Technology,Beijing Jiaotong University;School of Computer and Information Technology,Beijing Jiaotong University Haibin College;

【通讯作者】 王全新;

【机构】 北京交通大学计算机与信息技术学院北京交通大学海滨学院计算机与信息技术学院

【摘要】 文本表示和分类是自然语言理解领域的研究热点。目前已有很多文本分类方法,包括卷积网络、递归网络、自注意力机制以及它们的结合。但是,复杂的网络并不能从根本上提高文本分类的性能,好的文本表示才是文本分类的关键。为了获得好的文本表示,提高文本分类性能,构建了基于LSTM的表示学习-文本分类模型,其中表示学习模型利用语言模型为文本分类模型提供初始化的文本表示和网络参数。文中主要采用对抗训练方法训练语言模型,即在词向量上添加扰动构造对抗样本,再利用对抗样本和原始样本一起训练模型,通过提升模型对对抗样本的正确分类能力,提高文本表示的质量,增强模型的泛化性能,从而改善分类模型的分类效果。实验结果表明,基于对抗训练的文本分类方法在基准数据集AGNews,IMDB,DBpedia上分别实现了92.9%,93.2%,98.9%的准确率,证明了该方法能够有效提高文本分类模型的分类性能。

【Abstract】 Text representation and classification are hot topics in the field of natural language understanding.There are many text classification methods,including convolutional networks,recursive networks,self-attention mechanisms and their combinations.complex networks cannot fundamentally improve the performance of classificationtext representation is the key to text classification.In order to obtain a good text representation and improve the performance of text classification,an LSTM-based representation learning-text classification model is constructed,where the representation learning model uses a language model to provide the text classification model with initialized text representation and network parameters.main work is to adversarial training methods that is,add perturbations to word vectors to construct adversarial samples,and train the original samples.By improving the model’s ability adversarial samples,the quality of text representation,and the generalization performance of the model, the classification effect of the classification model.xperimental results show that the method based on adversarial training achieves 92.9%,93.2% and 98.9% on the benchmark datasets AGNews,IMDBDBpedia,that the method can improve the classification effect of the model.

【基金】 河北省高等教育科技攻关重点项目(ZD2017304)~~
  • 【文献出处】 计算机科学 ,Computer Science , 编辑部邮箱 ,2020年S1期
  • 【分类号】TP391.1;TP183
  • 【被引频次】7
  • 【下载频次】342
节点文献中: