DGDNN: decoupled graph diffusion neural network for stock movement prediction

  • Zinuo You
  • , Zijian Shi
  • , Hongbo Bo
  • , John Cartlidge
  • , Li Zhang
  • , Yan Ge

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Forecasting future stock trends remains challenging for academia and industry due to stochastic inter-stock dynamics and hierarchical intra-stock dynamics influencing stock prices. In recent years, graph neural networks have achieved remarkable performance in this problem by formulating multiple stocks as graph-structured data. However, most of these approaches rely on artificially defined factors to construct static stock graphs, which fail to capture the intrinsic interdependencies between stocks that rapidly evolve. In addition, these methods often ignore the hierarchical features of the stocks and lose distinctive information within. In this work, we propose a novel graph learning approach implemented without expert knowledge to address these issues. First, our approach automatically constructs dynamic stock graphs by entropy-driven edge generation from a signal processing perspective. Then, we further learn task-optimal dependencies between stocks via a generalized graph diffusion process on constructed stock graphs. Last, a decoupled representation learning scheme is adopted to capture distinctive hierarchical intra-stock features. Experimental results demonstrate substantial improvements over state-of-the-art baselines on real-world datasets. Moreover, the ablation study and sensitivity study further illustrate the effectiveness of the proposed method in modeling the time-evolving inter-stock and intra-stock dynamics.
Original languageEnglish
Title of host publicationProceedings of the 16th International Conference on Agents and Artificial Intelligence
Subtitle of host publicationFebruary 24-26, 2024, in Rome, Italy
EditorsAna Paula Rocha, Luc Steels, Jaap van den Herik
PublisherSciTePress
Pages431-442
Number of pages12
Volume2
ISBN (Print)9789897586804
DOIs
Publication statusPublished - 1 Oct 2024
Externally publishedYes

Publication series

NameConference Proceedings
PublisherSciTePress
ISSN (Print)2184-433X

Keywords

  • stock prediction
  • graph neural network
  • graph structure learning
  • information propagation

Fingerprint

Dive into the research topics of 'DGDNN: decoupled graph diffusion neural network for stock movement prediction'. Together they form a unique fingerprint.

Cite this