Multi - Level Attention and Contrastive Learning for Enhanced Text Classification with an Optimized Transformer

This paper studies a text classification algorithm based on an improved Transformer to improve the performance and efficiency of the model in text classification tasks. Aiming at the shortcomings of the traditional Transformer model in capturing deep semantic relationships and optimizing computational complexity., this paper introduces a multi-level attention mechanism and a contrastive learning strategy. The multi-level attention mechanism effectively models the global semantics and local features in the text by combining global attention with local attention; the contrastive learning strategy enhances the model's ability to distinguish between different categories by constructing positive and negative sample pairs while improving the classification effect. In addition., in order to improve the training and inference efficiency of the model on large-scale text data., this paper designs a lightweight module to optimize the feature transformation process and reduce the computational cost. Experimental results on the dataset show that the improved Transformer model outperforms the comparative models such as BiLSTM., CNN., standard Transformer., and BERT in terms of classification accuracy., Fl score., and recall rate., showing stronger semantic representation ability and generalization performance. The method proposed in this paper provides a new idea for algorithm optimization in the field of text classification and has good application potential and practical value. Future work will focus on studying the performance of this model in multi-category imbalanced datasets and cross-domain tasks and explore the integration with other technologies to further improve its performance.

Authors

Institutions

Publication Details

Published
2025-02-28
DOI
https://doi.org/10.1109/iccece65250.2025.10985142
Citations
7
Primary Topic
Text and Document Classification Technologies
Type
article
Field-Weighted Citation Impact
17.37
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Multi - Level Attention and Contrastive Learning for Enhanced Text Classification with an Optimized Transformer

Isabel Gao, Haibing Zheng, Shicheng Zhou, Binrong Zhu et al.
7 citations
Text and Document Classification Technologies
17.37
article

Multi - Level Attention and Contrastive Learning for Enhanced Text Classification with an Optimized Transformer

Isabel Gao, Haibing Zheng, Shicheng Zhou, Binrong Zhu, Guiran Liu, Xiaoxuan Liao
article en
7 citations

Abstract

This paper studies a text classification algorithm based on an improved Transformer to improve the performance and efficiency of the model in text classification tasks. Aiming at the shortcomings of the traditional Transformer model in capturing deep semantic relationships and optimizing computational complexity., this paper introduces a multi-level attention mechanism and a contrastive learning strategy. The multi-level attention mechanism effectively models the global semantics and local features in the text by combining global attention with local attention; the contrastive learning strategy enhances the model's ability to distinguish between different categories by constructing positive and negative sample pairs while improving the classification effect. In addition., in order to improve the training and inference efficiency of the model on large-scale text data., this paper designs a lightweight module to optimize the feature transformation process and reduce the computational cost. Experimental results on the dataset show that the improved Transformer model outperforms the comparative models such as BiLSTM., CNN., standard Transformer., and BERT in terms of classification accuracy., Fl score., and recall rate., showing stronger semantic representation ability and generalization performance. The method proposed in this paper provides a new idea for algorithm optimization in the field of text classification and has good application potential and practical value. Future work will focus on studying the performance of this model in multi-category imbalanced datasets and cross-domain tasks and explore the integration with other technologies to further improve its performance.

Stevens Institute of Technology (US), University of Minnesota (US), Chinese University of Hong Kong (HK), New York University (US), San Francisco State University (US)
Quality Education
Openalex Percentile: Top 1%
Text and Document Classification Technologies
17.37
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.