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).
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
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)
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)
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)
Atsalakis, G.S., Valavanis, K.P.: Surveying stock market forecasting techniques-part II: soft computing methods. Expert Syst. Appl. 36(3), 5932–5941 (2009)
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)
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)
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)
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)
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)
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)
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
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)
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)
Kingma, D.P., Ba, J.: Adam: a method for stochastic optimization. arXiv preprint arXiv:1412.6980 (2014)
LeCun, Y., Bengio, Y., et al.: Convolutional networks for images, speech, and time series. Handb. Brain Theory Neural Netw. 3361(10), 1995 (1995)
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
Liu, G., et al.: Multi-scale two-way deep neural network for stock trend prediction. In: IJCAI, pp. 4555–4561 (2020)
Luo, W., Yu, F.: Recurrent highway networks with grouped auxiliary memory. IEEE Access 7, 182037–182049 (2019)
Marquardt, D.W., Snee, R.D.: Ridge regression in practice. Am. Stat. 29(1), 3–20 (1975)
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)
Parlour, C.A., Seppi, D.J.: Limit order markets: a survey. Handb. Financ. Intermediat. Bank. 5, 63–95 (2008)
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)
Sangadiev, A., et al.: DeepFolio: convolutional neural networks for portfolios with limit order book data. arXiv preprint arXiv:2008.12152 (2020)
Schuster, M., Paliwal, K.K.: Bidirectional recurrent neural networks. IEEE Trans. Signal Process. 45(11), 2673–2681 (1997)
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)
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)
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)
Vaswani, A., et al.: Attention is all you need. In: Advances in Neural Information Processing Systems, pp. 5998–6008 (2017)
Yang, P., Yang, Q., Tang, K., Yao, X.: Parallel exploration via negatively correlated search. Front. Comp. Sci. 15(5), 1–13 (2021)
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)
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)
Zhang, Z., Zohren, S., Roberts, S.: DeepLOB: deep convolutional neural networks for limit order books. IEEE Trans. Signal Process. 67(11), 3001–3012 (2019)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
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)