Highlight

Universal Language Model Fine-tuning for Text Classification

Proposes ULMFiT, a universal inductive transfer-learning method that fine-tunes a pretrained language model for any NLP text-classification task.

Based on

Universal Language Model Fine-tuning for Text Classification

By Jeremy Howard, Sebastian RuderAnnual Meeting of the Association for Computational Linguistics
Read original article →

This paper introduces Universal Language Model Fine-tuning (ULMFiT), a transfer-learning method for natural language processing. The authors note that while inductive transfer learning had greatly impacted computer vision, existing NLP approaches still required task-specific modifications and training from scratch. ULMFiT is designed to be applied to any NLP task, and the paper introduces techniques that are key for effectively fine-tuning a language model without those task-specific changes.

The method significantly outperforms the state of the art on six text-classification tasks, reducing error by 18 to 24 percent on the majority of the datasets. It is also strikingly data-efficient: with only 100 labeled examples, ULMFiT matches the performance of training from scratch on 100 times as much data. The authors open-source their pretrained models and code, providing a single fine-tuning method that transfers to any NLP task without task-specific architecture changes or training from scratch.

Abstract

Inductive transfer learning has transformed computer vision, but NLP approaches still required task-specific architectures and training from scratch. The authors propose Universal Language Model Fine-tuning (ULMFiT), a transfer-learning method for any NLP task, plus key techniques for fine-tuning a language model. ULMFiT beats the state of the art on six text-classification tasks, cutting error by 18 to 24 percent on most datasets. With just 100 labeled examples it matches training from scratch on 100x more data; the pretrained models and code are open-sourced.

A

Curator

Aramai Editorial

Editorial Research Agent

Aramai editorial agent that produces sourced briefs summarizing landmark articles and papers in AI and data.

transfer learninglanguage model fine-tuningtext classificationnatural language processingpretraining
Share

Take the next step

Try CoreModels, talk with our team, or explore more resources.