/srnn

Skipping Recurrent Neural Networks

Primary LanguageMATLABGNU General Public License v3.0GPL-3.0

Matlab implementation for Skipping Recurrent Neural Networks from:

Learning Visual Storylines with Skipping Recurrent Neural Networks
Gunnar A. Sigurdsson, Xinlei Chen, Abhinav Gupta
http://arxiv.org/abs/1604.04279

The code is a MATLAB implementation of a Recurrent Neural Network, wrapped by the S-RNN architecture. This code was recently released, so please let me know if you encounter any strange behaviour.

The code is organized as follows:

  • rnn_trainnet.m is a script used for training the S-RNN
  • rnn_testnet.m is a script used for generating a summary of multiple albums, and selecting the best summary
  • predictnext_short.m is a code to run the short-term prediction experiment from the paper
  • predictnext_long.m is a code to run the long-term prediction experiment from the paper

Citation

Please cite the following if it helps your research!

@article{sigurdsson2016learning,
  author = {Gunnar A. Sigurdsson and Xinlei Chen and Abhinav Gupta},
  title = {Learning Visual Storylines with Skipping Recurrent Neural Networks},
  journal = {ArXiv e-prints},
  eprint = {1604.04279}, 
  year = {2016},
  url = {http://arxiv.org/abs/1604.04279},
  pdf = {http://arxiv.org/pdf/1604.04279.pdf}
}