journal article·2024-09-18Open access

Improving the Accuracy and Effectiveness of Text Classification Based on the Integration of the Bert Model and a Recurrent Neural Network (RNN_Bert_Based)

Chanthol Eang, Seungjae Lee

Applied Sciences· Volume 14· Issue 18

Abstract

This paper proposes a new robust model for text classification on the Stanford Sentiment Treebank v2 (SST-2) dataset in terms of model accuracy. We developed a Recurrent Neural Network Bert based (RNN_Bert_based) model designed to improve classification accuracy on the SST-2 dataset. This dataset consists of movie review sentences, each labeled with either positive or negative sentiment, making it a binary classification task. Recurrent Neural Networks (RNNs) are effective for text classification because they capture the sequential nature of language, which is crucial for understanding context and meaning. Bert excels in text classification by providing bidirectional context, generating contextual embeddings, and leveraging pre-training on large corpora. This allows Bert to capture nuanced meanings and relationships within the text effectively. Combining Bert with RNNs can be highly effective for text classification. Bert’s bidirectional context and rich embeddings provide a deep understanding of the text, while RNNs capture sequential patterns and long-range dependencies. Together, they leverage the strengths of both architectures, leading to improved performance on complex classification tasks. Next, we also developed an integration of the Bert model and a K-Nearest Neighbor based (KNN_Bert_based) method as a comparative scheme for our proposed work. Based on the results of experimentation, our proposed model outperforms traditional text classification models as well as existing models in terms of accuracy.
Source: crossref · Quality: baseline

Research context

OpenAlex topics: Topic Modeling · Text and Document Classification Technologies · Sentiment Analysis and Opinion Mining

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

Affiliations

Department of Computer Science and Engineering, Intelligent Robot Research Institute, Sun Moon University, Asan 31460, Republic of Korea

References (39)

Show source-provided references
1. Munikar, M., Shakya, S., and Shrestha, A. (2019). Fine-grained Sentiment Classification using BERT. arXiv.10.1109/AITB48515.2019.8947435 ↗
2. Text Mining: Classification of Text Documents Using Granular Hybrid Classification Technique10.32622/ijrat.76201910 ↗
3. Semary, N.A., Ahmed, W., Amin, K., Pławiak, P., and Hammad, M. (2024). Enhancing Machine Learning-Based Sentiment Analysis through Feature Extraction Techniques. PLoS ONE, 19.10.1371/journal.pone.0294968 ↗
4. Framework for Thought to Text Classification
5. (2018). Challenges in text classification using machine learning techniques. Int. J. Recent Trends Eng. Res., 4, 81–83.10.23883/IJRTER.2018.4068.K3ORB ↗
6. (2017). News Text Classification Model Based on Topic Model. Int. J. Recent Trends Eng. Res., 3, 48–52.10.23883/IJRTER.2017.3330.A9X5G ↗
7. Combining Sentiment-Combined Model with Pre-Trained BERT Models for Sentiment Analysis10.5626/JOK.2021.48.7.815 ↗
8. Zhang, T., Kishore, V., Wu, F., Weinberger, K.Q., and Artzi, Y. (2020). BERTScore: Evaluating Text Generation with BERT. arXiv.
9. Sellam, T., Das, D., and Parikh, A. (2020, January 9). BLEURT: Learning Robust Metrics for Text Generation. Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics, Online.10.18653/v1/2020.acl-main.704 ↗
10. Liu, S., Tao, H., and Feng, S. (2019, January 22–24). Text Classification Research Based on Bert Model and Bayesian network. Proceedings of the 2019 Chinese Automation Congress (CAC), Hangzhou, China.10.1109/CAC48633.2019.8996183 ↗
11. Joint Representations of Texts and Labels with Compositional Loss for Short Text Classification
12. Chinese Multilabel Short Text Classification Method Based on GAN and Pinyin Embedding10.1109/ACCESS.2024.3412649 ↗
13. Tree Framework with BERT Word Embedding for the Recognition of Chinese Implicit Discourse Relations10.1109/ACCESS.2020.3019500 ↗
14. A BERT-Enhanced Exploration of Web and Mobile Request Safety through Advanced NLP Models and Hybrid Architectures10.1109/ACCESS.2024.3406413 ↗
15. Improving BERT-Based Text Classification with Auxiliary Sentence and Domain Knowledge10.1109/ACCESS.2019.2953990 ↗
16. Zhang, H., Shan, Y., Jiang, P., and Cai, X. (2022, January 16–18). A Text Classification Method Based on BERT-Att-TextCNN Model. Proceedings of the 2022 IEEE 5th Advanced Information Management, Communicates, Electronic and Automation Control Conference (IMCEC), Chongqing, China.10.1109/IMCEC55388.2022.10019808 ↗
17. Leveraging Knowledge-Based Features with Multilevel Attention Mechanisms for Short Arabic Text Classification10.1109/ACCESS.2022.3175306 ↗
18. Joint Learning with BERT-GCN and Multi-Attention for Event Text Classification and Event Assignment10.1109/ACCESS.2022.3156918 ↗
19. Electric Power Audit Text Classification with Multi-Grained Pre-Trained Language Model10.1109/ACCESS.2023.3240162 ↗
20. Sentiment Analysis Classification System Using Hybrid BERT Models10.1186/s40537-023-00781-w ↗
21. Garrido-Muñoz, I., Montejo-Ráez, A., Martínez-Santiago, F., and Ureña-López, L.A. (2021). A Survey on Bias in Deep NLP. Appl. Sci., 11.10.20944/preprints202103.0049.v1 ↗
22. Wu, Y., Jin, Z., Shi, C., Liang, P., and Zhan, T. (2024). Research on the Application of Deep Learning-based BERT Model in Sentiment Analysis. arXiv.10.54254/2755-2721/71/2024MA ↗
23. Li, Q., Li, X., Du, Y., Fan, Y., and Chen, X. (2022). A New Sentiment-Enhanced Word Embedding Method for Sentiment Analysis. Appl. Sci., 12.10.3390/app122010236 ↗
24. A Long-Text Classification Method of Chinese News Based on BERT and CNN10.1109/ACCESS.2022.3162614 ↗
25. Solving Data Imbalance in Text Classification with Constructing Contrastive Samples10.1109/ACCESS.2023.3306805 ↗
26. Text Sentiment Analysis of Douban Film Short Comments Based on BERT-CNN-BiLSTM-Att Model10.1109/ACCESS.2024.3381515 ↗
27. Fine-Tuning BERT for Multi-Label Sentiment Analysis in Unbalanced Code-Switching Text10.1109/ACCESS.2020.3030468 ↗
28. Sentiment Classification for Chinese Text Based on Interactive Multitask Learning10.1109/ACCESS.2020.3007889 ↗
29. Learning Label-Adaptive Representation for Large-Scale Multi-Label Text Classification10.1109/TASLP.2024.3393722 ↗
30. Enhancing Machine-Generated Text Detection: Adversarial Fine-Tuning of Pre-Trained Language Models10.1109/ACCESS.2024.3396820 ↗
31. Three-Branch BERT-Based Text Classification Network for Gastroscopy Diagnosis Text10.26599/IJCS.2023.9100031 ↗
32. Fine-Tuning Transformer Models Using Transfer Learning for Multilingual Threatening Text Identification10.1109/ACCESS.2023.3320062 ↗
33. Bangla-BERT: Transformer-Based Efficient Model for Transfer Learning and Language Understanding10.1109/ACCESS.2022.3197662 ↗
34. An Automatic Sentiment Analysis Method for Short Texts Based on Transformer-BERT Hybrid Model10.1109/ACCESS.2024.3422268 ↗
35. Language Models for Hierarchical Classification of Radiology Reports with Attention Mechanisms, BERT, and GPT-410.1109/ACCESS.2024.3402066 ↗
36. Corizzo, R., and Leal-Arenas, S. (2023). One-Class Learning for AI-Generated Essay Detection. Appl. Sci., 13.10.3390/app13137901 ↗
37. Learning Deep Features for One-Class Classification10.1109/TIP.2019.2917862 ↗
38. He, K., Zhang, X., Ren, S., and Sun, J. (2016, January 27–30). Deep residual learning for image recognition. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA.10.1109/CVPR.2016.90 ↗
39. Socher, R., Bauer, J., Manning, C.D., and Ng, A.Y. (2013, January 4–9). Parsing with compositional vector grammars. Proceedings of the 51st Annual Meeting of the Association for Computational Linguistics, Sofia, Bulgaria. Long Papers.