Recurrent Neural Network Regularization

Journal: arXiv (Cornell University) • • Cornell University • ISSN: 2331-8422

Abstract

We present a simple regularization technique for Recurrent Neural Networks (RNNs) with Long Short-Term Memory (LSTM) units. Dropout, the most successful technique for regularizing neural networks, does not work well with RNNs and LSTMs. In this paper, we show how to correctly apply dropout to LSTMs, and show that it substantially reduces overfitting on a variety of tasks. These tasks include language modeling, speech recognition, image caption generation, and machine translation.

Subjects & Keywords

How to Cite this Article

Wojciech Zaremba, Ilya Sutskever, Oriol Vinyals (2014). "Recurrent Neural Network Regularization". arXiv (Cornell University). https://doi.org/10.48550/arxiv.1409.2329