arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.04849cs.CLcs.AI
收藏

Ensemble BERT:一种基于集成学习与BERT架构的学生社交网络文本情感分类模型

Ensemble BERT: A student social network text sentiment classification model based on ensemble learning and BERT architecture

Southwest University(西南大学);Central China Normal University(华中师范大学);Dalian Neusoft University of Information(大连东软信息学院);Zhejiang University of Science and Technology(浙江科技大学);University of Sydney(悉尼大学)

Kai Jiang, Honghao Yang, Yuexian Wang, Qianru Chen, Yiming Luo

AI总结:

本文提出基于BERT与多数投票的Ensemble BERT,用于微博中学生文本情感分类,发现其效果接近三层BERT但训练更耗时,深层BERT更适合兼顾效果与效率。

摘要:

中学生心理健康评估一直是教育领域的关注重点之一。本文介绍了一种基于BERT的新型集成学习网络,采用通过整合多个分类器来提升模型性能的理念。我们训练了一系列基于BERT的学习器,并使用多数投票法将其组合。我们通过中国的微博(Weibo)收集中学生社交网络文本数据,并将该方法应用于中学生社交网络文本情感倾向分类任务。实验结果表明,集成学习网络的性能优于基础模型;由三个单层BERT模型组成的集成学习模型,其性能与一个三层BERT模型几乎相同,但需要多花费11.58%的训练时间。因此,在平衡预测效果与效率方面,应优先选择更深层的BERT网络进行训练。不过,就可解释性而言,网络集成能够提供可接受的解决方案。

英文摘要:

The mental health assessment of middle school students has always been one of the focuses in the field of education. This paper introduces a new ensemble learning network based on BERT, employing the concept of enhancing model performance by integrating multiple classifiers. We trained a range of BERT-based learners, which combined using the majority voting method. We collect social network text data of middle school students through China's Weibo and apply the method to the task of classifying emotional tendencies in middle school students' social network texts. Experimental results suggest that the ensemble learning network has a better performance than the base model and the performance of the ensemble learning model, consisting of three single-layer BERT models, is barely the same as a three-layer BERT model but requires 11.58% more training time. Therefore, in terms of balancing prediction effect and efficiency, the deeper BERT network should be preferred for training. However, for interpretability, network ensembles can provide acceptable solutions.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑