A Novel Integrated Approach for Stock Prediction Based on Modal Decomposition Technology and Machine Learning

• Published in: IEEE Access
• Main Author: Sun Y.; Mutalib S.; Omar N.; Tian L.
• Format: Article
• Language: English
• Published: Institute of Electrical and Electronics Engineers Inc. 2024
• Online Access: https://www.scopus.com/inward/record.uri?eid=2-s2.0-85198240224&doi=10.1109%2fACCESS.2024.3425727&partnerID=40&md5=43304b6abed02aa073babe2b2165d8f1

After the COVID-19 ended, the global economy gradually recovered. Due to the nonlinearity, complexity, and high noise of financial time series, stock price prediction has become one of the most challenging tasks in the stock market. To tackle this challenge and enhance the prediction performance in the complicated stock markets, we propose a novel integrated approach based on Complete Ensemble Empirical Mode Decomposition with Adaptive Noise (CEEMDAN), Long Short-Term Memory (LSTM), and ensemble learning algorithm LightGBM to simultaneously improve the fitting and accuracy of stock price prediction. In addition, to prevent overfitting and improve predictive performance, this study adopted the Simulated Annealing (SA) algorithm for optimization. The predictive performance of the proposed hybrid model is comprehensively evaluated by comparing it with single LSTM, RNN, and other popular hybrid models. Three evaluation metrics, namely Root Mean Square Error (RMSE), Mean Absolute Error (MAE), and accuracy, are used to compare the aforementioned models. The experimental results indicate that the proposed hybrid CEEMDAN-LSTM-SA-LightGBM model outperforms all other comparative models in this study with better fitting and accuracy. © 2024 The Authors.