JACIII Vol.26 No.4 pp. 600-608
doi: 10.20965/jaciii.2022.p0600


CNN-GRUA-FC Stock Price Forecast Model Based on Multi-Factor Analysis

Shuying Yang, Haiming Guo, and Junguang Li

School of Computer Science and Engineering, Tianjin University of Technology
No.391 Bin Shui Xi Dao Road, Xiqing District, Tianjin 300384, China

Corresponding author

September 7, 2020
May 1, 2022
July 20, 2022
stock price prediction, CNN, GRU, attention mechanism, random forest

To predict stock prices, this paper proposes a CNN-GRUA-FC model based on multi-factor analysis for time series forecasting. First, we use the random forest algorithm to evaluate the importance of the factor series, selecting the factor series of greater importance as the input of the subsequent prediction model. We then use the convolutional neural network (CNN) to extract the spatial characteristics of stock data for prediction, taking advantage of the gated recurrent unit (GRU) neural network to extract the dynamic characteristics of stock data and make prediction. Additionally, we combine an attention mechanism with the GRU neural network (GRUA) to improve its prediction performance. Finally, the prediction results of the two different sub-models are passed through the fully connected (FC) layer to obtain the final predictions. The results show that the prediction accuracy of the CNN-GRUA-FC prediction model proposed in this paper is higher than that of other models.

Cite this article as:
S. Yang, H. Guo, and J. Li, “CNN-GRUA-FC Stock Price Forecast Model Based on Multi-Factor Analysis,” J. Adv. Comput. Intell. Intell. Inform., Vol.26 No.4, pp. 600-608, 2022.
Data files:
  1. [1] A. A. Ariyo, A. O. Adewumi, and C. K. Ayo, “Stock Price Prediction Using the ARIMA Model,” 2014 UKSim-AMSS 16th Int. Conf. on Computer Modelling and Simulation, pp. 106-112, doi: 10.1109/UKSim.2014.67, 2014.
  2. [2] W. Lina and X. Dongrong, “Analysis of Non-Steady Time-Series Forecast for Economy Based on ARMA Model,” J. of Wuhan University of Technology (Transportation Science and Engineering Edition), No.01, pp. 133-136, 2004 (in Chinese).
  3. [3] M. Zhou, Y. Nie, G. Li et al., “Wavelet analysis based ARIMA hourly electricity prices forecasting approach,” Power System Technology, No.09, pp. 50-55, 2005 (in Chinese).
  4. [4] M. Khashei, M. Bijari, and G. A. R. Ardali, “Improvement of Auto-Regressive Integrated Moving Average Models Using Fuzzy Logic and Artificial Neural Networks (ANNs),” Neurocomputing, Vol.72, Issue 4-6, pp. 956-967, 2009.
  5. [5] L. F. S. Vilela, R. C. Leme, C. A. M. Pinheiro et al., “Forecasting financial series using clustering methods and support vector regression,” Artificial Intelligence Review, Vol.52, No.2, pp. 743-773, 2019.
  6. [6] H. Liu and Y. Hou, “Application of Bayesian Neural Network in Prediction of Stock Time Series,” Computer Engineering and Applications, Vol.55, No.2, pp. 225-229+244, 2019 (in Chinese).
  7. [7] M. Qiu, Y. Song, and F. Akagi, “Application of artificial neural network for the prediction of stock market returns: The case of the Japanese stock market,” Chaos, Solitons & Fractals, Vol.85, pp. 1-7, 2016.
  8. [8] S. Siami-Namini, N. Tavakoli, and A. Siami Namin, “A Comparison of ARIMA and LSTM in Forecasting Time Series,” 2018 17th IEEE Int. Conf. on Machine Learning and Applications (ICMLA), pp. 1394-1401, doi: 10.1109/ICMLA.2018.00227, 2018.
  9. [9] F. Shen, J. Chao, and J. Zhao, “Forecasting exchange rate using deep belief networks and conjugate gradient method,” Neurocomputing, Vol.167, pp. 243-253, 2015.
  10. [10] J. Zhao, D. Zeng, S. Liang et al., “Prediction model for stock price trend based on recurrent neural network,” J. Ambient Intell. Human. Comput., Vol.12, pp. 745-753, 2021.
  11. [11] A. Tsantekidis, N. Passalis, A. Tefas et al., “Forecasting Stock Prices from the Limit Order Book Using Convolutional Neural Networks,” 2017 IEEE 19th Conf. on Business Informatics (CBI), pp. 7-12, doi: 10.1109/CBI.2017.23, 2017.
  12. [12] M. Hiransha, E. A. Gopalakrishnan, V. K. Menon et al., “NSE Stock Market Prediction Using Deep-Learning Models,” Procedia Computer Science, Vol.132, pp. 1351-1362, 2018.
  13. [13] G. P. Zhang, “Time series forecasting using a hybrid ARIMA and neural network model,” Neurocomputing, Vol.50, pp. 159-175, 2003.
  14. [14] Z. Xu, J. Zhang, J. Wang et al., “Prediction research of financial time series based on deep learning,” Soft Comput., Vol.24, pp. 8295-8312, 2020.
  15. [15] S. Jain, R. Gupta, and A. A. Moghe, “Stock Price Prediction on Daily Stock Data Using Deep Neural Networks,” 2018 Int. Conf. on Advanced Computation and Telecommunication (ICACAT), doi: 10.1109/ICACAT.2018.8933791, 2018.
  16. [16] L. Hang, “Statistical Learning Methods,” Tsinghua University Press, 2012.
  17. [17] C. Dewi and R.-C. Chen, “Random forest and support vector machine on features selection for regression analysis,” Int. J. of Innovative Computing, Information and Control, Vol.15, No.6, pp. 2027-2037, 2019.
  18. [18] S. Hochreiter and J. Schmidhuber, “Long short-term memory,” Neural Computation, Vol.9, No.8, pp. 1735-1780, 1997.
  19. [19] K. Cho, B. v. Merriënboer, C. Gulcehre et al., “Learning Phrase Representations Using RNN Encoder-Decoder for Statistical Machine Translation,” 2014 Conf. on Empirical Methods in Natural Language Processing (EMNLP), pp. 1724-1734, 2014.
  20. [20] J. Chung, C. Gulcehre, K. Cho et al., “Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modeling,” NIPS 2014 Workshop on Deep Learning, 2014.

*This site is desgined based on HTML5 and CSS3 for modern browsers, e.g. Chrome, Firefox, Safari, Edge, Opera.

Last updated on Jul. 12, 2024