- 专利标题: 一种基于对抗训练和对抗学习网络的半监督文本分类方法
-
申请号: CN201910395284.1申请日: 2019-05-13
-
公开(公告)号: CN110532377B公开(公告)日: 2021-09-14
- 发明人: 吴骏 , 闫梦奎 , 郜永祯 , 张洛一 , 谢俊元
- 申请人: 南京大学
- 申请人地址: 江苏省南京市鼓楼区汉口路22号
- 专利权人: 南京大学
- 当前专利权人: 南京大学
- 当前专利权人地址: 江苏省南京市鼓楼区汉口路22号
- 代理机构: 南京瑞弘专利商标事务所
- 代理商 彭雄
- 主分类号: G06F16/35
- IPC分类号: G06F16/35 ; G06K9/62 ; G06K9/66
摘要:
本发明公开了一种基于对抗训练和对抗学习网络的半监督文本分类方法,在初始化网络阶段,首先初始化预测网络,其次初始化评判网络;对抗学习训练阶段,通过预测网络和评判网络的对抗作用,利用policy gradient策略对预测网络进行不断的优化,直至达到最优的效果;模型预测阶段,使用已经训练好的预测网络对新样本的标签进行预测。该方法利用对抗训练可以有效提高模型的正则化能力,增强模型的鲁棒性,同时借鉴对抗学习可以直接利用未标记样本进行模型的训练,通过对抗学习网络中子网络间的对抗作用不断迭代,能有效提高网络的分类性能。
公开/授权文献
- CN110532377A 一种基于对抗训练和对抗学习网络的半监督文本分类方法 公开/授权日:2019-12-03