Hierarchical Adaptive Temporal-Relational Modeling for Stock Trend Prediction

Hierarchical Adaptive Temporal-Relational Modeling for Stock Trend Prediction

Heyuan Wang, Shun Li, Tengjiao Wang, Jiayi Zheng

Proceedings of the Thirtieth International Joint Conference on Artificial Intelligence
Main Track. Pages 3691-3698. https://doi.org/10.24963/ijcai.2021/508

Stock trend prediction is a challenging task due to the non-stationary dynamics and complex market dependencies. Existing methods usually regard each stock as isolated for prediction, or simply detect their correlations based on a fixed predefined graph structure. Genuinely, stock associations stem from diverse aspects, the underlying relation signals should be implicit in comprehensive graphs. On the other hand, the RNN network is mainly used to model stock historical data, while is hard to capture fine-granular volatility patterns implied in different time spans. In this paper, we propose a novel Hierarchical Adaptive Temporal-Relational Network (HATR) to characterize and predict stock evolutions. By stacking dilated causal convolutions and gating paths, short- and long-term transition features are gradually grasped from multi-scale local compositions of stock trading sequences. Particularly, a dual attention mechanism with Hawkes process and target-specific query is proposed to detect significant temporal points and scales conditioned on individual stock traits. Furthermore, we develop a multi-graph interaction module which consolidates prior domain knowledge and data-driven adaptive learning to capture interdependencies among stocks. All components are integrated seamlessly in a unified end-to-end framework. Experiments on three real-world stock market datasets validate the effectiveness of our model.
Keywords:
Multidisciplinary Topics and Applications: Economic and Finance
Data Mining: Classification