VGC-GAN: A multi-graph convolution adversarial network for stock price prediction

被引:17
作者
Ma, Dongbo [1 ]
Yuan, Da [1 ]
Huang, Maojun [1 ]
Dong, Ling [1 ]
机构
[1] Shandong Technol & Business Univ, Sch Comp Sci & Technol, Yantai 264005, Peoples R China
基金
中国国家自然科学基金;
关键词
Stock price prediction; Graph Convolutional Network; Adversarial training; Variational Mode Decomposition; TIME; INDEX; MODEL;
D O I
10.1016/j.eswa.2023.121204
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Not only market signals but also disturbances of related companies influence the stock volatility of a company. Currently, most approaches that utilize inter-stock correlations rely on predetermined graphs, such as industry graphs, shareholder graphs, or event graphs, necessitating supplementary financial data. Consequently, this approach often leads to limited generalization ability of the model and incomplete representation of inter -stock relationships. Thus, this study introduces VGC-GAN, a multi-graph convolutional adversarial framework, for predicting stock prices. Initially, the proposed model generates multiple correlation graphs by analyzing historical stock data, providing a comprehensive depiction of inter-stock correlations from diverse perspectives. Subsequently, a Generative Adversarial Network (GAN) framework, augmented with Mean Square Error (MSE) loss, is constructed to enhance the predictive performance of the model. The framework combines Multi-Graph Convolutional Network (Multi-GCN) and Gated Recurrent Unit (GRU) as the generator. It undergoes supervised and adversarial training with Convolutional Neural Network (CNN), facilitating the in-depth exploration of hidden correlations between stocks and time dependence of stocks. To mitigate the impact of noise, VGC-GAN uses subsequences after Variational Mode Decomposition (VMD) with optimized parameters as input to the generator. The proposed model is evaluated on several real datasets, and the experimental results confirm its effectiveness in stock price prediction.
引用
收藏
页数:14
相关论文
共 57 条
  • [1] ModAugNet: A new forecasting framework for stock market index value with an overfitting prevention LSTM module and a prediction LSTM module
    Baek, Yujin
    Kim, Ha Young
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2018, 113 : 457 - 480
  • [2] Predicting the direction of stock market prices using tree-based classifiers
    Basak, Suryoday
    Kar, Saibal
    Saha, Snehanshu
    Khaidem, Luckyson
    Dey, Sudeepa Roy
    [J]. NORTH AMERICAN JOURNAL OF ECONOMICS AND FINANCE, 2019, 47 : 552 - 567
  • [3] GENERALIZED AUTOREGRESSIVE CONDITIONAL HETEROSKEDASTICITY
    BOLLERSLEV, T
    [J]. JOURNAL OF ECONOMETRICS, 1986, 31 (03) : 307 - 327
  • [4] Cao DF, 2020, ADV NEUR IN, V33
  • [5] Challu C, 2022, Arxiv, DOI [arXiv:2201.12886, DOI 10.1016/J.APENERGY.2019.05.068]
  • [6] A novel graph convolutional feature based convolutional neural network for stock trend prediction
    Chen, Wei
    Jiang, Manrui
    Zhang, Wei-Guo
    Chen, Zhensong
    [J]. INFORMATION SCIENCES, 2021, 556 : 67 - 94
  • [7] Incorporating Corporation Relationship via Graph Convolutional Neural Networks for Stock Price Prediction
    Chen, Yingmei
    Wei, Zhongyu
    [J]. CIKM'18: PROCEEDINGS OF THE 27TH ACM INTERNATIONAL CONFERENCE ON INFORMATION AND KNOWLEDGE MANAGEMENT, 2018, : 1655 - 1658
  • [8] China?s commercial bank stock price prediction using a novel K-means-LSTM hybrid approach
    Chen, Yufeng
    Wu, Jinwang
    Wu, Zhongrui
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2022, 202
  • [9] Chung JY, 2014, Arxiv, DOI arXiv:1412.3555
  • [10] Adapting extreme value statistics to financial time series: dealing with bias and serial dependence
    de Haan, Laurens
    Mercadier, Cecile
    Zhou, Chen
    [J]. FINANCE AND STOCHASTICS, 2016, 20 (02) : 321 - 354