This repo has been deprecated. Please visit Megatron-LM for our up to date Large-scale unsupervised pretraining and finetuning code.
If you would still like to use this codebase, see our tagged releases and install required software/dependencies that was available publicly at that date.
This codebase contains pretrained binary sentiment and multimodel emotion classification models as well as code to reproduce results from our series of large scale pretraining + transfer NLP papers: Large Scale Language Modeling: Converging on 40GB of Text in Four Hours and Practical Text Classification With Large Pre-Trained Language Models. This effort was born out of a desire to reproduce, analyze, and scale the Generating Reviews and Discovering Sentiment paper from OpenAI.
The techniques used in this repository are general purpose and our easy to use command line interface can be used to train state of the art classification models on your own difficult classification datasets.
This codebase supports mixed precision training as well as distributed, multi-gpu, multi-node training for language models (support is provided based on the NVIDIA APEx project). In addition to training language models, this codebase can be used to easily transfer and finetune trained models on custom text classification datasets.
For example, a Transformer language model for unsupervised modeling of large text datasets, such as the amazon-review dataset, is implemented in PyTorch. We also support other tokenization methods, such as character or sentencepiece tokenization, and language models using various recurrent architectures.
The learned language model can be transferred to other natural language processing (NLP) tasks where it is used to featurize text samples. The featurizations provide a strong initialization point for discriminative language tasks, and allow for competitive task performance given only a few labeled samples. For example, we consider finetuning our models on the difficult task of multimodal emotion classification based on a subset of the plutchik wheel of emotions.
Created by Robert Plutchik, this wheel is used to illustrate different emotions in a compelling and nuanced way. He suggested that there are 8 primary bipolar emotions (joy versus sadness, anger versus fear, trust versus disgust, and surprise versus anticipation) with different levels of emotional intensity. For our classification task we utilize tweets from the SemEval2018 Task 1E-c emotion classification dataset to perform multilabel classification of anger, anticipation, disgust, fear, joy, sadness, surprise, and trust. This is a difficult task that suffers from real world classification problems such as class imbalance and labeler disagreement.
On the full SemEval emotion classification dataset we find that finetuning our model on the data achieves competitive state of the art performance with no additional domain-specific feature engineering.
Install the sentiment_discovery package with python3 setup.py install
in order to run the modules/scripts within this repo.
At this time we only support python3.
We've included our sentencepiece tokenizer model and vocab as a zip file:
We've included a transformer language model base as well as a 4096-d mlstm language model base. For examples on how to use these models please see our finetuning and transfer sections. Even though these models were trained with FP16 they can be used in FP32 training/inference.
We've also included classifiers trained on a subset of SemEval emotions corresponding to the 8 plutchik emotions (anger, anticipation, disgust, fear, joy, sadness, surprise, and trust):
Lastly, we've also included already trained classification models for SST and IMDB binary sentiment classification:
To use classification models that reproduce results from our original large batch language modeling paper please use the following commit hash and set of models.
We did not include pretrained models leveraging ELMo. To reproduce our papers' results with ELMo, please see our available resources.
Each file has a dictionary containing a PyTorch state_dict
consisting of a language model (lm_encoder keys) trained on Amazon reviews and a classifier (classifier key) as well as accompanying args
necessary to run a model with that state_dict
.
In the ./data
folder we've provided processed copies of the Binary Stanford Sentiment Treebank (Binary SST), IMDB Movie Review, and the SemEval2018 Tweet Emotion datasets as part of this repository. In order to train on the amazon dataset please download the "aggressively deduplicated data" version from Julian McAuley's original site. Access requests to the dataset should be approved instantly. While using the dataset make sure to load it with the --loose-json
flag.
In addition to providing easily reusable code of the core functionalities (models, distributed, fp16, etc.) of this work, we also provide scripts to perform the high-level functionalities of the original paper:
Classify an input csv/json using one of our pretrained models or your own.
Performs classification on Binary SST by default.
Output classification probabilities are saved to a .npy
file
python3 run_classifier.py --load_model ama_sst.pt # classify Binary SST
python3 run_classifier.py --load_model ama_sst_16.pt --fp16 # run classification in fp16
python3 run_classifier.py --load_model ama_sst.pt --text-key <text-column> --data <path.csv> # classify your own dataset
See here for more documentation.
Train a language model on a csv/json corpus. By default we train a weight-normalized, 4096-d mLSTM, with a 64-d character embedding.
This is the first step of a 2-step process to training your own sentiment classifier.
Saves model to lang_model.pt
by default.
python3 pretrain.py #train a large model on imdb
python3 pretrain.py --model LSTM --nhid 512 #train a small LSTM instead
python3 pretrain.py --fp16 --dynamic-loss-scale #train a model with fp16
python3 -m multiproc pretrain.py #distributed model training
python3 pretrain.py --data ./data/amazon/reviews.json --lazy --loose-json \ #train a model on amazon data
--text-key reviewText --label-key overall --optim Adam --split 1000,1,1
python3 pretrain.py --tokenizer-type SentencePieceTokenizer --vocab-size 32000 \ #train a model with our sentencepiece tokenization
--tokenizer-type bpe --tokenizer-path ama_32k_tokenizer.model
python3 pretrain.py --tokenizer-type SentencePieceTokenizer --vocab-size 32000 \ #train a transformer model with our sentencepiece tokenization
--tokenizer-type bpe --tokenizer-path ama_32k_tokenizer.model --model transformer \
--decoder-layers 12 --decoder-embed-dim 768 --decoder-ffn-embed-dim 3072 \
--decoder-learned-pos --decoder-attention-heads 8
bash ./experiments/train_mlstm_singlenode.sh #run our mLSTM training script on 1 DGX-1V
bash ./experiments/train_transformer_singlenode.sh #run our transformer training script on 1 DGX-1V
For more documentation of our language modeling functionality look here
In order to learn about our language modeling experiments and reproduce results see the training reproduction section in analysis.
For information about how we achieve numerical stability with FP16 training see our fp16 training analysis.
Given a trained language model, this script will featurize text from train, val, and test csv/json's. It then uses sklearn logistic regression to fit a classifier to predict sentiment from these features. Lastly it performs feature selection to try and fit a regression model to the top n most relevant neurons (features). By default only one neuron is used for this second regression.
python3 transfer.py --load mlstm.pt #performs transfer to SST, saves results to `<model>_transfer/` directory
python3 transfer.py --load mlstm.pt --neurons 5 #use 5 neurons for the second regression
python3 transfer.py --load mlstm.pt --fp16 #run model in fp16 for featurization step
bash ./experiments/run_sk_sst.sh #run transfer learning with mlstm on imdb dataset
bash ./experiments/run_sk_imdb.sh #run transfer learning with mlstm on sst dataset
Additional documentation of the command line arguments available for transfer can be found here
Given a trained language model and classification dataset, this script will build a classifier that leverages the trained language model as a text feature encoder.
The difference between this script and transfer.py
is that the model training is performed end to end: the loss from the classifier is backpropagated into the language model encoder as well.
This script allows one to build more complex classification models, metrics, and loss functions than transfer.py
.
This script supports building arbitrary multilable, multilayer, and multihead perceptron classifiers. Additionally it allows using language modeling as an auxiliary task loss during training and multihead variance as an auxiliary loss during training.
Lastly this script supports automatically selecting classification thresholds from validation performance. To measure validation performance this script includes more complex metrics including: f1-score, mathew correlation coefficient, jaccard index, recall, precision, and accuracy.
python3 finetune_classifier.py --load mlstm.pt --lr 2e-5 --aux-lm-loss --aux-lm-loss-weight .02 #finetune mLSTM model on sst (default dataset) with auxiliary loss
python3 finetune_classifier.py --load mlstm.pt --automatic-thresholding --threshold-metric f1 #finetune mLSTM model on sst and automatically select classification thresholds based on the validation f1 score
python3 finetune_classifier.py --tokenizer-type SentencePieceTokenizer --vocab-size 32000 \ #finetune transformer with sentencepiece on SST
--tokenizer-type bpe tokenizer-path ama_32k_tokenizer.model --model transformer --lr 2e-5 \
--decoder-layers 12 --decoder-embed-dim 768 --decoder-ffn-embed-dim 3072 \
--decoder-learned-pos --decoder-attention-heads 8 --load transformer.pt --use-final-embed
python3 finetune_classifier.py --automatic-thresholding --non-binary-cols l1 l2 l3 --lr 2e-5\ #finetune multilayer classifier with 3 classes and 4 heads per class on some custom dataset and automatically select classfication thresholds
--classifier-hidden-layers 2048 1024 3 --heads-per-class 4 --aux-head-variance-loss-weight 1. #`aux-head-variance-loss-weight` is an auxiliary loss to increase the variance between each of the 4 head's weights
--data <custom_train>.csv --val <custom_val>.csv --test <custom_test>.csv --load mlstm.pt
bash ./experiments/se_transformer_multihead.sh #finetune a multihead transformer on 8 semeval categories
See how to reproduce our finetuning experiments in the finetuning reproduction section of analysis.
Additional documentation of the command line arguments available for finetune_classifier.py
can be found here
A special thanks to our amazing summer intern Neel Kant for all the work he did with transformers, tokenization, and pretraining+finetuning classification models.
A special thanks to @csarofeen and @Michael Carilli for their help developing and documenting our RNN interface, Distributed Data Parallel model, and fp16 optimizer. The latest versions of these utilities can be found at the APEx github page.
Thanks to @guillitte for providing a lightweight pytorch port of openai's sentiment-neuron repo.
This project uses the amazon review dataset collected by J. McAuley
Want to help out? Open up an issue with questions/suggestions or pull requests ranging from minor fixes to new functionality.
May your learning be Deep and Unsupervised.