English

Universal Language Model Fine-tuning for Text Classification

Computation and Language 2018-05-24 v5 Machine Learning Machine Learning

Abstract

Inductive transfer learning has greatly impacted computer vision, but existing approaches in NLP still require task-specific modifications and training from scratch. We propose Universal Language Model Fine-tuning (ULMFiT), an effective transfer learning method that can be applied to any task in NLP, and introduce techniques that are key for fine-tuning a language model. Our method significantly outperforms the state-of-the-art on six text classification tasks, reducing the error by 18-24% on the majority of datasets. Furthermore, with only 100 labeled examples, it matches the performance of training from scratch on 100x more data. We open-source our pretrained models and code.

Keywords

Cite

@article{arxiv.1801.06146,
  title  = {Universal Language Model Fine-tuning for Text Classification},
  author = {Jeremy Howard and Sebastian Ruder},
  journal= {arXiv preprint arXiv:1801.06146},
  year   = {2018}
}

Comments

ACL 2018, fixed denominator in Equation 3, line 3

R2 v1 2026-06-22T23:49:05.764Z