一种基于Transformer的中文新闻分类方法

基本信息

申请号 CN202010375567.2 申请日 -
公开(公告)号 CN111666373A 公开(公告)日 2020-09-15
申请公布号 CN111666373A 申请公布日 2020-09-15
分类号 G06F16/33(2019.01)I 分类 计算;推算;计数;
发明人 范梦真;杨芳洲;刘金;罗轶凤;钱卫宁;周傲英 申请(专利权)人 上海瞰点科技有限责任公司
代理机构 上海蓝迪专利商标事务所(普通合伙) 代理人 华东师范大学;上海瞰点科技有限责任公司
地址 200241上海市闵行区东川路500号
法律状态 -

摘要

摘要 本发明公开了一种基于Transformer的中文新闻分类方法,其特点是采用可并行的Transformer‑Encoder来对文本序列进行编码,采用“词‑句‑文章”多层的网络来提取新闻的文本信息,利用attention机制来降维得到文章级别的向量,并且结合NER命名实体识别技术引入实体信息来丰富特征,最后融合两部分特征输入分类器,训练得到分类模型,提高分类模型训练的效率和预测的精度。本发明在某领域的文本分类任务上,相比于现有的分类模型,可并行化的序列编码方式降低了模型训练的时间开销,attention机制和加入实体信息提高了模型对重要信息的识别能力,同时,识别了新闻中的实体信息,减少了机器阅读文字的歧义。