https://github.com/nicklashansen/rnn_lstm_from_scratch

How to build RNNs and LSTMs from scratch with NumPy.
https://github.com/nicklashansen/rnn_lstm_from_scratch

deep-learning lstm-neural-networks natural-language-processing numpy pytorch recurrent-neural-networks

Last synced: 2 months ago
JSON representation

How to build RNNs and LSTMs from scratch with NumPy.

• Host: GitHub
• URL: https://github.com/nicklashansen/rnn_lstm_from_scratch
• Owner: nicklashansen
• License: gpl-3.0
• Created: 2019-10-08T20:42:22.000Z (over 5 years ago)
• Default Branch: master
• Last Pushed: 2020-10-04T08:39:49.000Z (over 4 years ago)
• Last Synced: 2024-08-07T23:12:56.256Z (6 months ago)
• Topics: deep-learning, lstm-neural-networks, natural-language-processing, numpy, pytorch, recurrent-neural-networks
• Language: Jupyter Notebook
• Homepage:
• Size: 218 KB
• Stars: 243
• Watchers: 6
• Forks: 70
• Open Issues: 2
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# How to build RNNs and LSTMs from scratch with NumPy

**\[Update 08/18/2020\] Improvements to dataset; exercises and descriptions have been made more clear.**

Originally developed by me (Nicklas Hansen), Peter E. Christensen and Alexander R. Johansen as educational material for the graduate deep learning course at the Technical University of Denmark (DTU). You can access the full course material [here](https://github.com/DeepLearningDTU/02456-deep-learning-with-PyTorch). Inspired by the great [Andrej Karpathy](https://karpathy.ai/).

____

In this lab we will introduce different ways of learning from sequential data.

As an example, we will train a neural network to do language modelling, i.e. predict the next token in a sentence. In the context of natural language processing a token could be a character or a word, but mind you that the concepts introduced here apply to all kinds of sequential data, such as e.g. protein sequences, weather measurements, audio signals or monetary transaction history, just to name a few.

To really get a grasp of what is going on inside the recurrent neural networks that we are about to teach you, we will carry out a substantial part of this exercise in NumPy rather than PyTorch. Once you get a hold of it, we will proceed to the PyTorch implementation.

In this notebook we will show you:

* How to represent categorical variables in networks

* How to build a recurrent neural network (RNN) from scratch

* How to build a LSTM network from scratch

* How to build a LSTM network in PyTorch

## Dataset

For this exercise we will create a simple dataset that we can learn from. We generate sequences of the form:

`a b EOS`,

`a a b b EOS`,

`a a a a a b b b b b EOS`

where `EOS` is a special character denoting the end of a sequence. The task is to predict the next token `t_n`, i.e. `a`, `b`, `EOS` or the unknown token `UNK` given the sequence of tokens `t_1`, `t_2`, `...`, `t_n-1` and we are to process sequences in a sequential manner. As such, the network will need to learn that e.g. 5 `b`s and an `EOS` token will follow 5 `a`s.

## Results

The RNN takes considerable effort to converge to a nice solution:

![RNN loss](https://i.imgur.com/aSeXXI5.png)

The LSTM learns much faster than the RNN:

![LSTM loss](https://i.imgur.com/MioBMl2.png)

And finally, the PyTorch LSTM learns even faster and converges to a better local minimum:

![PyTorch LSTM loss](https://i.imgur.com/6HrHHRp.png)

After working your way through these exercises, you should have a better understanding of how RNNs work, how to train them, and what they can be used for. And the conclusion? - use PyTorch.