A novel LASSO-ATT-LSTM model of stock price prediction based on multi-source heterogeneous data

Abstract

Stock price volatility is influenced by many factors, including unstructured data that is not easy to quantify, such as investor sentiment. Therefore, given the difficulty of quantifying investor sentiment and the complexity of stock price, the paper proposes a novel LASSO-ATT-LSTM intelligent stock price prediction system based on multi-source data. Firstly, establish a sentiment dictionary in the financial field, conduct sentiment analysis on news information and comments according to the dictionary, calculate sentiment scores, and then obtain daily investor sentiment. Secondly, the LASSO (Least absolute shrinkage and selection operator) is used to reduce the dimension of basic trading indicators, valuation indicators, and technical indicators. The processed indicators and investor sentiment are used as the input of the prediction model. Finally, the LSTM (Long short-term memory) model that introduces the attention mechanism is used for intelligent prediction. The results show that the prediction of the proposed model is close to the real stock price, MAPE, RMSE, MAE and R2 are 0.0118, 0.0685, 0.0515 and 0.8460, respectively. Compared with the existing models, LASSO-ATT-LSTM has higher accuracy and is an effective method for stock price prediction.