On the initialization of long short-term memory networks

TY - CHAP

T1 - On the initialization of long short-term memory networks

AU - Mehdipour Ghazi, Mostafa

AU - Nielsen, Mads

AU - Pai, Akshay

AU - Modat, Marc

AU - Cardoso, M. Jorge

AU - Ourselin, Sébastien

AU - Sørensen, Lauge

PY - 2019/1/1

Y1 - 2019/1/1

N2 - Weight initialization is important for faster convergence and stability of deep neural networks training. In this paper, a robust initialization method is developed to address the training instability in long short-term memory (LSTM) networks. It is based on a normalized random initialization of the network weights that aims at preserving the variance of the network input and output in the same range. The method is applied to standard LSTMs for univariate time series regression and to LSTMs robust to missing values for multivariate disease progression modeling. The results show that in all cases, the proposed initialization method outperforms the state-of-the-art initialization techniques in terms of training convergence and generalization performance of the obtained solution.

AB - Weight initialization is important for faster convergence and stability of deep neural networks training. In this paper, a robust initialization method is developed to address the training instability in long short-term memory (LSTM) networks. It is based on a normalized random initialization of the network weights that aims at preserving the variance of the network input and output in the same range. The method is applied to standard LSTMs for univariate time series regression and to LSTMs robust to missing values for multivariate disease progression modeling. The results show that in all cases, the proposed initialization method outperforms the state-of-the-art initialization techniques in terms of training convergence and generalization performance of the obtained solution.

KW - Deep neural networks

KW - Disease progression modeling

KW - Initialization

KW - Long short-term memory

KW - Time series regression

UR - http://www.scopus.com/inward/record.url?scp=85077503119&partnerID=8YFLogxK

U2 - 10.1007/978-3-030-36708-4_23

DO - 10.1007/978-3-030-36708-4_23

M3 - Conference paper

AN - SCOPUS:85077503119

SN - 9783030367077

T3 - Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)

SP - 275

EP - 286

BT - Neural Information Processing - 26th International Conference, ICONIP 2019, Proceedings

A2 - Gedeon, Tom

A2 - Wong, Kok Wai

A2 - Lee, Minho

PB - SPRINGER

T2 - 26th International Conference on Neural Information Processing, ICONIP 2019

Y2 - 12 December 2019 through 15 December 2019

ER -