CoRA: Boosting Time Series Foundation Models for Multivariate Forecasting through Correlation-aware Adapter
About
Most existing Time Series Foundation Models (TSFMs) use channel independent modeling and focus on capturing and generalizing temporal dependencies, while neglecting the correlations among channels or overlooking the different aspects of correlations. However, these correlations play a vital role in Multivariate time series forecasting. To address this, we propose a CoRrelation-aware Adapter (CoRA), a lightweight plug-and-play method that requires only fine-tuning with TSFMs and is able to capture different types of correlations, so as to improve forecast performance. Specifically, to reduce complexity, we innovatively decompose the correlation matrix into low-rank Time-Varying and Time-Invariant components. For the Time-Varying component, we further design learnable polynomials to learn dynamic correlations by capturing trends or periodic patterns. To learn positive and negative correlations that appear only among some channels, we introduce a novel dual contrastive learning method that identifies correlations through projection layers, regulated by a Heterogeneous-Partial contrastive loss during training, without introducing additional complexity in the inference stage. Extensive experiments on 10 real-world datasets demonstrate that CoRA can improve TSFMs in multivariate forecasting performance.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Time Series Forecasting | ETTm2 (5% train) | MSE0.254 | 34 | |
| Time Series Forecasting | ETTh2 (5% train) | MSE0.34 | 34 | |
| Multivariate Time-series Forecasting | ETTh1 96 | MSE0.357 | 20 | |
| Multivariate Time-series Forecasting | ETTh1 192 | MSE0.381 | 20 | |
| Multivariate Time-series Forecasting | ETTm1 336 | MSE0.359 | 20 | |
| Multivariate Time-series Forecasting | Electricity 96 | MSE0.137 | 20 | |
| Multivariate Time-series Forecasting | Electricity 192 | MSE0.152 | 20 | |
| Multivariate Time-series Forecasting | ETTm1 96 | MSE0.297 | 20 | |
| Multivariate Time-series Forecasting | ETTm2 336 | MSE0.274 | 20 | |
| Multivariate Time-series Forecasting | ETTh2 720 | MSE0.384 | 20 |