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

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.

Linxiao Yang, Xue Jiang, Gezheng Xu, Tian Zhou, Min Yang, ZhaoYang Zhu, Linyuan Geng, Zhipeng Zeng, Qiming Chen, Xinyue Gu, Rong Jin, Liang Sun• 2026

Related benchmarks

TaskDatasetResultRank
Time Series Forecasting27 real-world application datasets (test)
SQL0.3974
36
Time Series ForecastingPhotovoltaic datasets
SQL0.3743
14
Probabilistic Univariate Time Series Forecastingfev-bench-uni
SQL0.5664
14
Time Series Forecastingfev-bench-cov
SQL Score0.7997
8
Showing 4 of 4 rows

Other info

Follow for update