Eigenvalue Normalized Recurrent Neural Networks for Short Term Memory


  • Kyle Helfrich University of Kentucky
  • Qiang Ye University of Kentucky




Several variants of recurrent neural networks (RNNs) with orthogonal or unitary recurrent matrices have recently been developed to mitigate the vanishing/exploding gradient problem and to model long-term dependencies of sequences. However, with the eigenvalues of the recurrent matrix on the unit circle, the recurrent state retains all input information which may unnecessarily consume model capacity. In this paper, we address this issue by proposing an architecture that expands upon an orthogonal/unitary RNN with a state that is generated by a recurrent matrix with eigenvalues in the unit disc. Any input to this state dissipates in time and is replaced with new inputs, simulating short-term memory. A gradient descent algorithm is derived for learning such a recurrent matrix. The resulting method, called the Eigenvalue Normalized RNN (ENRNN), is shown to be highly competitive in several experiments.




How to Cite

Helfrich, K., & Ye, Q. (2020). Eigenvalue Normalized Recurrent Neural Networks for Short Term Memory. Proceedings of the AAAI Conference on Artificial Intelligence, 34(04), 4115-4122. https://doi.org/10.1609/aaai.v34i04.5831



AAAI Technical Track: Machine Learning