A model based LSTM and graph convolutional network for stock trend prediction

被引:0
|
作者
Ran, Xiangdong [1 ]
Shan, Zhiguang [2 ]
Fan, Yukang [3 ]
Gao, Lei [4 ]
机构
[1] Beijing Informat Technol Coll, Beijing, Peoples R China
[2] State Informat Ctr, Informatizat & Ind Res Dept, Beijing, Peoples R China
[3] NYU, Coll Arts & Sci, New York, NY USA
[4] Beijing Big Data Ctr, Stand & Safety Dept, Beijing, Peoples R China
关键词
Long short-term memory; Graph convolutional network; Stock trend prediction; Stock trading decisions; NEURAL-NETWORK;
D O I
10.7717/peerj-cs.2326
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Stock market is a complex system characterized by collective activity, where interdependencies between stocks have a significant influence on stock price trends. It is widely believed that modeling these dependencies can improve the accuracy of stock trend prediction and enable investors to earn more stable profits. However, these dependencies are not directly observable and need to be analyzed from stock data. In this paper, we propose a model based on Long short-term memory (LSTM) and graph convolutional network to capture these dependencies for stock trend prediction. Specifically, an LSTM is employed to extract the stock features, with all hidden state outputs utilized to construct the graph nodes. Subsequently, Pearson correlation coefficient is used to organize the stock features into a graph structure. Finally, a graph convolutional network is applied to extract the relevant features for accurate stock trend prediction. Experiments based on China A50 stocks demonstrate that our proposed model outperforms baseline methods in terms of prediction performance and trading backtest returns. In trading backtest, we have identified a set of effective trading strategies as part of the trading plan. Based on China A50 stocks, our proposed model shows promising results in generating desirable returns during both upward and downward channels of the stock market. The proposed model has proven beneficial for investors to seeking optimal timing and pricing when dealing with shares.
引用
收藏
页数:28
相关论文
共 50 条
  • [1] A novel graph convolutional feature based convolutional neural network for stock trend prediction
    Chen, Wei
    Jiang, Manrui
    Zhang, Wei-Guo
    Chen, Zhensong
    INFORMATION SCIENCES, 2021, 556 : 67 - 94
  • [2] A scientific research topic trend prediction model based on multi-LSTM and graph convolutional network
    Xu, Mingying
    Du, Junping
    Xue, Zhe
    Guan, Zeli
    Kou, Feifei
    Shi, Lei
    INTERNATIONAL JOURNAL OF INTELLIGENT SYSTEMS, 2022, 37 (09) : 6331 - 6353
  • [3] Stock Price Trend Prediction Model Based on Deep Residual Network and Stock Price Graph
    Liu, Heng
    Song, Bowen
    2018 11TH INTERNATIONAL SYMPOSIUM ON COMPUTATIONAL INTELLIGENCE AND DESIGN (ISCID), VOL 2, 2018, : 328 - 331
  • [4] Trend Prediction of Stock Index Based on Convolutional Neural Network
    An, Zhiqi
    Ding, Yongmei
    Wu, Qianqian
    2022 7th International Conference on Cloud Computing and Big Data Analytics, ICCCBDA 2022, 2022, : 17 - 21
  • [5] STAGE framework: A stock dynamic anomaly detection and trend prediction model based on graph attention network and sparse spatiotemporal convolutional network
    Shi, Ming
    Rasli, Roznim Mohamad
    Wang, Shir Li
    PLOS ONE, 2025, 20 (03):
  • [6] GCNET: Graph-based prediction of stock price movement using graph convolutional network
    Jafari, Alireza
    Haratizadeh, Saman
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2022, 116
  • [7] PROPOSED BAYESIAN OPTIMIZATION BASED LSTM-CNN MODEL FOR STOCK TREND PREDICTION
    Chan, Bey Kun
    Johnson, Olanrewaju Victor
    Chew, Xinying
    Khaw, Khai Wah
    Ha Lee, Ming
    Alnoor, Alhamzah
    COMPUTING AND INFORMATICS, 2024, 43 (02) : 38 - 63
  • [8] Study on the prediction of stock price based on the associated network model of LSTM
    Guangyu Ding
    Liangxi Qin
    International Journal of Machine Learning and Cybernetics, 2020, 11 : 1307 - 1317
  • [9] Study on the prediction of stock price based on the associated network model of LSTM
    Ding, Guangyu
    Qin, Liangxi
    INTERNATIONAL JOURNAL OF MACHINE LEARNING AND CYBERNETICS, 2020, 11 (06) : 1307 - 1317
  • [10] Link Prediction Model Based on Adversarial Graph Convolutional Network
    Tang C.
    Zhao J.
    Ye X.
    Yu S.
    Moshi Shibie yu Rengong Zhineng/Pattern Recognition and Artificial Intelligence, 2021, 34 (02): : 95 - 105