Skip to main content

Indicator-Specific Recurrent Neural Networks with Co-teaching for Stock Trend Prediction

  • Conference paper
  • First Online:
  • 203 Accesses

Part of the book series: Lecture Notes in Computer Science ((LNCS,volume 13729))

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

Chapter
USD   29.95
Price excludes VAT (USA)
  • Available as PDF
  • Read on any device
  • Instant download
  • Own it forever
eBook
USD   44.99
Price excludes VAT (USA)
  • Available as EPUB and PDF
  • Read on any device
  • Instant download
  • Own it forever
Softcover Book
USD   59.99
Price excludes VAT (USA)
  • Compact, lightweight edition
  • Dispatched in 3 to 5 business days
  • Free shipping worldwide - see info

Tax calculation will be finalised at checkout

Purchases are for personal use only

Learn about institutional subscriptions

References

  1. 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)

    Google Scholar 

  2. 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)

    Article  MathSciNet  Google Scholar 

  3. 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)

    Google Scholar 

  4. Black, F.: Noise. J. Financ. 41(3), 528–543 (1986)

    Article  Google Scholar 

  5. Brock, W., Lakonishok, J., LeBaron, B.: Simple technical trading rules and the stochastic properties of stock returns. J. Financ. 47(5), 1731–1764 (1992)

    Article  Google Scholar 

  6. 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)

    Google Scholar 

  7. Chen, T., Chen, F.: An intelligent pattern recognition model for supporting investment decisions in stock market. Inf. Sci. 346, 261–274 (2016)

    Article  Google Scholar 

  8. 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)

    Article  MathSciNet  Google Scholar 

  9. 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)

    Google Scholar 

  10. Egeli, B., Ozturan, M., Badur, B.: Stock market prediction using artificial neural networks. Decis. Support Syst. 22, 171–185 (2003)

    Google Scholar 

  11. Fama, E.F.: Efficient capital markets: a review of theory and empirical work. J. Financ. 25(2), 383–417 (1970)

    Article  Google Scholar 

  12. Fenghua, W., Jihong, X., Zhifang, H., Xu, G.: Stock price prediction based on SSA and SVM. Procedia Comput. Sci. 31, 625–631 (2014)

    Article  Google Scholar 

  13. 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)

    Google Scholar 

  14. Hellström, T., Holmström, K.: Predicting the stock market (1998)

    Google Scholar 

  15. Hoseinzade, E., Haratizadeh, S.: CNNpred: CNN-based stock market prediction using a diverse set of variables. Expert Syst. Appl. 129, 273–285 (2019)

    Article  Google Scholar 

  16. Huang, W., Nakamori, Y., Wang, S.Y.: Forecasting stock market movement direction with support vector machine. Compute. Oper. Res. 32(10), 2513–2522 (2005)

    Article  MATH  Google Scholar 

  17. 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)

    Article  Google Scholar 

  18. Kim, S., Kang, M.: Financial series prediction using attention LSTM. arXiv preprint arXiv:1902.10877 (2019)

  19. 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)

    Google Scholar 

  20. Kingma, D.P., Ba, J.: Adam: a method for stochastic optimization. arXiv preprint arXiv:1412.6980 (2014)

  21. 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)

    Google Scholar 

  22. Lawrence, R.: Using neural networks to forecast stock market prices. University of Manitoba, vol. 333, pp. 2006–2013 (1997)

    Google Scholar 

  23. 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)

    Article  Google Scholar 

  24. Malach, E., Shalev-Shwartz, S.: Decoupling “when to update” from “how to update”. In: Advances in Neural Information Processing Systems 30 (2017)

    Google Scholar 

  25. Malkiel, B.G.: The efficient market hypothesis and its critics. J. Econ. Perspect. 17(1), 59–82 (2003)

    Article  Google Scholar 

  26. Malkiel, B.G.: A Random Walk Down Wall Street: Including a Life-Cycle Guide to Personal Investing. WW Norton & Company, New York (1999)

    Google Scholar 

  27. 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)

    Google Scholar 

  28. Ou, J.A., Penman, S.H.: Financial statement analysis and the prediction of stock returns. J. Account. Econ. 11(4), 295–329 (1989)

    Article  Google Scholar 

  29. 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)

    Article  Google Scholar 

  30. 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)

    Article  Google Scholar 

  31. 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)

    Google Scholar 

  32. Vaswani, A., et al.: Attention is all you need. In: Advances in Neural Information Processing Systems 30 (2017)

    Google Scholar 

  33. 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)

    Google Scholar 

  34. White, H.: Economic prediction using neural networks: the case of IBM daily stock returns. In: ICNN, vol. 2, pp. 451–458 (1988)

    Google Scholar 

  35. 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)

    Google Scholar 

  36. 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)

    Google Scholar 

  37. 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)

    Article  Google Scholar 

Download references

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

Authors

Corresponding author

Correspondence to Ruifeng Xu .

Editor information

Editors and Affiliations

Rights and permissions

Reprints and permissions

Copyright information

© 2022 The Author(s), under exclusive license to Springer Nature Switzerland AG

About this paper

Check for updates. Verify currency and authenticity via CrossMark

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)

Publish with us

Policies and ethics