Baguan-TS: A Sequence-Native In-Context Learning Model for Time Series Forecasting with Covariates
About
Transformers enable in-context learning (ICL) for rapid, gradient-free adaptation in time series forecasting, yet most ICL-style approaches rely on tabularized, hand-crafted features, while end-to-end sequence models lack inference-time adaptation. We bridge this gap with a unified framework, Baguan-TS, which integrates the raw-sequence representation learning with ICL, instantiated by a 3D Transformer that attends jointly over temporal, variable, and context axes. To make this high-capacity model practical, we tackle two key hurdles: (i) calibration and training stability, improved with a feature-agnostic, target-space retrieval-based local calibration; and (ii) output oversmoothing, mitigated via context-overfitting strategy. On public benchmark with covariates, Baguan-TS consistently outperforms established baselines, achieving the highest win rate and significant reductions in both point and probabilistic forecasting metrics. Further evaluations across diverse real-world energy datasets demonstrate its robustness, yielding substantial improvements.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Time Series Forecasting | 27 real-world application datasets (test) | SQL0.3974 | 36 | |
| Time Series Forecasting | Photovoltaic datasets | SQL0.3743 | 14 | |
| Probabilistic Univariate Time Series Forecasting | fev-bench-uni | SQL0.5664 | 14 | |
| Time Series Forecasting | fev-bench-cov | SQL Score0.7997 | 8 |