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 条
  • [21] Protein Subcellular Localization Prediction Model Based on Graph Convolutional Network
    Tianhao Zhang
    Jiawei Gu
    Zeyu Wang
    Chunguo Wu
    Yanchun Liang
    Xiaohu Shi
    Interdisciplinary Sciences: Computational Life Sciences, 2022, 14 : 937 - 946
  • [22] MG-Conv: A spatiotemporal multi-graph convolutional neural network for stock market index trend prediction
    Wang, Changhai
    Liang, Hui
    Wang, Bo
    Cui, Xiaoxu
    Xu, Yuwei
    COMPUTERS & ELECTRICAL ENGINEERING, 2022, 103
  • [23] Time-weighted LSTM Model with Redefined Labeling for Stock Trend Prediction
    Zhao, Zhiyong
    Rao, Ruonan
    Tu, Shaoxiong
    Shi, Jun
    2017 IEEE 29TH INTERNATIONAL CONFERENCE ON TOOLS WITH ARTIFICIAL INTELLIGENCE (ICTAI 2017), 2017, : 1210 - 1217
  • [24] Drug-Drug Interaction Prediction Based on Knowledge Graph Embeddings and Convolutional-LSTM Network
    Karim, Md Rezaul
    Cochez, Michael
    Jares, Joao Bosco
    Uddin, Mamtaz
    Beyan, Oya
    Decker, Stefan
    ACM-BCB'19: PROCEEDINGS OF THE 10TH ACM INTERNATIONAL CONFERENCE ON BIOINFORMATICS, COMPUTATIONAL BIOLOGY AND HEALTH INFORMATICS, 2019, : 113 - 123
  • [25] An Attention-Based LSTM Model for Stock Price Trend Prediction Using Limit Order Books
    Li, Yunhao
    Li, Liuliu
    Zhao, Xudong
    Ma, Tianyi
    Zou, Ying
    Chen, Ming
    5TH ANNUAL INTERNATIONAL CONFERENCE ON INFORMATION SYSTEM AND ARTIFICIAL INTELLIGENCE (ISAI2020), 2020, 1575
  • [26] Big data based stock trend prediction using deep CNN with reinforcement-LSTM model
    Ishwarappa
    Anuradha, J.
    INTERNATIONAL JOURNAL OF SYSTEM ASSURANCE ENGINEERING AND MANAGEMENT, 2021,
  • [27] Combining graph neural network and convolutional LSTM network for multistep soil moisture spatiotemporal prediction
    Pan, Ziwei
    Xu, Lei
    Chen, Nengcheng
    JOURNAL OF HYDROLOGY, 2025, 651
  • [28] Stock linkage prediction based on optimized LSTM model
    Ma, Chi
    Liang, Yan
    Wang, Shaofan
    Lu, Shengliang
    MULTIMEDIA TOOLS AND APPLICATIONS, 2022, 81 (09) : 12599 - 12617
  • [29] A Graph Convolutional Neural Network Model for Trajectory Prediction
    Di, Zichao
    Zhou, Yue
    Chen, Kun
    Chen, Zongzhi
    THIRTEENTH INTERNATIONAL CONFERENCE ON GRAPHICS AND IMAGE PROCESSING (ICGIP 2021), 2022, 12083
  • [30] SP-LAN: A Stock Prediction Model Based on LSTM-Attention Network
    Sun, Jingyou
    Li, Dong
    Wang, Xing
    Kou, Yue
    Li, Peixuan
    Xie, Yang
    WEB AND BIG DATA. APWEB-WAIM 2022 INTERNATIONAL WORKSHOPS, KGMA 2022, SEMIBDMA 2022, DEEPLUDA 2022, 2023, 1784 : 63 - 74