MDGNN: Multi-Relational Dynamic Graph Neural Network for Comprehensive and Dynamic Stock Investment Prediction

Authors

  • Hao Qian Ant Group, Hangzhou, China
  • Hongting Zhou Ant Group, Hangzhou, China
  • Qian Zhao Ant Group, Hangzhou, China
  • Hao Chen Ant Group, Hangzhou, China
  • Hongxiang Yao Alibaba Group, Hangzhou, China
  • Jingwei Wang Ant Group, Hangzhou, China
  • Ziqi Liu Ant Group, Hangzhou, China
  • Fei Yu Ant Group, Hangzhou, China
  • Zhiqiang Zhang Ant Group, Hangzhou, China
  • Jun Zhou Ant Group, Hangzhou, China

DOI:

https://doi.org/10.1609/aaai.v38i13.29381

Keywords:

ML: Classification and Regression, ML: Time-Series/Data Streams

Abstract

The stock market is a crucial component of the financial system, but predicting the movement of stock prices is challenging due to the dynamic and intricate relations arising from various aspects such as economic indicators, financial reports, global news, and investor sentiment. Traditional sequential methods and graph-based models have been applied in stock movement prediction, but they have limitations in capturing the multifaceted and temporal influences in stock price movements. To address these challenges, the Multi-relational Dynamic Graph Neural Network (MDGNN) framework is proposed, which utilizes a discrete dynamic graph to comprehensively capture multifaceted relations among stocks and their evolution over time. The representation generated from the graph offers a complete perspective on the interrelationships among stocks and associated entities. Additionally, the power of the Transformer structure is leveraged to encode the temporal evolution of multiplex relations, providing a dynamic and effective approach to predicting stock investment. Further, our proposed MDGNN framework achieves the best performance in public datasets compared with the state-of-the-art stock investment methods.

Published

2024-03-24

How to Cite

Qian, H., Zhou, H., Zhao, Q., Chen, H., Yao, H., Wang, J., Liu, Z., Yu, F., Zhang, Z., & Zhou, J. (2024). MDGNN: Multi-Relational Dynamic Graph Neural Network for Comprehensive and Dynamic Stock Investment Prediction. Proceedings of the AAAI Conference on Artificial Intelligence, 38(13), 14642-14650. https://doi.org/10.1609/aaai.v38i13.29381

Issue

Section

AAAI Technical Track on Machine Learning IV