###
计算机系统应用英文版:,():1-16
本文二维码信息
码上扫一扫!
自适应层次化主题注意力增强的ELECTRA神经主题模型
(1.青岛科技大学 信息科学技术学院, 青岛 266061;2.中国联合网络通信有限公司青岛市分公司, 青岛 266071)
ELECTRA Neural Topic Model Enhanced by Adaptive Hierarchical Topic Attention
(1.College of Information Science and Technology, Qingdao University of Science & Technology, Qingdao 266061, China;2.Qingdao Branch of China United Network Communications Co. Ltd., Qingdao 266071, China)
摘要
图/表
参考文献
相似文献
本文已被:浏览 12次   下载 40
Received:February 03, 2026    Revised:February 27, 2026
中文摘要: 文本分类任务中, 预训练语言模型在语义表示方面取得了显著进展, 但在复杂文档建模时仍存在主题粒度单一、主题与上下文交互不足等问题. 针对上述不足, 本文提出了一种自适应层次化主题注意力增强的 ELECTRA神经主题模型HTA-ELECTRA. 该模型通过层次化变分主题建模同时学习全局与局部主题表征, 并引入自适应阈值机制动态筛选与当前文档相关的主题信息; 在此基础上, 设计多级主题融合与主题感知注意力机制, 将主题结构信息逐层注入ELECTRA上下文表示, 从而增强文档语义建模能力. 实验结果表明, 在20Newsgroups、MRD和Yelp这3个数据集上, 所提模型在AUCMacro-F1、Macro-RecallMacro-PrecisionAccuracy等指标上均优于多种对比模型, 验证了该方法的有效性与鲁棒性.
Abstract:In text classification tasks, pre-trained language models have made remarkable progress in semantic representation. However, when modeling complex documents, they still suffer from limitations such as a single topic granularity and insufficient interaction between topics and context. To this end, this study proposes an adaptive hierarchical topic-aware ELECTRA neural topic model termed as HTA-ELECTRA. The model employs hierarchical variational topic modeling to jointly learn global and local topic representations, and introduces an adaptive threshold mechanism to dynamically select topics that are relevant to the current document. On this basis, a multi-level topic fusion strategy and a topic-aware attention mechanism are designed to progressively inject topic structure information into ELECTRA-based contextual representations, thereby enhancing document-level semantic modeling ability. Experimental results show that on the 20Newsgroups, MRD, and Yelp datasets, the proposed model outperforms multiple comparison models in terms of AUC, Macro-F1, Macro-Recall, Macro-Precision, and Accuracy, validating the effectiveness and robustness of the proposed method.
文章编号:     中图分类号:    文献标志码:
基金项目:青岛市科技惠民示范项目(23-2-8-smjk-20-nsh)
引用文本:
关权,秦玉华,张磊,任盈盈,于静.自适应层次化主题注意力增强的ELECTRA神经主题模型.计算机系统应用,,():1-16
GUAN Quan,QIN Yu-Hua,ZHANG Lei,REN Ying-Ying,YU Jing.ELECTRA Neural Topic Model Enhanced by Adaptive Hierarchical Topic Attention.COMPUTER SYSTEMS APPLICATIONS,,():1-16