基于 ERNIE-RCNN 模型的中文短文本分类

时间:2022-06-30 00:27:12
作者:王浩畅,孙铭泽
关键字:中文短文本分类,ERNIE 模型,ERNIE-RCNN 模型,词向量,特征提取,深度学习
DOI:10.3969/j.issn.1673-629X.2022.06.005
查看次数:848

如需要完整文档点击下方 "点击下载文档" 按钮

由于中文短文本存在特征词少、规范性差、数据规模量大等难点,ERNIE 预训练模型占用内存大,进行短文本分类时会造成向量空间稀疏、文本预训练不准确、时间复杂度高等问题。 针对以上短文本分类存在的问题,提出基于 ERNIE-RCNN 模型的中文短文本分类。 模型运用 ERNIE 模型作为词向量,对实体和词语义单元掩码,后连接 Transformer 的编码层,对 ERNIE 层输出的词嵌入向量进行编码,优化模型过拟合问题,增强泛化能力,RCNN 模型对 ERNIE 输入的词向量进行特征提取,卷积层利用大小不同的卷积核提取大小不同的特征值,池化层进行映射处理,最后通过 softmax 进行分类。 将该模型与七种深度学习文本分类模型在中文新闻数据集上进行训练实验,得到了模型在准确率、精准率、召回率、F1 值、迭代次数、运行时间上的对比结果,表明 ERNIE-RCNN 模型能够很好地提取文本中的特征信息,减少了训练时间,有效解决了中文短文本分类的难点,具有很好的分类效果。

如需要完整文档点击下方 "点击下载文档" 按钮

基于 ERNIE-RCNN 模型的中文短文本分类
《基于 ERNIE-RCNN 模型的中文短文本分类》
完整文档 下载到本地,方便收藏和查阅
文件号:059604
基于 ERNIE-RCNN 模型的中文短文本分类
点击下载文档
基于 ERNIE-RCNN 模型的中文短文本分类

点击下载 文件号:059604(点击复制) 公众号(点击复制)

x