Stock investment is an economic activity characterized by high risks and high returns. Therefore, the prediction of stock prices or fluctuations is of great importance to investors. Stock price prediction is a challenging task due to the nonlinearity and high volatility of stock time series. Existing deep learning models may not capture the periodic and non-periodic features of stock data effectively. In this paper, we propose a novel model that leverages Complete Ensemble Empirical Mode Decomposition (CEEMD), Time2Vec, and Transformer to better capture and utilize various patterns in stock data for enhanced prediction performance, and we call it ETT. CEEMD decomposes the stock data into different frequency components based on their intrinsic scales. Time2Vec provides a time vector representation that captures both periodic and non-periodic patterns while being invariant to time scaling. Transformer learns the long-term dependencies and global information from the data. We apply ETT to predict stock prices in the Chinese A-share market and compare it with several baseline models. The results show that ETT reduces the mean squared error (MSE) by an average of 4% and increases the average cumulative return by 58% on the CSI 100 and Hushen 300 datasets.
Read full abstract