An efficient and effective second-order training algorithm for LSTM-based adaptive learning

Date
2021-04-07
Editor(s)
Advisor
Supervisor
Co-Advisor
Co-Supervisor
Instructor
Source Title
IEEE Transactions on Signal Processing
Print ISSN
1053-587X
Electronic ISSN
1941-0476
Publisher
IEEE
Volume
69
Issue
Pages
2541 - 2554
Language
English
Journal Title
Journal ISSN
Volume Title
Series
Abstract

We study adaptive (or online) nonlinear regression with Long-Short-Term-Memory (LSTM) based networks, i.e., LSTM-based adaptive learning. In this context, we introduce an efficient Extended Kalman filter (EKF) based second-order training algorithm. Our algorithm is truly online, i.e., it does not assume any underlying data generating process and future information, except that the target sequence is bounded. Through an extensive set of experiments, we demonstrate significant performance gains achieved by our algorithm with respect to the state-of-the-art methods. Here, we mainly show that our algorithm consistently provides 10 to 45% improvement in the accuracy compared to the widely-used adaptive methods Adam, RMSprop, and DEKF, and comparable performance to EKF with a 10 to 15 times reduction in the run-time.

Course
Other identifiers
Book Title
Citation
Published Version (Please cite this version)