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
869
Time Series ForecastingWeather
MSE0.163
497
Long-term forecastingETTm1
MSE0.326
452
Traffic ForecastingMETR-LA
MAE0.369
415
Long-term time-series forecastingETTh1 (test)
MSE0.447
410
Long-term forecastingETTm2
MSE0.177
391
Long-term forecastingETTh2
MSE0.377
376
Time Series ForecastingECL
MAE0.241
345
Time Series ForecastingETTm2
MSE0.177
300
Long-term time-series forecastingWeather (test)
MSE0.261
240
Showing 10 of 70 rows

Other info

Code

Follow for update