HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Universal Language Model Fine-tuning for Text Classification

Jeremy Howard; Sebastian Ruder

Universal Language Model Fine-tuning for Text Classification

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.

Code Repositories

dpalominop/ULMFit
Mentioned in GitHub
mamamot/Russian-ULMFit
pytorch
Mentioned in GitHub
alexandra-chron/wassa-2018
pytorch
Mentioned in GitHub
LamLauChiu/Tensorflow_Learning
tf
Mentioned in GitHub
Julian1070/Deep-Learning
pytorch
Mentioned in GitHub
PrideLee/sentiment-analysis
pytorch
Mentioned in GitHub
cstorm125/thai2fit
pytorch
Mentioned in GitHub
prajjwal1/language-modelling
pytorch
Mentioned in GitHub
mdaniluk/language-detector
Mentioned in GitHub
magic-lantern/nlp-transfer-learning
pytorch
Mentioned in GitHub
amagooda/SummaRuNNer_coattention
pytorch
Mentioned in GitHub
rodrigopivi/aida
tf
Mentioned in GitHub
uchange/ulangel
pytorch
Mentioned in GitHub
alantancr/Hotel-Recommender
tf
Mentioned in GitHub
khumbuai/keras_wiki_lm
tf
Mentioned in GitHub
apmoore1/language-model
pytorch
Mentioned in GitHub
lukexyz/Language-Models
pytorch
Mentioned in GitHub
benjaminvdb/110kDBRD
Mentioned in GitHub
muellerzr/CodeFest_2019
Mentioned in GitHub
rania000/SentAnalyser
pytorch
Mentioned in GitHub
alexandra-chron/ntua-slp-wassa-iest2018
pytorch
Mentioned in GitHub
floraxinru/NLP_HotelReviews
Mentioned in GitHub
varshinireddyt/ULMFiT
Mentioned in GitHub
AnttiKarlsson/finnish_ulmfit
pytorch
Mentioned in GitHub
algoashutosh/sentiment-analysis-project
pytorch
Mentioned in GitHub
fastai/fastai
Official
pytorch
simecek/Czech-ULMFiT
Mentioned in GitHub
haianhle/ULMFiT-Sentiment
Mentioned in GitHub
noise-field/Russian-ULMFit
pytorch
Mentioned in GitHub
Deepayan137/Adapting-OCR
pytorch
Mentioned in GitHub
nishee0521/Sarcasm-Detector
Mentioned in GitHub
ahmadelsallab/READ
tf
Mentioned in GitHub
AbhimanyuAryan/IMDB-NLP
Mentioned in GitHub
jannenev/ulmfit-language-model
pytorch
Mentioned in GitHub
floleuerer/fastai_ulmfit
Mentioned in GitHub
benjaminvdb/DBRD
Mentioned in GitHub
lukashedegaard/ride
pytorch
Mentioned in GitHub
Mees-Molenaar/protein_location
pytorch
Mentioned in GitHub
heye0507/individualNLPClassifier
pytorch
Mentioned in GitHub
akari0216/Paddle-awdlstm
paddle
Mentioned in GitHub
neburseni/NLP
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
sentiment-analysis-on-imdbULMFiT
Accuracy: 95.4
sentiment-analysis-on-yelp-binaryULMFiT
Error: 2.16
sentiment-analysis-on-yelp-fine-grainedULMFiT
Error: 29.98
text-classification-on-ag-newsULMFiT
Error: 5.01
text-classification-on-dbpediaULMFiT
Error: 0.80
text-classification-on-trec-6ULMFiT
Error: 3.6

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Universal Language Model Fine-tuning for Text Classification | Papers | HyperAI