arXiv · 2105.00357
RotLSTM: Rotating Memories in Recurrent Neural Networks
Abstract
Long Short-Term Memory (LSTM) units have the ability to memorise and use long-term dependencies between inputs to generate predictions on time series data. We introduce the concept of modifying the cell state (memory) of LSTMs using rotation matrices parametrised by a new set of trainable weights. This addition shows significant increases of performance on some of the tasks from the bAbI dataset.
Explore related subjects
Keep this discovery
Vlad Velici, Adam Prügel-Bennett. 2021-05-01. RotLSTM: Rotating Memories in Recurrent Neural Networks. https://arxiv.org/abs/2105.00357
Cite the original work for its findings. Save a collection to share your selection of sources.