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 条
  • [31] Stock linkage prediction based on optimized LSTM model
    Chi Ma
    Yan Liang
    Shaofan Wang
    Shengliang Lu
    Multimedia Tools and Applications, 2022, 81 : 12599 - 12617
  • [32] Stock volatility prediction based on convolutional neural network
    Xu, Lili
    BASIC & CLINICAL PHARMACOLOGY & TOXICOLOGY, 2021, 128 : 178 - 178
  • [33] Conceptual-temporal graph convolutional neural network model for stock price movement prediction and application
    Zhang, Fuping
    SOFT COMPUTING, 2023, 27 (10) : 6329 - 6344
  • [34] Conceptual-temporal graph convolutional neural network model for stock price movement prediction and application
    Zhang Fuping
    Soft Computing, 2023, 27 : 6329 - 6344
  • [35] Stock prediction based on random forest and LSTM neural network
    Ma, Yilin
    Han, Ruizhu
    Fu, Xiaoling
    2019 19TH INTERNATIONAL CONFERENCE ON CONTROL, AUTOMATION AND SYSTEMS (ICCAS 2019), 2019, : 126 - 130
  • [36] Traffic Flow Prediction Model Based on Attention Spatiotemporal Graph Convolutional Network
    Sun, HongXian
    2023 3rd International Symposium on Computer Technology and Information Science, ISCTIS 2023, 2023, : 148 - 153
  • [37] Pedestrian Trajectory Prediction Based on Deep Convolutional LSTM Network
    Song, Xiao
    Chen, Kai
    Li, Xu
    Sun, Jinghan
    Hou, Baocun
    Cui, Yong
    Zhang, Baochang
    Xiong, Gang
    Wang, Zilie
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2021, 22 (06) : 3285 - 3302
  • [38] Multi-Channel Temporal Graph Convolutional Network for Stock Return Prediction
    Sun, Jifeng
    Lin, Jianwu
    Zhou, Yi
    2020 IEEE 18TH INTERNATIONAL CONFERENCE ON INDUSTRIAL INFORMATICS (INDIN), VOL 1, 2020, : 423 - 428
  • [39] Graph Convolutional Network based Link State Prediction
    Yeom, Sungwoong
    Choi, Chulwoong
    Kolekar, Shivani Sanjay
    Kim, Kyungbaek
    2021 22nd Asia-Pacific Network Operations and Management Symposium, APNOMS 2021, 2021, : 246 - 249
  • [40] Graph Convolutional Network based Link State Prediction
    Yeom, Sungwoong
    Choi, Chulwoong
    Kolekar, Shivani Sanjay
    Kim, Kyungbaek
    2021 22ND ASIA-PACIFIC NETWORK OPERATIONS AND MANAGEMENT SYMPOSIUM (APNOMS), 2021, : 246 - 249