Yayınlanmış 1 Ocak 2021 | Sürüm v1
Dergi makalesi Açık

An Efficient and Effective Second-Order Training Algorithm for LSTM-Based Adaptive Learning

  • 1. Bilkent Univ, Dept Elect & Elect Engn, TR-06800 Ankara, Turkey
  • 2. Turkcell Technol, 5G R&D Team, TR-34854 Istanbul, Turkey

Açıklama

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.

Dosyalar

bib-9b021690-cf4d-4038-9dc2-fb055cc7125a.txt

Dosyalar (194 Bytes)

Ad Boyut Hepisini indir
md5:554a8e57caa509c1bd03619628ddde1b
194 Bytes Ön İzleme İndir