Stock Price Prediction Based on Optimized LSTM Model

Wandong Zhai
DOI: https://doi.org/10.4108/eai.7-7-2023.2338038
Abstract:: The variation trend of stock prices is often disturbed by various factors such as investor sentiment, market sentiment, and corporate performance. The combined effect of these factors leads to stochastic fluctuations in stock price change, which results in its high-noise and unstable characteristics. In responding to the problem that results of stock price prediction of original long short-term memory neural network model (LSTM) are not accurate enough, this study combines the various analytical indicators about stock prices when applying the structure of LSTM neural network. The indicators as KDJ, BOLL, MACD, ARBR, and CR, are used as the training set data together with the basic stock price trading data to expand the training set data volume and change the model parameters for prediction. After initially improving the prediction accuracy, the study uses Pearson correlation coefficient and Principal Component Analysis (PCA) to further improve the model, and an optimized LSTM neural network model (P-I-LSTM) was proposed to improve the accuracy while reducing the amount of data in the training set and improving the training speed.
Computer Science,Business
What problem does this paper attempt to address?