Stock Index Prediction Based on Multi-Time Scale Learning with Multi-Graph Attention Networks

Liu Yuxia,Zhang Qi,Chu Tianguang
DOI: https://doi.org/10.1007/s10489-022-04285-7
IF: 5.3
2022-01-01
Applied Intelligence
Abstract:We present a stock index prediction model based on a multi-time scale learning (MTSL) and the multi-graph attention network (MGAT) approach. Instead of dealing with individual stock markets, we consider a group of stock markets simultaneously and exploit the effects of their interactions when making predictions. Our model consists of a boosting Hodrick-Prescott (bHP) filter and MGATs, along with MTSL processes. The bHP filter decomposes the stock index series into the slow-varying growth trend and the fast-varying cyclical volatility for training the MGAT to facilitate the learning of multi-time scale data. The MGAT exploits the interplays of stock markets with three different types of graphs: a regionality graph that qualitatively describes the linkages among stock markets within the same region or similar financial systems; a correlativity graph that quantifies the Pearson correlation between stock markets; and a causality graph that measures the convergence cross mapping (CCM) causality between stock markets. Particularly, the last graph is essentially a directed graph, which captures the nonreciprocal relationship between different stock markets. Experimental results on real stock indices reveal the effectiveness and merit of the proposed model in comparison with other models adopted in this paper.
What problem does this paper attempt to address?