English

Time-Aware Prior Fitted Networks for Zero-Shot Forecasting with Exogenous Variables

Machine Learning 2026-03-18 v1

Abstract

In many time series forecasting settings, the target time series is accompanied by exogenous covariates, such as promotions and prices in retail demand; temperature in energy load; calendar and holiday indicators for traffic or sales; and grid load or fuel costs in electricity pricing. Ignoring these exogenous signals can substantially degrade forecasting accuracy, particularly when they drive spikes, discontinuities, or regime and phase changes in the target series. Most current time series foundation models (e.g., Chronos, Sundial, TimesFM, TimeMoE, TimeLLM, and LagLlama) ignore exogenous covariates and make forecasts solely from the numerical time series history, thereby limiting their performance. In this paper, we develop ApolloPFN, a prior-data fitted network (PFN) that is time-aware (unlike prior PFNs) and that natively incorporates exogenous covariates (unlike prior univariate forecasters). Our design introduces two major advances: (i) a synthetic data generation procedure tailored to resolve the failure modes that arise when tabular (non-temporal) PFNs are applied to time series; and (ii) time-aware architectural modifications that embed inductive biases needed to exploit the time series context. We demonstrate that ApolloPFN achieves state-of-the-art results across benchmarks, such as M5 and electric price forecasting, that contain exogenous information.

Keywords

Cite

@article{arxiv.2603.15802,
  title  = {Time-Aware Prior Fitted Networks for Zero-Shot Forecasting with Exogenous Variables},
  author = {Andres Potapczynski and Ravi Kiran Selvam and Tatiana Konstantinova and Shankar Ramasubramanian and Malcolm Wolff and Kin G. Olivares and Ruijun Ma and Mengfei Cao and Michael W. Mahoney and Andrew Gordon Wilson and Boris N. Oreshkin and Dmitry Efimov},
  journal= {arXiv preprint arXiv:2603.15802},
  year   = {2026}
}
R2 v1 2026-07-01T11:23:03.511Z