A Stock Price Prediction Model Based on Investor Sentiment and Optimized Deep Learning

• Published in: IEEE access Vol. 11; pp. 51353 - 51367
• Main Authors: Mu, Guangyu, Gao, Nan, Wang, Yuhan, Dai, Li
• Format: Journal Article
• Language: English
• Published: Piscataway IEEE 2023; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Data mining; Data models; Deep learning; Long short term memory; LSTM model; Machine learning; Performance prediction; Prediction algorithms; Prediction models; Predictive models; Prices; Pricing; Recurrent neural networks; Search algorithms; Search methods; Sentiment analysis; sentiment dictionary; sparrow search algorithm; Stock market; stock price prediction; Support vector machines; Swarm intelligence
• ISSN: 2169-3536; 2169-3536
• DOI: 10.1109/ACCESS.2023.3278790

Accurate prediction of stock prices can reduce investment risks and increase returns. This paper combines the multi-source data affecting stock prices and applies sentiment analysis, swarm intelligence algorithm, and deep learning to build the MS-SSA-LSTM model. Firstly, we crawl the East Money forum posts information to establish the unique sentiment dictionary and calculate the sentiment index. Then, the Sparrow Search Algorithm (SSA) optimizes the Long and Short-Term Memory network (LSTM) hyperparameters. Finally, the sentiment index and fundamental trading data are integrated, and LSTM is used to forecast stock prices in the future. Experiments demonstrate that the MS-SSA-LSTM model outperforms the others and has high universal applicability. Compared with standard LSTM, the <inline-formula> <tex-math notation="LaTeX">R^{2} </tex-math></inline-formula> of MS-SSA-LSTM is improved by 10.74% on average. We found that: 1) Adding the sentiment index can enhance the model's predictive performance. 2) The LSTM's hyperparameters are optimized using SSA, which objectively explains the model parameter settings and improves the prediction effect. 3) The high volatility of China's financial market is more suitable for short-term prediction.