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 条
  • [11] Prediction model for stock price trend based on recurrent neural network
    Zhao, Jinghua
    Zeng, Dalin
    Liang, Shuang
    Kang, Huilin
    Liu, Qinming
    JOURNAL OF AMBIENT INTELLIGENCE AND HUMANIZED COMPUTING, 2021, 12 (01) : 745 - 753
  • [12] Prediction model for stock price trend based on convolution neural network
    Lin, Hongbo
    Zhao, Jinghua
    Liang, Shuang
    Kang, Huilin
    JOURNAL OF INTELLIGENT & FUZZY SYSTEMS, 2020, 39 (04) : 4999 - 5008
  • [13] Prediction model for stock price trend based on recurrent neural network
    Jinghua Zhao
    Dalin Zeng
    Shuang Liang
    Huilin Kang
    Qinming Liu
    Journal of Ambient Intelligence and Humanized Computing, 2021, 12 : 745 - 753
  • [14] Financial market trend prediction model based on LSTM neural network algorithm
    Dong, Peilin
    Wang, Xiaoyu
    Shi, Zhouhao
    JOURNAL OF COMPUTATIONAL METHODS IN SCIENCES AND ENGINEERING, 2024, 24 (02) : 745 - 755
  • [15] A graph-based convolutional neural network stock price prediction with leading indicators
    Wu, Jimmy Ming-Tai
    Li, Zhongcui
    Srivastava, Gautam
    Tasi, Meng-Hsiun
    Lin, Jerry Chun-Wei
    SOFTWARE-PRACTICE & EXPERIENCE, 2021, 51 (03): : 628 - 644
  • [16] Stock Price Movement Prediction based on Relation Type guided Graph Convolutional Network
    Peng, Hao
    Dong, Ke
    Yang, Jie
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2023, 126
  • [17] Correlation Matters: A Stock Price Predication Model Based on the Graph Convolutional Network
    Xin, Chengkun
    Han, Qian
    Pan, Gang
    ADVANCED INTELLIGENT COMPUTING TECHNOLOGY AND APPLICATIONS, PT III, ICIC 2024, 2024, 14864 : 228 - 239
  • [18] A Network Traffic Prediction Model Based on Layered Training Graph Convolutional Network
    Li, Yulian
    Su, Yang
    IEEE ACCESS, 2025, 13 : 24398 - 24410
  • [19] Short-Term Traffic Flow Prediction Based on Graph Convolutional Network Embedded LSTM
    Huang, Yanguo
    Zhang, Shuo
    Wen, Junlin
    Chen, Xinqiang
    INTERNATIONAL CONFERENCE ON TRANSPORTATION AND DEVELOPMENT 2020 - TRAFFIC AND BIKE/PEDESTRIAN OPERATIONS, 2020, : 159 - 168
  • [20] Protein Subcellular Localization Prediction Model Based on Graph Convolutional Network
    Zhang, Tianhao
    Gu, Jiawei
    Wang, Zeyu
    Wu, Chunguo
    Liang, Yanchun
    Shi, Xiaohu
    INTERDISCIPLINARY SCIENCES-COMPUTATIONAL LIFE SCIENCES, 2022, 14 (04) : 937 - 946