改进的BERT-UNet模型的新闻文本分类系统

我要开发同款
LiZeqin2024年07月31日
65阅读
开发技术torch、自然语言处理、node.js、python
所属分类人工智能

作品详情

新闻文本分类任务在信息检索、舆情检测与分析、信息智能推送等领域发挥着重要的作用。为了解决传统卷积神经网络在新闻文本分类中效果不佳的问题,本文提出了一种改进的BERT-UNet文本分类模型,来增强捕捉长距离文本特征和可视化效果。首先该模型使用BERT预训练文本词向量,然后将其嵌入映射到UNet模型中,提取上下文的关键特征,同时通过Softmax函数实现文本分类,最后利用前端技术对新闻舆情信息文本分类结果进行可视化监控展示。为验证模型在文本分类任务上的优越性,本文进在THUCNews数据集上进行了对比实验,实验结果显示,相较于传统模的TextCNN模型和独立使用BERT的方法,BERT-UNet模型在宏平均F1值上分别提高了3.11%和0.29%。这表明改进的BERT-UNet模型在捕捉文本特征关系方面更有效,在提升分类性能,改善传统新闻文本分类方法及可视化监控提供了新思路。
声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论