Prediction model for stock price trend based on recurrent neural network

被引:41
|
作者
Zhao, Jinghua [1 ]
Zeng, Dalin [2 ]
Liang, Shuang [1 ]
Kang, Huilin [3 ]
Liu, Qinming [1 ]
机构
[1] Univ Shanghai Sci & Technol, Sch Business, Shanghai, Peoples R China
[2] Shandong Jianzhu Univ, Sch Management Engn, Jinan, Peoples R China
[3] Tongji Univ, Sch Econ & Management, Shanghai, Peoples R China
关键词
Recurrent neural network (RNN); Attention mechanism; Stock price trend prediction;
D O I
10.1007/s12652-020-02057-0
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Stock data have a long memory, that is, changes in stock prices are closely related to historical transaction data. Also, Recurrent Neural Networks have good time series feature extraction capabilities. The paper proposed prediction models based on RNN/LSTM/GRU respectively. The attention mechanism has the ability to select and focus "key information". Therefore, based on the conventional Recurrent Neural Network, this paper introduced the attention mechanism and proposed a prediction model based on AT-RNN/AT-LSTM/AT-GRU. And the paper modeled and experimented with it. The results showed that: (1) In the most basic comparison test of RNN-M, LSTM-M, and GRU-M prediction models, the GRU-M and LSTM -M was significantly better than the RNN-M and the GRU-M was slightly better than the LSTM-M; (2) The introduction of the attention mechanism layer was helpful to improve the accuracy of the stock fluctuation prediction model;(3) Deeper neural networks did not necessarily achieve better results.
引用
收藏
页码:745 / 753
页数:9
相关论文
共 50 条
  • [1] Prediction model for stock price trend based on recurrent neural network
    Jinghua Zhao
    Dalin Zeng
    Shuang Liang
    Huilin Kang
    Qinming Liu
    [J]. Journal of Ambient Intelligence and Humanized Computing, 2021, 12 : 745 - 753
  • [2] Prediction model for stock price trend based on convolution neural network
    Lin, Hongbo
    Zhao, Jinghua
    Liang, Shuang
    Kang, Huilin
    [J]. JOURNAL OF INTELLIGENT & FUZZY SYSTEMS, 2020, 39 (04) : 4999 - 5008
  • [3] Stock Trend Prediction Algorithm Based on Deep Recurrent Neural Network
    Lu, Ruochen
    Lu, Muchao
    [J]. WIRELESS COMMUNICATIONS & MOBILE COMPUTING, 2021, 2021
  • [4] Stock Price Trend Prediction Model Based on Deep Residual Network and Stock Price Graph
    Liu, Heng
    Song, Bowen
    [J]. 2018 11TH INTERNATIONAL SYMPOSIUM ON COMPUTATIONAL INTELLIGENCE AND DESIGN (ISCID), VOL 2, 2018, : 328 - 331
  • [5] Neural Network Prediction of Stock Price Trend Based on RS with Entropy Discretization
    Huang Haiqing
    Gan Xusheng
    Lei Lei
    [J]. 2017 INTERNATIONAL CONFERENCE ON MATERIALS, ENERGY, CIVIL ENGINEERING AND COMPUTER (MATECC 2017), 2017, : 183 - 189
  • [6] RECURRENT NEURAL NETWORK BASED STOCK PRICE PREDICTION USING MULTIPLE STOCK BRANDS
    Rikukawa, Shota
    Mori, Hiroki
    Harada, Taku
    [J]. INTERNATIONAL JOURNAL OF INNOVATIVE COMPUTING INFORMATION AND CONTROL, 2020, 16 (03): : 1093 - 1099
  • [7] Prediction of Stock Price Trend Based on Wavelet Neural Network and RS Attributes Reduction
    Wei Yanming
    Lou Yuanwei
    Lei Lei
    [J]. PROCEEDINGS OF THE 2017 INTERNATIONAL CONFERENCE ON EDUCATION, ECONOMICS AND MANAGEMENT RESEARCH (ICEEMR 2017), 2017, 95 : 95 - 98
  • [8] Stock Price Trend Prediction using Artificial Neural Network Techniques
    Lertyingyod, Weerachart
    Benjamas, Nunnapus
    [J]. 2016 20TH INTERNATIONAL COMPUTER SCIENCE AND ENGINEERING CONFERENCE (ICSEC), 2016,
  • [9] Prediction of Short-Term Stock Price Trend Based on Multiview RBF Neural Network
    Lv, Bailin
    Jiang, Yizhang
    [J]. COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2021, 2021
  • [10] Stock Price Trend Prediction Based on RBF Neural Network and Artificial Fish Swarm Algorithm
    Wei Yanming
    Gan Xusheng
    Lei Lei
    [J]. 2017 INTERNATIONAL CONFERENCE ON MATERIALS, ENERGY, CIVIL ENGINEERING AND COMPUTER (MATECC 2017), 2017, : 210 - 215