Abstract
Stock trend prediction is a challenging problem due to the complexity of stock data. Recently, many works applied deep learning methods for stock trend prediction and achieve impressive results. However, these methods still suffer from two limitations: 1) Various types of technical indicators are input into a single model, making it difficult for the model to learn differentiated features. 2) Noisy data in the stocks is not handled effectively. Therefore, in this paper, we propose a stock trend prediction framework using indicator-specific recurrent neural networks with co-teaching. Specifically, we first collect data from Chinese stock market and divide them into fourteen categories. Then we apply multiple RNNs to extract features separately from different technical indicator categories which can learn comprehensive features. In addition, we leverage multi-head attention for effective feature interaction and fusion. At last, we utilize co-teaching method during the training process to reduce the impact of noisy data. Experimental results show both the effectiveness and superiority of our method.
This is a preview of subscription content, log in via an institution.
Buying options
Tax calculation will be finalised at checkout
Purchases are for personal use only
Learn about institutional subscriptionsReferences
Agrawal, J., Chourasia, V., Mittra, A.: State-of-the-art in stock prediction techniques. Int. J. Adv. Res. Electr. Electron. Instrum. Eng. 2(4), 1360–1366 (2013)
Ahmed, N.K., Atiya, A.F., Gayar, N.E., El-Shishiny, H.: An empirical comparison of machine learning models for time series forecasting. Econom. Rev. 29(5–6), 594–621 (2010)
Ariyo, A.A., Adewumi, A.O., Ayo, C.K.: Stock price prediction using the ARIMA model. In: 2014 UKSim-AMSS 16th International Conference on Computer Modelling and Simulation, pp. 106–112. IEEE (2014)
Black, F.: Noise. J. Financ. 41(3), 528–543 (1986)
Brock, W., Lakonishok, J., LeBaron, B.: Simple technical trading rules and the stochastic properties of stock returns. J. Financ. 47(5), 1731–1764 (1992)
Chen, K., Zhou, Y., Dai, F.: 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), pp. 2823–2824. IEEE (2015)
Chen, T., Chen, F.: An intelligent pattern recognition model for supporting investment decisions in stock market. Inf. Sci. 346, 261–274 (2016)
Chen, W., Jiang, M., Zhang, W.G., Chen, Z.: A novel graph convolutional feature based convolutional neural network for stock trend prediction. Inf. Sci. 556, 67–94 (2021)
Deng, S., Zhang, N., Zhang, W., Chen, J., Pan, J.Z., Chen, H.: Knowledge-driven stock trend prediction and explanation via temporal convolutional network. In: Companion Proceedings of the 2019 World Wide Web Conference, pp. 678–685 (2019)
Egeli, B., Ozturan, M., Badur, B.: Stock market prediction using artificial neural networks. Decis. Support Syst. 22, 171–185 (2003)
Fama, E.F.: Efficient capital markets: a review of theory and empirical work. J. Financ. 25(2), 383–417 (1970)
Fenghua, W., Jihong, X., Zhifang, H., Xu, G.: Stock price prediction based on SSA and SVM. Procedia Comput. Sci. 31, 625–631 (2014)
Han, B., et al.: Co-teaching: robust training of deep neural networks with extremely noisy labels. In: Advances in Neural Information Processing Systems 31 (2018)
Hellström, T., Holmström, K.: Predicting the stock market (1998)
Hoseinzade, E., Haratizadeh, S.: CNNpred: CNN-based stock market prediction using a diverse set of variables. Expert Syst. Appl. 129, 273–285 (2019)
Huang, W., Nakamori, Y., Wang, S.Y.: Forecasting stock market movement direction with support vector machine. Compute. Oper. Res. 32(10), 2513–2522 (2005)
Kara, Y., Boyacioglu, M.A., Baykan, Ö.K.: Predicting direction of stock price index movement using artificial neural networks and support vector machines: the sample of the Istanbul stock exchange. Expert Syst. Appl. 38(5), 5311–5319 (2011)
Kim, S., Kang, M.: Financial series prediction using attention LSTM. arXiv preprint arXiv:1902.10877 (2019)
Kimoto, T., Asakawa, K., Yoda, M., Takeoka, M.: Stock market prediction system with modular neural networks. In: 1990 IJCNN International Joint Conference on Neural Networks, pp. 1–6. IEEE (1990)
Kingma, D.P., Ba, J.: Adam: a method for stochastic optimization. arXiv preprint arXiv:1412.6980 (2014)
Kumar, I., Dogra, K., Utreja, C., Yadav, P.: A comparative study of supervised machine learning algorithms for stock market trend prediction. In: 2018 Second International Conference on Inventive Communication and Computational Technologies (ICICCT), pp. 1003–1007. IEEE (2018)
Lawrence, R.: Using neural networks to forecast stock market prices. University of Manitoba, vol. 333, pp. 2006–2013 (1997)
Lu, W., Li, J., Wang, J., Qin, L.: A CNN-BILSTM-AM method for stock price prediction. Neural Comput. Appl. 33(10), 4741–4753 (2021)
Malach, E., Shalev-Shwartz, S.: Decoupling “when to update” from “how to update”. In: Advances in Neural Information Processing Systems 30 (2017)
Malkiel, B.G.: The efficient market hypothesis and its critics. J. Econ. Perspect. 17(1), 59–82 (2003)
Malkiel, B.G.: A Random Walk Down Wall Street: Including a Life-Cycle Guide to Personal Investing. WW Norton & Company, New York (1999)
Nelson, D.M., Pereira, A.C., De Oliveira, R.A.: Stock market’s price movement prediction with LSTM neural networks. In: 2017 International Joint Conference on Neural Networks (IJCNN), pp. 1419–1426. IEEE (2017)
Ou, J.A., Penman, S.H.: Financial statement analysis and the prediction of stock returns. J. Account. Econ. 11(4), 295–329 (1989)
Rather, A.M., Agarwal, A., Sastry, V.: Recurrent neural network and a hybrid model for prediction of stock returns. Expert Syst. Appl. 42(6), 3234–3241 (2015)
Saad, E.W., Prokhorov, D.V., Wunsch, D.C.: Comparative study of stock trend prediction using time delay, recurrent and probabilistic neural networks. IEEE Trans. Neural Netw. 9(6), 1456–1470 (1998)
Trafalis, T.B., Ince, H.: Support vector machine for regression and applications to financial forecasting. In: Proceedings of the IEEE-INNS-ENNS International Joint Conference on Neural Networks. IJCNN 2000. Neural Computing: New Challenges and Perspectives for the New Millennium, vol. 6, pp. 348–353. IEEE (2000)
Vaswani, A., et al.: Attention is all you need. In: Advances in Neural Information Processing Systems 30 (2017)
Wei, H., Feng, L., Chen, X., An, B.: Combating noisy labels by agreement: a joint training method with co-regularization. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 13726–13735 (2020)
White, H.: Economic prediction using neural networks: the case of IBM daily stock returns. In: ICNN, vol. 2, pp. 451–458 (1988)
Yoo, P.D., Kim, M.H., Jan, T.: Machine learning techniques and use of event information for stock market prediction: a survey and evaluation. In: International Conference on Computational Intelligence for Modelling, Control and Automation and International Conference on Intelligent Agents, Web Technologies and Internet Commerce (CIMCA-IAWTIC 2006), vol. 2, pp. 835–841. IEEE (2005)
Yu, X., Han, B., Yao, J., Niu, G., Tsang, I., Sugiyama, M.: How does disagreement help generalization against label corruption? In: International Conference on Machine Learning, pp. 7164–7173. PMLR (2019)
Zhang, X., Li, A., Pan, R.: Stock trend prediction based on a new status box method and AdaBoost probabilistic support vector machine. Appl. Soft Comput. 49, 385–398 (2016)
Acknowledgments
This work was partially supported by the National Natural Science Foundation of China (62006062, 62176076), Shenzhen Foundational Research Funding JCYJ20200109113441941, Shenzhen Key Technology Project JSGG20210802154400001 and Joint Lab of HITSZ and China Merchants Securities.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2022 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Xu, H. et al. (2022). Indicator-Specific Recurrent Neural Networks with Co-teaching for Stock Trend Prediction. In: Pan, X., Jin, T., Zhang, LJ. (eds) Artificial Intelligence and Mobile Services – AIMS 2022. AIMS 2022. Lecture Notes in Computer Science, vol 13729. Springer, Cham. https://doi.org/10.1007/978-3-031-23504-7_6
Download citation
DOI: https://doi.org/10.1007/978-3-031-23504-7_6
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-23503-0
Online ISBN: 978-3-031-23504-7
eBook Packages: Computer ScienceComputer Science (R0)