Stock Type Prediction Model Based on Hierarchical Graph Neural Network
Stock Type Prediction Model Based on Hierarchical Graph Neural Network
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
- MHierarchical Graph Neural Network (HGNN) for multi-level feature extraction.
- MGraph convolution operation to capture relational structures among stocks.
- MTemporal attention aggregator to model time-series dependencies from historical prices.
- MConstruction of stock industry relationship graph based on sector information.
Results
- RThe HGNN model effectively integrates stock relationship data and hierarchical attributes for prediction.
- RTemporal attention aggregator successfully captures macro market states from historical price sequences.
- RGraph convolution enhances representation learning by leveraging industry relationships.
- RThe model outperforms baseline methods in stock type classification tasks (implied).
- RHierarchical structure enables modeling of both micro-level stock features and macro-level market trends.
Analyze Paper
Generate insights from "Stock Type Prediction Model Based on Hierarchical Graph Neur...".