Recurrent Neural Network Generator

RnnGen is a generative natural language processing library using Word2Vec. Everything is built from scratch for learning purposes, using numpy for machine learning math.
With RnnGen, you can use your own set of text or an already implemented, create word vector embeddings of it and
train your own word generator!

Installation

Use the package manager pip to install RnnGen.

pip install rnngen

Dependencies

Python 3.6.7 numpy=<1.15.4
matplotlib=<3.0.2
sklearn (scikit-learn)=<0.19.1 (only for cosine similarity)

Usage

import rnngen  # Imports package
from rnngen import datasets  # Imports dataset directories

# Will start training the model with default settings and pre-trained embeddings. 
# Only datasets.SIMPSONS_PROCESSED will work if using default embeddings_dir parameter.
rnngen.generate(datasets.SIMPSONS_PROCESSED)

Advanced Usage

All parameters should be tuned in setupvariables.py for optimal results. To use your own data, you will need a txt file at least the size of a book, and some variables might need to get changed. Read setupvariables.py for more info.

Preprocessing rnngen.pre_process()

Preprocessing of data makes the text clean and workable.
pre_process takes a text directory and a directory to save the processed text.

import rnngen

rnngen.pre_process('dir_file_to_process', 'dir_to_save_processed_text')

Expected output:

Preprocessing 'dir_file_to_process' to directory 'dir_to_save_processed_text'
0 of 14000
10000 of 14000
...
Preprocessing done.

You can now access your processed file and use it to train word2vec and generator. (The same processed text MUST be used to generate words AND to train word2vec.)

Word2Vec training rnngen.word2vec()

Word2Vec uses a processed text to train word embeddings. The term Word Embeddings are explained in 'Knowledge Prerequisites' below.

Already trained and ready to use datasets can be used: datasets.SIMPSONS_PROCESSED (400 episodes of simpsons) and datasets.BOOKS_PROCESSED(25 books).

Word2Vec takes a processed text and a directory to save the embeddings.

rnngen.word2vec('dir_to_processed_file', 'dir_to_save_embeddings')

or to use already processed data:

rnngen.word2vec(datasets.SIMPSONS_PROCESSED, 'dir_to_save_embeddings')

Expected output:

While training, word2vec will continuously verbose loss/earlier loss, number of iterations and word2vec cosine similarity (explained in Understanding Prerequisies). The cosine similarity will take 2 random words and see how similar they are, and for us humans to judge the quality of the embeddings. He and She is always verbose and should be as close to 1 as possible.

Loss: 0.2342 [1.46, 0.4549, 0.3594, 0.3191, 0.256, 0.2449]  # Current loss followed by earlier losses
Iter: 600000 of 646860  # Current iteration
Epoch: 9 of 10  # Current epoch
he | she:  0.8017 # Two tested word embeddings, followed by a high word similarity.
almost | tv:  0.0279 # These are two very different words, therefore low similarity.
problem | window:  0.1334 # Can often be used interchangeably, therefore medium cosine similarity.

After training is done, test_embeddings will be called. It will take 10 random words, and print out the 5 most similar to them.

The metric here is also cosine similarity. If these look correct/similar and the words can be used interchangeably, your embeddings are probably good.

These words have the highest cosine similarity (most similar) to "great".
great 1.0  # The embedding of great is exactly the same as itself (no surprise)
good 0.762  # The embedding of great is very similar to the embedding of 'good'. 
terrible 0.567  # Even though terrible has the opposite meaning, both can still be used at the same places.
wonderful 0.553  
perfect 0.456

# More interchangeable examples
These words have the highest cosine similarity (most similar) to "he".
he 1.0
she 0.935
youve 0.865
ive 0.832
weve 0.831

These words have the highest cosine similarity (most similar) to "cat".
cat 1.0
dog 0.752
clown 0.486  # Although very different words, they are both objects, and therefore have high similarity.
bus 0.454
husband 0.423

Generator training rnngen.generate()

The generator generates sentences using word embeddings. It takes a processed text and directory to embeddings. You can also choose to train using sparse vectors, by setting word2vec to False.
If embeddings_dir is not specified to something, it will use datasets.SIMPSONS_EMBEDDINGS_300 as default, and then only datasets.SIMPSONS_PROCESSED can be used as text.

Default: rnngen.generate('dir_to_processed_file', use_word2vec=True, embeddings_dir=datasets.SIMPSONS_EMBEDDINGS_300)
The pretrained embeddings are: datasets.SIMPSONS_EMEBDDINGS_300 and SIMPSONS_EMBEDDINGS_100 (300 respectively 100 embedding_size)

The model will have to train for a few days to give good results.

import rnngen
from rnngen import datasets

rnngen.generate('dir_to_processed_text',  use_word2vec=True, embeddings_dir=datasets.SIMPSONS_EMBEDDINGS_300)

Expected output

Every 10 seconds it will verbose a loss and two predicted text. The upper text is chosen using a probability distribution and the below one is independently predicted using a greedy 'take the one with highest probability'.

loss: 6.82 #START# relax pretty cheese guy 
           #START# im what school say  
           
loss: 4.9 #START# seeing everybody eating ruined you and me 
          #START# im sorry , nice and think opening  

Example of full usage

# Example using datasets.BOOKS (which is './rnngen/resources/datasets/books.txt')

import rnngen
from rnngen import datasets

rnngen.pre_process(datasets.BOOKS, 'processed_text')
rnngen.word2vec('processed_text', 'word_embeddings')
rnngen.generate('processed_text', embeddings_dir='word_embeddings') # Must use SAME processed_text as in word2vec

Knowledge prerequisites for understanding

Word2Vec:

Instead of sparse vector representations of words, you can use dense representations. These are called embeddings.
Link for more info: https://en.wikipedia.org/wiki/Word2vec

Common words

Embedding: A vector of size 300 (default) that represent a word. Embeddings in plural is a matrix (vocabulary size, embedding size)

Cosine Similarity: Is used for testing embeddings. The cosine similarity between two word vectors is the semantic similarity. A cosine similarity of 1 means interchangeable words, and a cosine similarity of -1 means completely different words. The words 'he' and 'she' should have close to 1. https://en.wikipedia.org/wiki/Cosine_similarity

Recurrent neural network:

Instead of normal neural networks, recurrent neural networks can be used to look not just at the current input, but at inputs and probabilities in the past, and is therefore good for forecasting or creating generative models that depend on earlier inputs, such as text. You need to have the past context to keep on writing.
Link for more info: https://medium.com/explore-artificial-intelligence/an-introduction-to-recurrent-neural-networks-72c97bf0912

Common words

Backpropagation look back (bp_look_back): The number of words the model will look back at while training. If set to 5, the model will look at the last 4 words and use them to predict the next one.

Hidden state: The special part with recurrent networks is the hidden state, which saves information from past training and uses the information to hopefully generate legitimate sentences.

Important parameters in setupvariables.py

parameters_setup:

Is responsible for parameters when training Rnn, and is essential to tune for good results.

preprocess_setup:

'WORD_THRESHOLD'

Removes all sentences that contain a word that appears less than 'WORD_THRESHOLD' times. It is defaulted at 40 because of huge datasets, but if you try on your own smaller dataset, this needs to be lowered to under 3.

'TRAINING_TYPE'

Is defaulted to 'words' but can be set to 'letters'. When set to 'letters', the generative model will divide the words into letters instead, and predict letter for letter. If 'letters', do not forget to increase 'BP_LOOK_BACK'. If it is set to 4, it will only be able to look 4 letters back in time and can therefore not create any complex words. Also set 'MIN_WORDS' to around 20 to prevent short sentences.

word2vec_params_setup:

Contains all word2vec training parameters. Can be tuned so training takes shorter time and more optimal learning rate decay, but it can also increase quality on word embeddings.

More info about each variable is in the setupvariables.py file.

Conclusion and what I have learned

Word2Vec is a very interesting form of representing words, and by experimenting with them, there is a lot to be learned. For example, if you would take the embedding of king, subtract man and add woman, the embedding would become very similar to queen. (Can be done with more sophisticated embeddings for example google)

Recurrent neural networks do find some correlations in texts, for example, knows when to stop and when to put a comma, and how some word groups appear after another, but the recurrent algorithm with hidden states is not powerful enough to create longer meaningful sentences based on other text. But I'm still amazed that an the algorithms can make sense of spoken english and even create some of its own, even of limited quality.

To do the backpropagation with numpy in the recurrent neural network has really helped me to understand the mechanics of backpropagation and how to deploy it on different algorithms.

TODO/What could be done in the future

The next obvious step is to evolve the vanilla recurrent algorithm into a GRU or LSTM, which are similar algorithms but that has a much better memory of what has happened in the past, which means longer meaningful sentences can be generated. If we also combine this with an attention mechanism (which essentially is something that reminds the algorithm of the subject it generates text on) we could create long texts and even articles, which is already made today. https://digiday.com/media/washington-posts-robot-reporter-published-500-articles-last-year/

Other improvements that are of diminishing returns are: Increase text data, training all models for longer with increased nodes and test different processing techniques.