Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modeling

Venue

abs/1412.3555, vol. CoRR

Publication Year

2014

Keywords

Computer Science - Machine Learning,Computer Science - Neural and Evolutionary Computing

Authors

  • Junyoung Chung
  • Caglar Gulcehre
  • KyungHyun Cho
  • Yoshua Bengio

Abstract

In this paper we compare different types of recurrent units in recurrent neural networks (RNNs). Especially, we focus on more sophisticated units that implement a gating mechanism, such as a long short-term memory (LSTM) unit and a recently proposed gated recurrent unit (GRU). We evaluate these recurrent units on the tasks of polyphonic music modeling and speech signal modeling. Our experiments revealed that these advanced recurrent units are indeed better than more traditional recurrent units such as tanh units. Also, we found GRU to be comparable to LSTM.