arrow
返回

Enhancing text classification with attention matrices based on BERT

delete2023-11-22
delete6
PRE
AI
Z
Zhiyi Yu *
李
李红 (Hong Li)
J
J. Feng
DOI:10.1111/exsy.13512delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Text classification is a critical task in the field of natural language processing. While pre-trained language models like BERT have made significant strides in improving performance in this area, the distinctive dependency information that is present in text has not been fully exploited. Besides, BERT mostly captures phrase-level information in lower layers, which becomes progressively weaker with the increasing depth of layers. To address these limitations, our work focuses on enhancing text classification through the incorporation of Attention Matrices, particularly in the fine-tuning process of pre-trained models like BERT. Our approach, named AM-BERT, leverages learned dependency relationships as external knowledge to enhance the pre-trained model by generating attention matrices. In addition, we introduce a new learning strategy that enables the model to retain learned phrase-level structure information. Extensive experiments and detailed analysis on multiple benchmark datasets demonstrate the effectiveness of our approach in text classification tasks. Furthermore, we show that AM-BERT achieves stable performance improvements also in named entity recognition tasks.
Keyword:
attention matrix
dependency parsing
phrase-level structure information
text classification

期刊

Expert Systems 封面图
Expert Systems
IF:
2.3
论文数:
2.6K
被引数:
3.8K

机构

C
Central South University
学者数:
10.0W
论文数: 7.2W
被引数: 10.9W
引用论文

引用论文

Palpatory Apraxia
err2008-01-30
err0
PREAI
errAtsushi Yamadori
err分享
err收藏
err分享
err收藏
AP-BERT: enhanced pre-trained model through average pooling
err2022-03-19
err24
PREAI
errZhao, Shuai; Zhang, Tianyu; Hu, Man; Chang, Wen; You, Fucheng
err分享
err收藏
YAC transgene-mediated olfactory receptor gene choiceYAC转基因介导的嗅觉受体基因选择
err2000-02-01
err0
errOAAI
errFarah A.W. Ebrahimi; James Edmondson; Rodney Rothstein; Andrew Chess
err分享
err收藏
UHRF1 upregulation mediates exosome release and tumor progression in osteosarcoma
err
IF0
err2020-05-25
err0
errOAAI
errStephanie C. Wu; Ahhyun Kim; Loredana Zocchi; Claire Chen; Jocelyne Lopez; Kelsey Salcido; Jie Wu; Claudia A. Benavente
err分享
err收藏
Hydrodynamic studies of dilute pitch solutions: The shape and size of pitch molecules
err1983-01-01
err0
PREAI
errMototsugu Sakai; Kiyomi Sasaki; Michio Inagaki
err分享
err收藏
err分享
err收藏
学者 查看更多内容