Abstract
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.
Similar content being viewed by others
References
Chang PC, Liu CH (2008) A TSK type fuzzy rule based system for stock price prediction. Expert Syst Appl 34(1):135–144
Chen K, Zhou Y, Dai F (2015) A LSTM-based method for stock returns prediction: a case study of China stock market. In: 2015 IEEE international conference on big data (Big Data). IEEE, 2823–2824.
Hochreiter S, Schmidhuber J (1997) Long short-term memory. Neural Comput 9(8):1735–1780
Kim K, Han I (2000) Genetic algorithms approach to feature discretization in artificial neural networks for the prediction of stock price index. Expert Syst Appl 19(2):125–132
Kimoto T, Asakawa K, Yoda M, et al. (1990) Stock market prediction system with modular neural networks. In: 1990 IJCNN international joint conference on neural networks. IEEE, 1–6.
Nair BB, Mohandas VP, Sakthivel NR (2010) A decision tree—rough set hybrid system for stock market trend prediction[J]. Int J Comput Appl 6(9):1–6
Nelson DMQ, Pereira ACM, de Oliveira RA (2017) Stock market's price movement prediction with LSTM neural networks. In: 2017 international joint conference on neural networks (IJCNN). IEEE, 1419–1426.
Sun Ruiqi. Research on the US Stock Index Price Trend Prediction Model Based on LSTM Neural Network [D].
Sokolova M, Japkowicz N, Szpakowicz S (2006) Beyond accuracy, F-score and ROC: a family of discriminant measures for performance evaluation. In: Australasian joint conference on artificial intelligence. Springer, Berlin, Heidelberg, 1015–1021.
Treisman A (1998) Feature binding, attention and object perception. Philosophical Trans R Soc Lond Ser B 353(1373):1295–1306
White H (1988) Economic prediction using neural networks: the case of IBM daily stock returns. Proc IEEE Int Conf Neural Netw. v2(2) 451–458
Yang H, Chan L, King I (2002) Support vector machine regression for volatile stock market prediction. In: International conference on intelligent data engineering and automated learning. Springer, Berlin, Heidelberg, 391–396.
Acknowledgements
This work was supported by National Natural Science Fund (71672128), National Key Research and Development Program of China (2018YFC0830400), Shanghai Natural Science Foundation (19ZR1435600), and Natural Science Fostering Foundation (1F-19-303-001).
Author information
Authors and Affiliations
Corresponding author
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
Zhao, J., Zeng, D., Liang, S. et al. Prediction model for stock price trend based on recurrent neural network. J Ambient Intell Human Comput 12, 745–753 (2021). https://doi.org/10.1007/s12652-020-02057-0
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s12652-020-02057-0