Stock Type Prediction Model Based on Hierarchical Graph Neural Network

—

Stock Type Prediction Model Based on Hierarchical Graph Neural Network

Jianhua Yao
Yuxin Dong
Jiajing Wang
Bingxing Wang
Hongye Zheng
Honglin Qin
Published on 12/9/2024
Stocks
Equities
Machine learning
Deep learning
Stock picking
Factor investing

This paper introduces a novel Hierarchical Graph Neural Network (HGNN) model for stock type prediction, addressing the challenge of capturing multi-level information and relational structures in the stock market. The model constructs a stock industry relationship graph to represent inter-stock dependencies and extracts temporal information from historical price sequences using a temporal attention aggregator. A graph convolution operation is designed to model macro market states, integrating these features into a comprehensive prediction framework. The approach effectively utilizes stock relationship data and hierarchical attributes, which are often overlooked in traditional models. The results demonstrate that the HGNN model outperforms baseline methods, highlighting its ability to capture both micro-level stock characteristics and macro-level market dynamics. This work contributes to the field of quantitative finance by providing a robust method for stock type classification that leverages graph neural networks and temporal attention mechanisms.

Highlights

  • 1Proposes a Hierarchical Graph Neural Network (HGNN) model for stock type prediction that captures multi-level information and relational structures.
  • 2Constructs a stock industry relationship graph to model inter-stock dependencies.
  • 3Integrates temporal information from historical price sequences using a temporal attention aggregator.
  • 4Designs a graph convolution operation to model macro market states.
  • 5Addresses challenges of utilizing stock relationship data and hierarchical attributes in stock market analysis.

Methods

  • M
    Hierarchical Graph Neural Network (HGNN) for multi-level feature extraction.
  • M
    Graph convolution operation to capture relational structures among stocks.
  • M
    Temporal attention aggregator to model time-series dependencies from historical prices.
  • M
    Construction of stock industry relationship graph based on sector information.

Results

  • R
    The HGNN model effectively integrates stock relationship data and hierarchical attributes for prediction.
  • R
    Temporal attention aggregator successfully captures macro market states from historical price sequences.
  • R
    Graph convolution enhances representation learning by leveraging industry relationships.
  • R
    The model outperforms baseline methods in stock type classification tasks (implied).
  • R
    Hierarchical structure enables modeling of both micro-level stock features and macro-level market trends.
0/5

Analyze Paper

Generate insights from "Stock Type Prediction Model Based on Hierarchical Graph Neur...".

Suggested Actions