UnetTSF: A Better Performance Linear Complexity Time Series Prediction Model (2401.03001v1)
Abstract: Recently, Transformer-base models have made significant progress in the field of time series prediction which have achieved good results and become baseline models beyond Dlinear. The paper proposes an U-Net time series prediction model (UnetTSF) with linear complexity, which adopts the U-Net architecture. We are the first to use FPN technology to extract features from time series data, replacing the method of decomposing time series data into trend and seasonal terms, while designing a fusion structure suitable for time series data. After testing on 8 open-source datasets, compared to the best linear model DLiner. Out of 32 testing projects, 31 achieved the best results. The average decrease in mse is 10.1%, while the average decrease in mae is 9.1%. Compared with the complex transformer-base PatchTST, UnetTSF obtained 9 optimal results for mse and 15 optimal results for mae in 32 testing projects.
- Chen Yuzhou , Segovia Ignacio and Gel Yulia R,”Z-GCNETs: time zigzags at graph convolutional networks for time series forecasting”,International Conference on Machine Learning,2021.
- Bryan Lim and Stefan Zohren, ”Time-series forecasting with deep learning: a survey”,”Phil. Trans. R. Soc. A”,2021.
- Ronneberger Olaf , Fischer Philipp and Brox Thomas,”U-net: Convolutional networks for biomedical image segmentation”,Medical Image Computing and Computer-Assisted Intervention–MICCAI 2015: 18th International Conference,2015.