Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Fredformer: Frequency Debiased Transformer for Time Series Forecasting

About

The Transformer model has shown leading performance in time series forecasting. Nevertheless, in some complex scenarios, it tends to learn low-frequency features in the data and overlook high-frequency features, showing a frequency bias. This bias prevents the model from accurately capturing important high-frequency data features. In this paper, we undertook empirical analyses to understand this bias and discovered that frequency bias results from the model disproportionately focusing on frequency features with higher energy. Based on our analysis, we formulate this bias and propose Fredformer, a Transformer-based framework designed to mitigate frequency bias by learning features equally across different frequency bands. This approach prevents the model from overlooking lower amplitude features important for accurate forecasting. Extensive experiments show the effectiveness of our proposed approach, which can outperform other baselines in different real-world time-series datasets. Furthermore, we introduce a lightweight variant of the Fredformer with an attention matrix approximation, which achieves comparable performance but with much fewer parameters and lower computation costs. The code is available at: https://github.com/chenzRG/Fredformer

Xihao Piao, Zheng Chen, Taichi Murayama, Yasuko Matsubara, Yasushi Sakurai• 2024

Related benchmarks

TaskDatasetResultRank
Time Series ForecastingETTh1
MSE0.373
836
Time Series ForecastingWeather
MSE0.163
497
Long-term forecastingETTm1
MSE0.326
422
Long-term time-series forecastingETTh1 (test)
MSE0.447
410
Long-term forecastingETTm2
MSE0.177
350
Traffic ForecastingMETR-LA
MAE0.369
329
Long-term forecastingETTh2
MSE0.377
310
Time Series ForecastingETTm2
MSE0.177
300
Time Series ForecastingECL
MSE0.147
294
Time Series ForecastingPeMS08
MSE0.171
229
Showing 10 of 70 rows

Other info

Code

Follow for update