How to Fine-Tune BERT for Text Classification?

被引:720
|
作者
Sun, Chi [1 ]
Qiu, Xipeng [1 ]
Xu, Yige [1 ]
Huang, Xuanjing [1 ]
机构
[1] Fudan Univ, Sch Comp Sci, Shanghai Key Lab Intelligent Informat Proc, 825 Zhangheng Rd, Shanghai, Peoples R China
来源
CHINESE COMPUTATIONAL LINGUISTICS, CCL 2019 | 2019年 / 11856卷
关键词
Transfer learning; BERT; Text classification;
D O I
10.1007/978-3-030-32381-3_16
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Language model pre-training has proven to be useful in learning universal language representations. As a state-of-the-art language model pre-training model, BERT (Bidirectional Encoder Representations from Transformers) has achieved amazing results in many language understanding tasks. In this paper, we conduct exhaustive experiments to investigate different fine-tuning methods of BERT on text classification task and provide a general solution for BERT fine-tuning. Finally, the proposed solution obtains new state-of-the-art results on eight widely-studied text classification datasets.
引用
收藏
页码:194 / 206
页数:13
相关论文
共 50 条
  • [1] Utilizing BERT Pretrained Models with Various Fine-Tune Methods for Subjectivity Detection
    Huo, Hairong
    Iwaihara, Mizuho
    WEB AND BIG DATA, PT II, APWEB-WAIM 2020, 2020, 12318 : 270 - 284
  • [2] Contextual Embeddings-Based Web Page Categorization Using the Fine-Tune BERT Model
    Nandanwar, Amit Kumar
    Choudhary, Jaytrilok
    SYMMETRY-BASEL, 2023, 15 (02):
  • [3] A general fine-tune method for catastrophic forgetting
    Tao, Yang
    Zhu, Mingming
    Li, Hao
    Yuan, Cao
    MIPPR 2019: AUTOMATIC TARGET RECOGNITION AND NAVIGATION, 2020, 11429
  • [4] Federated Freeze BERT for text classification
    Omar Galal
    Ahmed H. Abdel-Gawad
    Mona Farouk
    Journal of Big Data, 11
  • [5] Federated Freeze BERT for text classification
    Galal, Omar
    Abdel-Gawad, Ahmed H.
    Farouk, Mona
    JOURNAL OF BIG DATA, 2024, 11 (01)
  • [6] How Deeply to Fine-Tune a Convolutional Neural Network: A Case Study Using a Histopathology Dataset
    Kandel, Ibrahem
    Castelli, Mauro
    APPLIED SCIENCES-BASEL, 2020, 10 (10):
  • [7] Turkish Medical Text Classification Using BERT
    Celikten, Azer
    Bulut, Hasan
    29TH IEEE CONFERENCE ON SIGNAL PROCESSING AND COMMUNICATIONS APPLICATIONS (SIU 2021), 2021,
  • [8] Federated Split BERT for Heterogeneous Text Classification
    Li, Zhengyang
    Si, Shijing
    Wang, Jianzong
    Xiao, Jing
    2022 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2022,
  • [9] Rethinking of BERT sentence embedding for text classification
    Omar Galal
    Ahmed H. Abdel-Gawad
    Mona Farouk
    Neural Computing and Applications, 2024, 36 (32) : 20245 - 20258
  • [10] An Automated Text Document Classification Framework using BERT
    Shah, Momna Ali
    Iqbal, Muhammad Javed
    Noreen, Neelum
    Ahmed, Iftikhar
    INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2023, 14 (03) : 279 - 285