Skip to main content

OCET: One-Dimensional Convolution Embedding Transformer for Stock Trend Prediction

  • Conference paper
  • First Online:
Bio-Inspired Computing: Theories and Applications (BIC-TA 2022)

Part of the book series: Communications in Computer and Information Science ((CCIS,volume 1801))

  • 708 Accesses

Abstract

Due to the strong data fitting ability of deep learning, the use of deep learning for quantitative trading has gradually sprung up in recent years. As a classical problem of quantitative trading, Stock Trend Prediction (STP) mainly predicts the movement of stock price in the future through the historical price information to better guide quantitative trading. In recent years, some deep learning work has made great progress in STP by effectively grasping long-term timing information. However, as a kind of real-time series data, short-term timing information is also very important, because stock trading is high-frequency and price fluctuates violently. And with the popularity of Transformer, there is a lack of an effective combination of feature extraction and Transformer in STP tasks. To make better use of short term information, we propose One-dimensional Convolution Embedding (OCE). Simultaneously, we introduce effective feature extraction with Transformer into STP problem to extract feature information and capture long-term timing information. By combining OCE and Transformer organically, we propose a noval STP prediction model, One-dimensional Convolution Embedding Transformer (OCET), to capture long-term and short-term time series information. Finally, OCET achieves a highest accuracy up to 0.927 in public benchmark FI-2010 When reasoning speed is twice that of SOTA models and a highest accuracy of 0.426 in HKGSAS-2020. Empirical results on these two datasets show that our OCET is significantly superior than other algorithms in STP tasks. Code are available at https://github.com/langgege-cqu/OCET.

This work was supported partly by the National Natural Science Foundation of China (Grants 62272210), partly by the Guangdong Provincial Key Laboratory (Grant 2020B121201001), partly by the Program for Guangdong Introducing Innovative and Entrepreneurial Teams (Grant 2017ZT07X386), and partly by the Stable Support Plan Program of Shenzhen Natural Science Fund (Grant 20200925154942002).

This is a preview of subscription content, log in via an institution to check access.

Access this chapter

Chapter
USD 29.95
Price excludes VAT (USA)
  • Available as PDF
  • Read on any device
  • Instant download
  • Own it forever
eBook
USD 99.00
Price excludes VAT (USA)
  • Available as EPUB and PDF
  • Read on any device
  • Instant download
  • Own it forever
Softcover Book
USD 129.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

Institutional subscriptions

References

  1. Ahn, H.J., Cai, J., Hamao, Y., Ho, R.Y.: The components of the bid-ask spread in a limit-order market: evidence from the Tokyo stock exchange. J. Empir. Financ. 9(4), 399–430 (2002)

    Article  Google Scholar 

  2. Aitken, M.J., Berkman, H., Mak, D.: The use of undisclosed limit orders on the Australian stock exchange. J. Bank. Financ. 25(8), 1589–1603 (2001)

    Article  Google Scholar 

  3. Anagnostidis, P., Papachristou, G., Thomaidis, N.S.: Liquidity commonality in order-driven trading: evidence from the Athens stock exchange. Appl. Econ. 48(22), 2007–2021 (2016)

    Article  Google Scholar 

  4. Atsalakis, G.S., Valavanis, K.P.: Surveying stock market forecasting techniques-part II: soft computing methods. Expert Syst. Appl. 36(3), 5932–5941 (2009)

    Article  Google Scholar 

  5. Cao, Q., Leggio, K.B., Schniederjans, M.J.: A comparison between Fama and French’s model and artificial neural networks in predicting the Chinese stock market. Comput. Oper. Res. 32(10), 2499–2512 (2005)

    Article  MATH  Google Scholar 

  6. Cavalcante, R.C., Brasileiro, R.C., Souza, V.L., Nobrega, J.P., Oliveira, A.L.: Computational intelligence and financial markets: a survey and future directions. Expert Syst. Appl. 55, 194–211 (2016)

    Article  Google Scholar 

  7. Devlin, J., Chang, M.W., Lee, K., Toutanova, K.: BERT: pre-training of deep bidirectional transformers for language understanding. arXiv preprint arXiv:1810.04805 (2018)

  8. Gould, M.D., Porter, M.A., Williams, S., McDonald, M., Fenn, D.J., Howison, S.D.: Limit order books. Quantit. Financ. 13(11), 1709–1742 (2013)

    Article  MathSciNet  MATH  Google Scholar 

  9. Greff, K., Srivastava, R.K., Koutník, J., Steunebrink, B.R., Schmidhuber, J.: LSTM: a search space odyssey. IEEE Trans. Neural Netw. Learn. Syst. 28(10), 2222–2232 (2016)

    Article  MathSciNet  Google Scholar 

  10. He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 770–778 (2016)

    Google Scholar 

  11. Hong, W., Li, G., Liu, S., Yang, P., Tang, K.: Multi-objective evolutionary optimization for hardware-aware neural network pruning. Fundam. Res. (2022). https://doi.org/10.1016/j.fmre.2022.07.013

  12. Hu, J., Shen, L., Sun, G.: Squeeze-and-excitation networks. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 7132–7141 (2018)

    Google Scholar 

  13. Huang, C., Ge, W., Chou, H., Du, X.: Benchmark dataset for short-term market prediction of limit order book in china markets. J. Financ. Data Sci. 3(4), 171–183 (2021)

    Article  Google Scholar 

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

  15. LeCun, Y., Bengio, Y., et al.: Convolutional networks for images, speech, and time series. Handb. Brain Theory Neural Netw. 3361(10), 1995 (1995)

    Google Scholar 

  16. Li, G., Yang, P., Qian, C., Hong, R., Tang, K.: Stage-wise magnitude-based pruning for recurrent neural networks. IEEE Trans. Neural Netw. Learn. Syst. (2022). https://doi.org/10.1109/TNNLS.2022.3184730

  17. Liu, G., et al.: Multi-scale two-way deep neural network for stock trend prediction. In: IJCAI, pp. 4555–4561 (2020)

    Google Scholar 

  18. Luo, W., Yu, F.: Recurrent highway networks with grouped auxiliary memory. IEEE Access 7, 182037–182049 (2019)

    Article  Google Scholar 

  19. Marquardt, D.W., Snee, R.D.: Ridge regression in practice. Am. Stat. 29(1), 3–20 (1975)

    MATH  Google Scholar 

  20. Ntakaris, A., Magris, M., Kanniainen, J., Gabbouj, M., Iosifidis, A.: Benchmark dataset for mid-price forecasting of limit order book data with machine learning methods. J. Forecast. 37(8), 852–866 (2018)

    Article  MathSciNet  Google Scholar 

  21. Parlour, C.A., Seppi, D.J.: Limit order markets: a survey. Handb. Financ. Intermediat. Bank. 5, 63–95 (2008)

    Google Scholar 

  22. Qilong, W., Banggu, W., Pengfei, Z., Peihua, L., Wangmeng, Z., Qinghua, H.: ECA-net: efficient channel attention for deep convolutional neural networks 2020 ieee. In: CVF Conference on Computer Vision and Pattern Recognition (CVPR) (2020)

    Google Scholar 

  23. Sangadiev, A., et al.: DeepFolio: convolutional neural networks for portfolios with limit order book data. arXiv preprint arXiv:2008.12152 (2020)

  24. Schuster, M., Paliwal, K.K.: Bidirectional recurrent neural networks. IEEE Trans. Signal Process. 45(11), 2673–2681 (1997)

    Article  Google Scholar 

  25. Szegedy, C., Vanhoucke, V., Ioffe, S., Shlens, J., Wojna, Z.: Rethinking the inception architecture for computer vision. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2818–2826 (2016)

    Google Scholar 

  26. Tran, D.T., Iosifidis, A., Kanniainen, J., Gabbouj, M.: Temporal attention-augmented bilinear network for financial time-series data analysis. IEEE Trans. Neural Netw. Learn. Syst. 30(5), 1407–1418 (2018)

    Article  Google Scholar 

  27. Tsantekidis, A., Passalis, N., Tefas, A., Kanniainen, J., Gabbouj, M., Iosifidis, A.: Forecasting stock prices from the limit order book using convolutional neural networks. In: 2017 IEEE 19th Conference on Business Informatics (CBI), vol. 1, pp. 7–12. IEEE (2017)

    Google Scholar 

  28. Vaswani, A., et al.: Attention is all you need. In: Advances in Neural Information Processing Systems, pp. 5998–6008 (2017)

    Google Scholar 

  29. Yang, P., Yang, Q., Tang, K., Yao, X.: Parallel exploration via negatively correlated search. Front. Comp. Sci. 15(5), 1–13 (2021)

    Google Scholar 

  30. Yang, P., Zhang, H., Yu, Y., Li, M., Tang, K.: Evolutionary reinforcement learning via cooperative coevolutionary negatively correlated search. Swarm Evol. Comput. 68, 100974 (2022)

    Google Scholar 

  31. Zhang, Z., Zohren, S.: Multi-horizon forecasting for limit order books: novel deep learning approaches and hardware acceleration using intelligent processing units. arXiv preprint arXiv:2105.10430 (2021)

  32. Zhang, Z., Zohren, S., Roberts, S.: DeepLOB: deep convolutional neural networks for limit order books. IEEE Trans. Signal Process. 67(11), 3001–3012 (2019)

    Article  MATH  Google Scholar 

Download references

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Guiying Li .

Editor information

Editors and Affiliations

Rights and permissions

Reprints and permissions

Copyright information

© 2023 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.

About this paper

Check for updates. Verify currency and authenticity via CrossMark

Cite this paper

Yang, P., Fu, L., Zhang, J., Li, G. (2023). OCET: One-Dimensional Convolution Embedding Transformer for Stock Trend Prediction. In: Pan, L., Zhao, D., Li, L., Lin, J. (eds) Bio-Inspired Computing: Theories and Applications. BIC-TA 2022. Communications in Computer and Information Science, vol 1801. Springer, Singapore. https://doi.org/10.1007/978-981-99-1549-1_29

Download citation

  • DOI: https://doi.org/10.1007/978-981-99-1549-1_29

  • Published:

  • Publisher Name: Springer, Singapore

  • Print ISBN: 978-981-99-1548-4

  • Online ISBN: 978-981-99-1549-1

  • eBook Packages: Computer ScienceComputer Science (R0)

Publish with us

Policies and ethics