Stock Trend Prediction: an Effective Hybrid Deep Model Based on Lead and Lag Correlation Graphs

Zhichao Li,Xianghui Yuan,Liwei Jin,Chencheng Zhao
DOI: https://doi.org/10.1109/cscwd61410.2024.10580251
2024-01-01
Abstract:The recognition of the importance of incorporating relevant stock data in the prediction of stock trends has been acknowledged. Nevertheless, the challenge of effectively modeling the relation between stocks and utilizing this relation to enhance the quality of forecasts remains a significant hurdle. Considering this context, we propose a deep-learning architecture designed to predict stock trends. The proposed framework involves the construction and utilization of a lead graph and lag graph. First, the lead graph and lag graph are constructed using the Dynamic Time Warping (DTW) method to reflect the interrelation between stocks. Then, we apply the Graph Convolutional Network (GCN) to extract characteristics from the prices of each stock and the lead or lag stocks. Ultimately, the stock features obtained by GCN are utilized as input for the Gated Recurrent Unit (GRU) network to predict the future movement of the stock. We conducted experiments with real stock data, and the findings demonstrate that the method proposed in this article outperforms approaches that do not account for lead and lag stocks or only consider similar stocks. These results provide empirical evidence supporting the effectiveness of incorporating lead graphs and lag graphs in the stock prediction process.
What problem does this paper attempt to address?