Efficient Online Learning Algorithms Based on LSTM Neural Networks | IEEE Journals & Magazine | IEEE Xplore

Efficient Online Learning Algorithms Based on LSTM Neural Networks


Abstract:

We investigate online nonlinear regression and introduce novel regression structures based on the long short term memory (LSTM) networks. For the introduced structures, w...Show More

Abstract:

We investigate online nonlinear regression and introduce novel regression structures based on the long short term memory (LSTM) networks. For the introduced structures, we also provide highly efficient and effective online training methods. To train these novel LSTM-based structures, we put the underlying architecture in a state space form and introduce highly efficient and effective particle filtering (PF)-based updates. We also provide stochastic gradient descent and extended Kalman filter-based updates. Our PF-based training method guarantees convergence to the optimal parameter estimation in the mean square error sense provided that we have a sufficient number of particles and satisfy certain technical conditions. More importantly, we achieve this performance with a computational complexity in the order of the first-order gradient-based methods by controlling the number of particles. Since our approach is generic, we also introduce a gated recurrent unit (GRU)-based approach by directly replacing the LSTM architecture with the GRU architecture, where we demonstrate the superiority of our LSTM-based approach in the sequential prediction task via different real life data sets. In addition, the experimental results illustrate significant performance improvements achieved by the introduced algorithms with respect to the conventional methods over several different benchmark real life data sets.
Published in: IEEE Transactions on Neural Networks and Learning Systems ( Volume: 29, Issue: 8, August 2018)
Page(s): 3772 - 3783
Date of Publication: 13 September 2017

ISSN Information:

PubMed ID: 28920911

Funding Agency:


Contact IEEE to Subscribe

References

References is not available for this document.