https://doi.org/10.1140/epjds/s13688-024-00517-7
Research
A hybrid stock prediction method based on periodic/non-periodic features analyses
1
School of Economics, Zhejiang University of Technology, Hangzhou, China
2
School of Computer Science, Zhejiang University of Technology, Hangzhou, China
Received:
21
June
2023
Accepted:
19
December
2024
Published online:
3
January
2025
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.
Key words: CEEMD / Time2Vec / Transformer / Periodic/Non-Periodic Features
© The Author(s) 2025
Open Access This article is licensed under a Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International License, which permits any non-commercial use, sharing, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if you modified the licensed material. You do not have permission under this licence to share adapted material derived from this article or parts of it. The images or other third party material in this article are included in the article’s Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article’s Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by-nc-nd/4.0/.