Stock price prediction based on LSTM and LightGBM hybrid model

Liwei Tian,Li Feng,Lei Yang,Yuankai Guo
DOI: https://doi.org/10.1007/s11227-022-04326-5
IF: 3.3
2022-02-18
The Journal of Supercomputing
Abstract:Finding an accurate, stable and effective model to predict the rise and fall of stocks has become a task increasingly favored by scholars. This paper proposes a long short-term memory (LSTM) network based on Pearson's correlation coefficient and a Bayesian-optimized LightGBM hybrid model, named as LSTM-BO-LightGBM, to solve the problem of stock price fluctuation prediction. The multilayer bidirectional LSTM-BO-LightGBM prediction model is compared with the LSTM-BO-XGBoost hybrid model, the LSTM-LightGBM hybrid model, the LSTM-XGBoost hybrid model, the single LSTM network model and the RNN network model. The prediction result of the LSTM-BO-LightGBM model for the "ES = F" stock is an RMSE value of 596.04, MAE value of 15.24, accuracy value of 0.639 and f1_score value of 0.799, which are improved compared with the prediction results of the other five models. At the same time, when applying the model to "YM = F", "CL = F", "^TNX", "^N225", "NQ = F", "AAPL", "GC = F", "JPY = X" and "SI = F", all of the nine stocks showed good forecasting performance. The results demonstrate that the multilayer bidirectional LSTM-BO-LightGBM model proposed in this paper has better approximation ability and generalization ability in the stock fluctuation forecast than previous models and can well fit the stock fluctuation.
What problem does this paper attempt to address?