ArXiv TLDR

STEPS: A Temporal Smooth Error Propagation Solver on the Manifolds for Test-Time Adaptation in Time Series Forecasting

🐦 Tweet
2605.08005

Jiaqi Liu, Yifan Ouyang, Zhifei Song, Sim Kuan Goh, Ashwaq Qasem

cs.LG

TLDR

STEPS introduces a manifold-based error propagation solver for robust Test-Time Adaptation in time series forecasting, significantly improving accuracy.

Key contributions

  • Reformulates Test-Time Adaptation as a Dirichlet Boundary Value Problem on a temporal manifold.
  • Uses Local and Global Solvers to propagate prefix errors and retrieve stable cross-window error memory.
  • Integrates solutions via Spatiotemporal Manifold Fusion (SMF) for robust and smooth prediction corrections.
  • Achieves 26.82% MSE reduction and outperforms strongest TTA baselines by 12.77% across six benchmarks.

Why it matters

This paper addresses critical challenges in Test-Time Adaptation for time series forecasting, such as error accumulation and instability under distribution shifts. STEPS offers a robust, manifold-based solution that significantly improves forecasting accuracy and reliability, especially with limited or noisy data.

Original Abstract

Test-Time Adaptation (TTA) aims to improve time series forecasting under distribution shifts by using limited observations revealed during inference. However, forecasting TTA must operate in a source-free online setting, where the adaptation signal is short, temporally correlated, and potentially noisy. Existing methods can therefore suffer from weak identifiability, error accumulation, and unstable long-horizon corrections when the revealed prefix is sparse or contaminated. To address these issues, we propose STEPS, a Smooth Temporal Error Propagation Solver for TTA in time-series forecasting. STEPS reformulates forecasting TTA as a Dirichlet Boundary Value Problem on a temporal manifold, where the revealed prefix error serves as the boundary condition for the unknown future error field. Then, STEPS solves a smooth and bounded correction field in prediction space: a Local Solver propagates prefix errors under temporal smoothness, a Global Solver retrieves stable cross-window error memory and Spatiotemporal Manifold Fusion (SMF) integrates both solutions into the final correction. Across six standard benchmarks and four frozen backbones, STEPS achieves an average relative MSE reduction of 26.82% over the zero-shot backbone, exceeding the strongest compared TTA baseline by 12.77%. Additional sparse prefix and contamination tests confirm the robustness of STEPS under limited and noisy prefixes.

📬 Weekly AI Paper Digest

Get the top 10 AI/ML arXiv papers from the week — summarized, scored, and delivered to your inbox every Monday.