sentiment-discovery
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.
numpy
pytorch (>= 0.4.1)
pandas
scikit-learn
matplotlib
unidecode
sentencepiece
seaborn
emoji
We've included our sentencepiece tokenizer model and vocab as a zip file:
sentencepiece tokenizer [1MB]
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.
FP16 Transformer LM [311MB]
FP16 mLSTM LM [169MB]
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):
Finetuned Plutchik Transformer [673MN]
Finetuned Plutchik mLSTM [433MB]
Lastly, we've also included already trained classification models for SST and IMDB binary sentiment classification:
Finetuned SST Transformer [621MB]
Transferred SST mLSTM [325MB]
Transferred IMDB mLSTM [325MB]
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:
sentiment classification of input text
unsupervised reconstruction/language modeling of a corpus of text (+ script for launching distributed workers)
transfer of learned language model to perform sentiment analysis on a specified corpus
sampling from language model to generate text (possibly of fixed sentiment) + heatmap visualization of sentiment in text
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.
上一篇:ising-gpu
下一篇:dlinput-tf
还没有评论,说两句吧!
热门资源
Keras-ResNeXt
Keras ResNeXt Implementation of ResNeXt models...
seetafaceJNI
项目介绍 基于中科院seetaface2进行封装的JAVA...
spark-corenlp
This package wraps Stanford CoreNLP annotators ...
capsnet-with-caps...
CapsNet with capsule-wise convolution Project ...
inferno-boilerplate
This is a very basic boilerplate example for pe...
智能在线
400-630-6780
聆听.建议反馈
E-mail: support@tusaishared.com