2015

A Theoretically Grounded Application of Dropout in Recurrent Neural Networks

Yarin Gal

citations

Cite Score

57

AI summary

This paper introduces a new variational inference based dropout technique applied to LSTM and GRU models, achieving state-of-the-art results on language modeling with the Penn Treebank dataset and improving performance on sentiment analysis tasks. The technique outperforms existing methods and offers insights into using dropout with RNN models.

Main Contributions

  • Proposes a new variational inference based dropout technique for RNNs.
  • Applies the technique to LSTM and GRU models.
  • Achieves state-of-the-art results on language modeling with the Penn Treebank dataset (73.4 test perplexity).
  • Improves performance on sentiment analysis tasks.
  • Provides a theoretical grounding for dropout in approximate Bayesian inference for RNNs.

Abstract

Recurrent neural networks (RNNs) stand at the forefront of many recent developments in deep learning. Yet a major difficulty with these models is their tendency to overfit, with dropout shown to fail when applied to recurrent layers. Recent results at the intersection of Bayesian modelling and deep learning offer a Bayesian interpretation of common deep learning techniques such as dropout. This grounding of dropout in approximate Bayesian inference suggests an extension of the theoretical results, offering insights into the use of dropout with RNN models. We apply this new variational inference based dropout technique in LSTM and GRU models, assessing it on language modelling and sentiment analysis tasks. The new approach outperforms existing techniques, and to the best of our knowledge improves on the single model state-of-the-art in language modelling with the Penn Treebank (73.4 test perplexity). This extends our arsenal of variational tools in deep learning.

Citation Graph

Loading graph...

References [30]

Sort:
Filter:

D. P. Kingma, Jimmy Lei Ba - 2014

49 papers in library cite

Sepp Hochreiter, Jürgen Schmidhuber - 1997

94 papers in library cite

N. Srivastava, Geoffrey E. Hinton, Alex Krizhevsky, Ilya Sutskever, Ruslan R. Salakhutdinov - 2014

20 papers in library cite

Kyunghyun Cho, B. V. Merrienboer, C. G. Gulcehre, D. Bahdanau, F. Bougares, Holger Schwenk, Yoshua Bengio - 2014

38 papers in library cite

Ilya Sutskever, Oriol Vinyals, Quoc V. Le - 2014

58 papers in library cite

Geoffrey Hinton - 2013

13 papers in library cite

Geoffrey E. Hinton, N. Srivastava, Alex Krizhevsky, Ilya Sutskever, Ruslan R. Salakhutdinov - 2012

25 papers in library cite

Wojciech Zaremba, Ilya Sutskever, Oriol Vinyals - 2014

22 papers in library cite

M. Sundermeyer, R. Schluter, Hermann Ney - 2010

7 papers in library cite

N. Kalchbrenner, Phil Blunsom - 2013

27 papers in library cite

James Bergstra, O. Breuleux, F. Bastien, P. Lamblin, Razvan Pascanu, G. Desjardins, J. Turian, D. W. Farley, Yoshua Bengio - 2010

22 papers in library cite

C. Blundell, J. Cornebise, Koray Kavukcuoglu, Daan Wierstra - 2015

3 papers in library cite

V. Pham, T. Bluche, C. Kermorvant, J. Louradour - 2014

5 papers in library cite

Alex Graves - 2011

8 papers in library cite

R. Neal - 1995

3 papers in library cite

Yarin Gal, Zoubin Ghahramani - 2015

3 papers in library cite

Geoffrey E. Hinton, D. V. Camp - 1993

3 papers in library cite

J. Bayer - 2013

3 papers in library cite

D. J. Rezende, S. Mohamed, Daan Wierstra - 2014

3 papers in library cite

D. J. Mackay - 1992

2 papers in library cite

J. M. H. Lobato, R. A. Adams - 2015

2 papers in library cite

M. Pachitariu, M. Sahani - 2013

2 papers in library cite

D. Kingma, T. Salimans, M. Welling - 2015

2 papers in library cite

Yarin Gal, Zoubin Ghahramani - 2015

1 paper in library cites

A. K. Balan, V. Rathod, Kevin P. Murphy, M. Welling - 2015

1 paper in library cites

D. Barber, C. M. Bishop - 1998

1 paper in library cites

[28]Keras

Fchollet - 2015

1 paper in library cites

T. Moon, H. Choi, Honglak Lee, I. Song - 2015

1 paper in library cites

T. Bluche, C. Kermorvant, J. Louradour - 2015

1 paper in library cites

Cited by

9

papers in your library

Cites

13

papers in your library

Read

on December 25, 2025

Your review

Tags

Paper Aliases

No aliases