Izvorni znanstveni članak
https://doi.org/10.31534/engmod.2022.2.ri.01d
Multi-model Fusion Attention Network for News Text Classification
Zhengpeng Li
; School of Computer and Software, University of Science and Technology Liaoning, Anshan 114044, CHINA
Jiansheng Wu
orcid.org/0000-0003-1201-5915
; School of Computer and Software, University of Science and Technology Liaoning, Anshan 114044, CHINA
Jiawei Miao
; School of Computer and Software, University of Science and Technology Liaoning, Anshan 114044, CHINA
Xinmiao Yu
; School of Computer and Software, University of Science and Technology Liaoning, Anshan 114044, CHINA
Sažetak
At present, the classification prediction task based on news content or news headline has the problems of inaccurate classification and attention deviation. In this paper, a multi-model fusion attention network for news text classification (MFAN) is proposed to train news content and news titles in parallel. Firstly, the multi-head attention mechanism is used to obtain the category information of news content through a dynamic word vector, focusing on the semantic information that significantly influences the downstream classification task. Secondly, the semantic information of news headlines is obtained by using the improved version of the long-short-term memory network, and the attention is focused on the words that have a great influence on the final results, which improves the effectiveness of model classification. Finally, the classification fusion module fuses the probability scores of news text and news headlines in proportion to improve the accuracy of text classification. The experimental test on the Tenth China Software cup dataset shows that the F1 - Score index of the MFAN model reaches 97.789 %. The experimental results show that the MFAN model can effectively and accurately predict the classification of news texts.
Ključne riječi
multi-model fusion; attentional mechanism; text classification; neural network
Hrčak ID:
276036
URI
Datum izdavanja:
5.5.2022.
Posjeta: 1.016 *