OCET: One-Dimensional Convolution Embedding Transformer for Stock Trend Prediction
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.
First ; Corresponding
Cited Times [WOS]:0
|Document Type||Conference paper|
|Department||Department of Statistics and Data Science|
1.Department of Statistics and Data Science,Southern University of Science and Technology,Shenzhen,518055,China
2.Guangdong Provincial Key Laboratory of Brain-Inspired Intelligent Computation,Department of Computer Science and Engineering,Southern University of Science and Technology,Shenzhen,518055,China
3.Research Institute of Trustworthy Autonomous Systems,Southern University of Science and Technology,Shenzhen,518055,China
4.Shenzhen Securities Information Co.,Ltd.,Shenzhen,China
|First Author Affilication||Department of Statistics and Data Science; Department of Computer Science and Engineering|
|Corresponding Author Affilication||Department of Computer Science and Engineering; Research Institute of Trustworthy Autonomous Systems|
|First Author's First Affilication||Department of Statistics and Data Science; Department of Computer Science and Engineering|
Yang，Peng,Fu，Lang,Zhang，Jian,et al. OCET: One-Dimensional Convolution Embedding Transformer for Stock Trend Prediction[C],2023:370-384.
|Files in This Item:||There are no files associated with this item.|
|Recommend this item|
|Export to Endnote|
|Export to Excel|
|Export to Csv|
|Similar articles in Google Scholar|
|Similar articles in Baidu Scholar|
|Similar articles in Bing Scholar|
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.