Prediction model for stock price trend based on recurrent neural network

摘要

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.