How to Fine-Tune BERT for Text Classification?

被引:647
|
作者
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
来源
关键词
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] HOW TO FINE-TUNE A SUPERCONDUCTOR
    EMSLEY, J
    NEW SCIENTIST, 1990, 128 (1739) : 23 - 23
  • [2] How to Fine-Tune an Epigenetic Switch
    Pires, Nuno D.
    Grossniklaus, Ueli
    DEVELOPMENTAL CELL, 2012, 23 (03) : 453 - 454
  • [3] Should You Fine-Tune BERT for Automated Essay Scoring?
    Mayfield, Elijah
    Black, Alan W.
    INNOVATIVE USE OF NLP FOR BUILDING EDUCATIONAL APPLICATIONS, 2020, : 151 - 162
  • [4] Fine-tune BERT with Sparse Self-Attention Mechanism
    Cui, Baiyun
    Li, Yingming
    Chen, Ming
    Zhang, Zhongfei
    2019 CONFERENCE ON EMPIRICAL METHODS IN NATURAL LANGUAGE PROCESSING AND THE 9TH INTERNATIONAL JOINT CONFERENCE ON NATURAL LANGUAGE PROCESSING (EMNLP-IJCNLP 2019): PROCEEDINGS OF THE CONFERENCE, 2019, : 3548 - 3553
  • [5] Pre-train, Interact, Fine-tune: a novel interaction representation for text classification
    Zheng, Jianming
    Cai, Fei
    Chen, Honghui
    de Rijke, Maarten
    INFORMATION PROCESSING & MANAGEMENT, 2020, 57 (06)
  • [6] How to fine-tune your business processes
    Epelman, M
    Brimley, D
    Maher, D
    O'Brien, D
    QUALITY PROGRESS, 1998, 31 (10) : 55 - 60
  • [7] FINE-TUNE LANTASTIC
    ISMAIL, AR
    COPLEY, R
    BYTE, 1994, 19 (08): : 55 - &
  • [8] How to fine-tune the cell's force producers
    Lappalainen, Pekka
    NATURE, 2019, 565 (7739) : 297 - 298
  • [9] Hepatitis C: How to fine-tune your approach
    Wangensteen, Laura
    Wangensteen, Kirk J.
    Evans, Susanna G.
    Everts, Leslie E.
    Trooskin, Stacey B.
    JOURNAL OF FAMILY PRACTICE, 2015, 64 (09): : 535 - 540
  • [10] Fine-tune SCR operations
    Divinia, D
    HYDROCARBON PROCESSING, 2005, 84 (01): : 81 - 82