LSTM Network Hyperparameter Optimization for Stock Price Prediction Using the Optuna Framework

Ismanto, Edi and Vitriani, Vitriani (2023) LSTM Network Hyperparameter Optimization for Stock Price Prediction Using the Optuna Framework. Jurnal Ilmiah Teknik Elektro Komputer dan Informatika (JITEKI), 9 (1). pp. 22-35.

[thumbnail of 3-LSTM Network Hyperparameter Optimization for Stock Price Prediction Using the Optuna Framework.pdf] Text
3-LSTM Network Hyperparameter Optimization for Stock Price Prediction Using the Optuna Framework.pdf

Download (1MB)

Abstract

In recent years, the application of deep learning-based financial modeling tools has grown in popularity. Research on stock forecasting is crucial to understanding how a nation's economy is doing. The study of intrinsic value and stock market forecasting has significant theoretical implications and a broad range of potential applications. One of the trickiest challenges in projects involving deep learning and machine learning is hyperparameter search. In this paper, we evaluate and analyze the optimal hyperparameter search in the long short-term memory (LSTM) model developed to forecast stock prices using the Optuna framework. This study contributes to developing the LSTM algorithm model for predicting stock prices. Applying the optuna framework to the LSTM model to improves the search for the ideal hyperparameter. We examined a number of hyperparameters with several LSTM architectures, including optimizers (SGD, Adagrad, RMSprop, Nadam, Adamax, dan Adam), LSTM hidden units, dropout rates, epochs, batch size, and learning rate. The results of the experiment indicated that of the four LSTM models tested—model 1 single LSTM, model 2 single LSTM, model 1 LSTM stacked, and model 2 LSTM stacked—model 1 single LSTM was the most effective. Single LSTM version 1 offers the lowest losses when compared to other models and had the lowest root mean square error (RMSE) score of 7.21. When compared to manual hyperparameter tuning, automatic hyperparameter tuning has lower losses and is better.

Item Type: Artikel Umum
Subjects: T Technology > TK Electrical engineering. Electronics Nuclear engineering
Divisi / Prodi: Faculty of Industrial Technology (Fakultas Teknologi Industri) > S1-Electrical Engineering (S1-Teknik Elektro)
Depositing User: M.Eng. Alfian Ma'arif
Date Deposited: 14 Mar 2023 08:30
Last Modified: 14 Mar 2023 08:30
URI: http://eprints.uad.ac.id/id/eprint/41357

Actions (login required)

View Item View Item