Deterministic convergence analysis for regularized long short-term memory and its application to regression and multi-classification problems

Qian Kang, Dengxiu Yu, Kang Hao Cheong, Zhen Wang

Research output: Contribution to journalArticlepeer-review

4 Scopus citations

Abstract

Long short-term memory (LSTM) is a recurrent neural network (RNN) framework designed to solve the gradient disappearance and explosion problems of traditional RNNs. In recent years, LSTM has become a state-of-the-art model for solving various machine-learning problems. This paper propose a novel regularized LSTM based on the batch gradient method. Specifically, the L2 regularization is appended to the objective function as a systematic external force, effectively controlling the excessive growth of weights in the network and preventing the overfitting phenomenon. In addition, a rigorous convergence analysis of the proposed method is carried out, i.e., monotonicity, weak convergence, and strong convergence results are obtained. Finally, comparative simulations are conducted on the benchmark data set for regression and classification problems, and the simulation results verify the effectiveness of the method.

Original languageEnglish
Article number108444
JournalEngineering Applications of Artificial Intelligence
Volume133
DOIs
StatePublished - Jul 2024

Keywords

  • Batch gradient algorithm
  • Convergence
  • Long short-term memory
  • Regularization

Fingerprint

Dive into the research topics of 'Deterministic convergence analysis for regularized long short-term memory and its application to regression and multi-classification problems'. Together they form a unique fingerprint.

Cite this