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 条
  • [41] Social graph convolutional LSTM for pedestrian trajectory prediction
    Zhou, Yutao
    Wu, Huayi
    Cheng, Hongquan
    Qi, Kunlun
    Hu, Kai
    Kang, Chaogui
    Zheng, Jie
    IET INTELLIGENT TRANSPORT SYSTEMS, 2021, 15 (03) : 396 - 405
  • [42] Canonical Correlation Analysis based Bi-Graph Convolutional Network for Stock Price Movement Prediction
    Zhang, Kexin
    Cai, Jia
    2022 INTERNATIONAL CONFERENCE ON IMAGE PROCESSING, COMPUTER VISION AND MACHINE LEARNING (ICICML), 2022, : 538 - 542
  • [43] Knowledge-Driven Stock Trend Prediction and Explanation via Temporal Convolutional Network
    Deng, Shumin
    Zhang, Ningyu
    Zhang, Wen
    Chen, Jiaoyan
    Pan, Jeff Z.
    Chen, Huajun
    COMPANION OF THE WORLD WIDE WEB CONFERENCE (WWW 2019 ), 2019, : 678 - 685
  • [44] GCN-LSTM: A hybrid graph convolutional network model for schizophrenia classification
    Gosala, Bethany
    Singh, Avnish Ramvinay
    Tiwari, Himanshu
    Gupta, Manjari
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2025, 105
  • [45] GRAPH CONVOLUTIONAL LSTM MODEL FOR SKELETON-BASED ACTION RECOGNITION
    Zhang, Han
    Song, Yonghong
    Zhang, Yuanlin
    2019 IEEE INTERNATIONAL CONFERENCE ON MULTIMEDIA AND EXPO (ICME), 2019, : 412 - 417
  • [46] A Spatiotemporal Model for Global Earthquake Prediction Based on Convolutional LSTM
    Zhang, Zhongchang
    Wang, Yubing
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [47] Stock Price Prediction Based on FinBERT-LSTM Model
    Fan, Shijia
    Chen, Xu
    Wang, Xu-an
    COMPLEX, INTELLIGENT AND SOFTWARE INTENSIVE SYSTEMS, CISIS-2024, 2024, 87 : 37 - 47
  • [48] Particle swarm optimization LSTM based stock prediction model
    Yuan, Xueyu
    He, Chun
    Xu, Heng
    Sun, Yuyang
    2023 3RD ASIA-PACIFIC CONFERENCE ON COMMUNICATIONS TECHNOLOGY AND COMPUTER SCIENCE, ACCTCS, 2023, : 513 - 516
  • [49] Stock price prediction based on LSTM and LightGBM hybrid model
    Tian, Liwei
    Feng, Li
    Yang, Lei
    Guo, Yuankai
    JOURNAL OF SUPERCOMPUTING, 2022, 78 (09): : 11768 - 11793
  • [50] A Global Earthquake Prediction Model Based on Spherical Convolutional LSTM
    Zhang, Zhongchang
    Wang, Yubing
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62 : 1 - 10