Research on Stock Price Prediction Model Based on LSTM

Authors

  • Junrong Nie

DOI:

https://doi.org/10.54097/56k4q130

Keywords:

Stock price prediction; LSTM; Neural network; Sliding windows.

Abstract

Stock prices are a non-linear, long-term series of data. The research topic of this paper is to use the Long Short-Term Memory Network (LSTM) model to analyze the historical stock price data of BYD (BYD) to predict the future stock price trend, and to provide intelligent reference for investors by evaluating the model performance, aiming to improve the accuracy of stock price prediction. The accuracy and consistency of stock price prediction can be increased by using this method, which successfully takes advantage of the time characteristics in historical data. After explaining the idea and architecture of LSTM, this article creates a stock price prediction model based on it, using Mean Absolute Error (MAE) and Root Mean Square Error (RMSE) as evaluation metrics. At the same time, network structures with different layers were designed to verify the effectiveness of the model, and the experimental results were shown. The advantages and limitations of the model are highlighted, and the feasibility of deep learning in the stock financial market and the credibility and advantages of strategic decision-making are demonstrated.

Downloads

Download data is not yet available.

References

Liu J, Lin C M M, Chao F. Gradient boost with convolution neural network for stock forecast. Advances in Computational Intelligence Systems: Contributions Presented at the 19th UK Workshop on Computational Intelligence, September 4-6, 2019, Portsmouth, UK 19. Springer International Publishing, 2020: 155-165.

Yuhan W, Yameng Z, Guoliang W. Stock Price prediction based on BP neural network and high order fuzzy cognitive graph. Intelligent Computers and Applications, 2023, 13(08): 100-6+13.

Ariyo A A, Adewumi A O, Ayo C K. Stock price prediction using the ARIMA model. 2014 UKSim-AMSS 16th international conference on computer modelling and simulation. IEEE, 2014: 106-112.

Jinchen H, Shunyu L, Nan S, et al. Application of Deep Learning in Stock forecasting. Information Technology and Informatization, 2023, (09): 190-3.

Rather A M, Agarwal A, Sastry V N. Recurrent neural network and a hybrid model for prediction of stock returns. Expert Systems with Applications, 2015, 42(6): 3234-3241.

Peng Y, Yuhong L, Rongfen Z. Modeling and Analysis of Stock price prediction based on LSTM. Computer Engineering and Applications, 2019, 55(11): 209-12.

Yuping H, Weixuan L, Zuhuan X. Comparative analysis of deep learning frameworks based on TensorFlow and PyTorch. Modern Information Technology, 2020, 4(04): 80-2+7.

Hongrui Z, Lei X. Research on stock prediction based on LSTM-CNN-CBAM model. Computer Engineering and Applications, 2021, 57(03): 203-7.

Haowen W, Juan L, Ya G. Stock prediction model based on feature optimization and loss function improvement. Information Systems Engineering, 2023, (05): 125-8.

Lu M, Xu X. TRNN: An efficient time-series recurrent neural network for stock price prediction. Information Sciences, 2024, 657: 119951.

Mehmood F, Ahmad S, Whangbo T K. An efficient optimization technique for training deep neural networks. Mathematics, 2023, 11(6): 1360.

Adebiyi A A, Adewumi A O, Ayo C K. Comparison of ARIMA and artificial neural networks models for stock price prediction. Journal of Applied Mathematics, 2014.

Downloads

Published

26-04-2024

How to Cite

Nie, J. (2024). Research on Stock Price Prediction Model Based on LSTM. Highlights in Science, Engineering and Technology, 94, 486-492. https://doi.org/10.54097/56k4q130