- Yay another push, another update!
- As I said in my previous update yesterday, I continue to develop a specific OpenAI version of the Transformer.
- In today's update, I added the LanguageModelHead definition w.r.t. original Tensorflow implementation + huggingface's PyTorch implementation.
- In addition, I added "Block" layer which contains (1) an attention operation followed by (2) a layer normalization followed by (3) an MLP and finally followed by (4) a layer normalization.
- In overall, the implementation is again similar to the reference codes; however, I added an extention to the "Block" and "Attention" part.
- OpenAI version uses Conv1D for its attention calculations while the "All you need is attention" version uses Linear layer (Nope Conv1D does not equal to a Linear layer: Source)
- Since I want to use built-in nn.MultiHeadAttention function, just for fun, I put a conditional into the "Block" definition. If you trigger it, you will use the attention with linear. Otherwise, your model will call OpenAI version.
- Lastly, the new model is still under construction. I am highly confident that it has many bugs even now =)
Table Of Contents
- Library Dependencies
- Project Skeleton
- Project Details
- Previous Updates
- References for Code Development
This is my personal, pet project which I apply machine learning and natural language processing stuffs by using PyTorch. I stopped working with Tensorflow after some hellish times that I could not do some basic extentions (such fasttext based oov embeddings, details are below). Also, Tensorflow's updates and functionality deprecation rate is annoying for me.
In this repository, I implement popular learning models and extend them with different minor adjustments (like variational dropouts). Even though it is really slow, I execute experiments by using these models on a dataset which me and my old colleagues in Huawei constructed (details are below, again) and try to announce experiment results.
Before diving into details, the python and library versions are as follows:
- python 3.6 (works well with 3.7, too)
- torch 1.0.1
- torchtext 0.3.1
- numpy 1.15.4 (due to PyTorch 1.0)
- setuptools 40.8.0 (Hell no idea why pipreqs put this into requirements.txt)
- spacy 2.0.16 (for interactive evaluation only)
- gensim 3.6.0 (for fasttext embeddings, as well as OOV Embedding generation.)
- adabound 0.0.5
I try to keep every part of the project clean and easy to follow. Even though the folders are self explanatory for me, let me explain them for those who may have hard time to understand.
./crf/CRF.pycontains the conditional random field implementation (not finished yet).
./datahelper/dataset_reader.pycontains the "DatasetLoader" object that reads a text dataset, splits it into 3 subsets (train/vali/test), creates vocabulary and iterators. It is a little bit hard-coded for the dataset I am using now. However, it is easy to make changes to use it for your own dataset.
./datahelper/embedding_helper.pyis a helper class to generate OOV word embeddings. To use Fasttext-based OOV embedding generation, it leverages Gensim!
./datahelper/preprocessor.pycontains the "Preprocessor" object and actions to apply on sentences.
./dropout_models/gaussian_dropout.pycontains the Gaussian Dropout object.
./dropout_models/variational_dropout.pycontains the Variational Dropout object.
./dropout_models/dropout.pycontains the Dropout object which you can select your dropout type among Bernoulli (basic), Gaussian and Variational dropout types.
./evaluation/evaluator.pyis the factory for evaluation objects that are used in model trainings as well as interactive evaluation.
./evaluation/xyz_evaluator.pymethods are the evaluator functions for specified models.
./model/xyz.pycontains network objects.
./model/Util_xyz.pycontains custom-defined objects that are used in
./optimizer/custom_optimizer.pycontains custom-defined optimizer objects.
./scorer/accuracy_scorer.pycontains classification accuracy metric calculations.
./scorer/ner_scorer.pycontains NER-task related metric calculations.
./training/trainer.pyis a class that returns the necessary trainer for the user's selected learning model
./training/xyz_trainer.pymethods are the trainer functions for specified models.
./utils/utils.pycontains both utility and common methods that are being used in several places in the project.
./main.pyis the main code. To execute this project, one needs to provide a valid
config.jsonfile which contains the necessary configuration properties.
./config/config.jsonis the configuration file.
- As the other Tensorflow-based repository, I will use the dataset that me and my old colleagues constructed 3 years ago. "English/Turkish Wikipedia Named-Entity Recognition and Text Categorization Dataset" is publicly available: https://data.mendeley.com/datasets/cdcztymf4k/1
- Text CNN, CharCNN, VDCNN, Conv-Deconv CNN, basic LSTM/GRU and Transformer (Google version) models are currently available to train and evaluate in the repository. More models will be added.
- Fasttext embeddings are used (by default but it can be changed). Eventually, one can use Torchtext to download the "pre-defined" embedding files. However, since Turkish embeddings were not included in, I manually edit the Torchtext backend codes (please check the "changes in the torchtext.txt" file). Also note that, everytime you update Torchtext, you need to re-add those changes again.
- Embeddings (whether random or pretrained) can be "static", "nonstatic", or "multichannel".
- For OOV words, OOVEmbeddingCreator is developed (under datahelper/embedding_helper). There are 5 different basic approaches defined to generate OOV embeddings: (1) zeros vector, (2) ones vector, (3) random vector (between 0, 1), (4) (r1, r2) ranged uniformly random vector, (5) Fasttext CharNgram-based vectors.
- Even though I am focusing on Turkish versions of the dataset, I believe "Preprocessor" can work for English dataset, too. In future, I may add more language specific methods.
- Main code loads properties from config.json (inside config folder).
- I tested all training, evaluation, model/vocabulary saving/loading aspects of the code for several epochs without any problem (except out of memory errors =)).
Better configuration/property reading, handling, instead of hard-coded dictionaries(Update: 11-Jan-2019)
Variational Dropout. Update: Variational and Gaussian dropout methods are added. Reference: Variational Dropout and the Local Reparameterization Trick
Extend main flow and learning models with respect to new dropout models.
Add character-level data preprocessing.
Add character-level data loading.
- Run the current piece of code for the aforementioned datasets and define a text categorization baseline (for both Turkish and English).
- Variational Dropout related extensions (current version is from 2015 paper but obviously more recent versions are out there for me to implement =)) + bayes by backprop for CNN (a.k.a. Bayesian CNN)
- Different learning algorithms (DeepCNN, LSTM, GRU, any-kind-of-hybrid versions of those algorithms, transformers).
Multilayer CNN(I removed this model and decided to continue with CharCNN and VDCNN instead).
- VDCNN (Very Deep CNN)
- Transformer (Attention is All You Need version) (Modified for Text Classification/NER!)
- Transformer (Improving Language Understanding by Generative Pre-Training version)
- Transformer-XL (Transformer-XL: Attentive Language Models Beyond a Fixed-Length Context version)
- Conv-Deconv CNN
- Encoder-Decoder GRU
- Encoder-Decoder LSTM
- Hybrid stuff (Like CNN+LSTM/GRU)
CRF layer to be able to do NER experiments.
- Add new models that will use CRF as their last layer (such as LSTMCRF, GRUCRF, CNNCRF, etc.)
- Develop NER-related performance metrics and update training/evaluation flows to use these metrics.
- For Turkish, I plan to add morphological disambiguation (https://github.com/erayyildiz/Neural-Morphological-Disambiguation-for-Turkish).
- Different language models.
- ELMO (pretrained Turkish/English embeddings)
- BERT (pretrained Turkish/English embeddings)
- Document length categorization/NER support (Conv-Deconv CNN implementation supports document-length tasks, but more support will come with ELMO and BERT update).
Important Note Before Start
I had to make some changes in the torchtext backend codes to be able to do several stuffs:
- I don't know why, torchtext does not split a dataset into 3 subsets (train/val/test) even if there is a function for it. I changed it to fix that issue. Hopefully, one day torchtext will fix it offically =)
- To be able to work with Turkish Fasttext embeddings, I added its respective alias.
- To be able to apply Fasttext's CharNGram to OOV words to generate OOV embeddings, a minor change has been made to Vector object.
- To be able to read any dataset without any problem, a minor change has been made to torchtext's utils.py.
Configuration JSON Format
To be able to run the main code, you need to provide a valid JSON file which contains 4 main properties. These are
dataset_propertiescontains dataset-related information such as path, embedding, batch information.
model_propertiescontains model-related parameters. Inside this property,
common_model_propertiescontains common properties for all models like embeddings, vocabulary size, etc.
model_name(like text_cnn, char_cnn, etc.) contains model-specific properties.
training_propertiescontains training-related properties.
evaluation_propertiescontains evaluation-related properties.
Details of the
config.json can be found in "/config/README.md" folder.
How to Run Main
If you make the necessary changes described in "changes in torchtext.txt" and prepare "config.json", you have two ways to run the code.
- If you are using an IDE, copy/paste your "config.json" file's path as an argument and press run button.
- If you are an old-school command window lover, type
python main.py --config /path/to/config.json.
Training from Scratch-Training from Checkpoint-Interactive Evaluation
You can train your model from 0th epoch until max_epoch, and/or continue your training from xth epoch to the end. You do not need to do anything extra for the first case; however, to be able to continue your training you need to make necessary changes in "config.json":
dataset_properties/checkpoint_pathis empty, the code will start a new training process. If you type your saved PyTorch model, the main flow will automatically load it and continue from where it left.
- Additionally, you can provide saved vocabulary files for words (
dataset_properties/saved_sentence_vocab(don't ask why it is sentence)) and labels (
- Additionally, you can provide saved vocabulary files for words (
To be able to activate interactive evaluation, you need to make necessary changes in "config.json":
model_properties/common_model_properties/run_mode's value to "eval_interactive".
- Provide your model's path to be evaluated and your saved vocabulary files' path by using
This section presents the Top-1 and Top-5 test accuracies for text categorization task of my experiments. Due to computational resource limit, I cannot test every single parameter/hyperparameter. In general, I hold algorithm parameters same for all experiments; however, I change embedding related parameters. I assume the result table is self-explanatory. As a final note, I won't share my best models and I won't guarantee reproducibility. Dataset splits (training/validation/test) are deterministic for all experiments, but anything else that needs random initialization is non-deterministic.
Note: Epoch is set to 20 for all experiments, until further notice (last update: 31-10-2018). However, if I believe that results may improve, I let the experiment run for 10 more epochs (at most 30 epoch per experiments).
Note 2 (Update: 22-01-2019): Most of the English-language experiments are executed in Google Cloud (by using 300$ initial credit). Since, I want to finish as many experiments as possible, I cannot increase the max_epoch from 20 to 30. In this experiments, I saw that validation loss and accuracies were improving in every epoch until the 20th, and I am pretty sure models can improve further. Unfortunately, I chose the maximum number of experiment runs instead of best results for each experiment in this trade-off.
Test Results for TextCNN
|#||Language||# Of Categories||Pre-trained Embedding||OOV Embedding||Embedding Training||Top-1 Test Accuracy||Top-5 Test Accuracy|
- Note that the experiment 14 resulted with a better score than 16, unlike other similar setups. The main reason is, I changed the "learning_rate" of the optimizer to a smaller value for the experiment 16 (well, for the sake of the experiment =)), and it appears that smaller learning rate made the learning process a bit slower (in terms of number of epochs). If I can find a chance to run this experiment again in Google Cloud (a.k.a. have enough credit to run it one more time), I will update the learning rate properly.
In this title, I will save the previous updates for me and the visitors to keep track.
- I've started to work on Improving Language Understanding by Generative Pre-Training version of Transformer architecture.
- The project will contain only languange model and classifier heads, and I will also add NER head. Rest of the original heads in the paper can be found in huggingface's github =)
- And yes, I know huggingface has a PyTorch implementation which I wrote to my References ages ago. Obviously, I won't c/p the already implemented code since it does not fit to my architectural flow as well as its PyTorch version is pretty old.
- The end product would be provide same results (if you be a good human being and pray to the randomization gods), however, the implementation will differ due to my architectural concerns and capabilities of the latest PyTorch.
- I've also updated my library versions (using PyTorch 1.1 now) but I forgot updating requirement.txt (It will be updated in next commit).
- As a side note, in the last version of PyTorch, MultiHeadedAttention is developed under "torch.nn". For testing and comparison purposes, I tried to use it; however, I could not make it work in my TransformerGoogle code (you can find the commented-out line in model file). Return value of this new method does not fit the rest of the methods in the model (it is solvable but I won't waste time for it).
- I know I said that I will push more updates 2 months ago, but failed to deliver it since I am a filthy lazy dude =)
- Hopefully, I will push some new stuff in 2-3 weeks. Next week, I have a US conference trip and I will buy myself a brand new monster laptop for both gaming and faster training times. After that I will
- Develop the last 2 transformer papers.
- Develop BERT and ELMO embeddings to the flow (not the algorithms but loading/inferring the pretrained model stuff). Maybe I can include GPT-2 models from the latest OpenAI paper but I am not sure about it for now (since that model is being used mainly for text generation).
- Semi-relevant update: After I got several e-mails from researchers around the world (even from European Commission =)), I decided to improve my dataset. I came up with a plan which I will execute after I push the updates I mentioned above. So, if you have any ideas and/or requests for the new dataset, just let me know.
- Recently, I read "Adaptive Gradient Methods with Dynamic Bound of Learning Rate" paper (paper - github) and decided to add it into my project. The presented results are promising, but I have not tested this new optimizer in my own experiments (for now).
- Since Adabound can be installed via
pip install, I updated the requirement.txt.
config.jsonis also updated, two new parameters are added related to Adabound.
- Since Adabound can be installed via
- Finally, personal issues are finalizing (new job, better state of mind =)).
- I plan to add more models for text categorization starting with new Transformer codes (check To-Do list for details).
- After two new Transformer code, I decide to add Elmo and BERT support (I won't train my own models but will use pretrained Turkish-English models).
References for Code Development
Below repositories really helped me to write a decent and working code: