Research on a hybrid prediction model for stock price based on long short-term memory and variational mode decomposition
Yang Yujun,Yang Yimei,Zhou Wang
DOI: https://doi.org/10.1007/s00500-021-06122-4
IF: 3.732
2021-09-06
Soft Computing
Abstract:The stock market plays a vital role in the economic and social organization of many countries. Since stock price time series are highly noisy, nonparametric, volatility, complexity, nonlinearity, dynamics, and chaos, the stock market prediction is an important issue for investors and professional analysts. In the financial field, stock market prediction is not only an important task but also an important research topic. For different problems, researchers have proposed many prediction methods. Many papers provide strong evidence that stock prices can be predicted from past price data. In this paper, we propose a hybrid prediction model for stock price based on long short-term memory (LSTM) and variational mode decomposition (VMD). We use the variational mode decomposition method to decompose the complex time series of stock prices into several relatively flat, regular, and stable subsequences. Then, we use each subsequence to train the long- and short-term memory neural network and predict each subsequence. Finally, we merge the predicted values of several subsequences to form the predicted results of the stock price complex original time series. To verify fully the method, we selected four experimental data for testing. Compared with the prediction results of various prediction methods, the prediction accuracy of our proposed model is higher. Especially in the R2 index, the experimental effect is very good. The proposed method achieves good results of more than 0.991 on each data set. Therefore, our proposed hybrid prediction model is accurate and effective in forecasting stock prices and has practical significance and reference value.
computer science, artificial intelligence, interdisciplinary applications