Hierarchical Node Representation Learning for Stock Prediction.

Zhihan Yue,Ying Tan
DOI: https://doi.org/10.1007/978-3-031-36625-3_37
2023-01-01
Abstract:The stock market is a highly complex and dynamic system, where relationships between stocks play a critical role in predicting price movements. To capture these relationships, we propose a novel approach called the hierarchical predictive representation (HPR). The pairwise attention network is first employed to identify effective relationships between stocks. Then, the hierarchical node matching identifies the most predictive relationship subset at various hierarchical levels. By concatenating representations from various levels, our method achieves a comprehensive representation that reflects local to global information. We further introduce a representation ensemble mechanism to leverage multiple relationships, enhancing the model's predictive performance. Extensive experiments on various datasets demonstrate the superiority of HPR compared to existing state-of-the-art methods.
What problem does this paper attempt to address?