English

CAPS: Unifying Attention, Recurrence, and Alignment in Transformer-based Time Series Forecasting

Machine Learning 2026-02-04 v1 Artificial Intelligence

Abstract

This paper presents CAPS\textbf{CAPS} (Clock-weighted Aggregation with Prefix-products and Softmax), a structured attention mechanism for time series forecasting that decouples three distinct temporal structures: global trends, local shocks, and seasonal patterns. Standard softmax attention entangles these through global normalization, while recent recurrent models sacrifice long-term, order-independent selection for order-dependent causal structure. CAPS combines SO(2) rotations for phase alignment with three additive gating paths -- Riemann softmax, prefix-product gates, and a Clock baseline -- within a single attention layer. We introduce the Clock mechanism, a learned temporal weighting that modulates these paths through a shared notion of temporal importance. Experiments on long- and short-term forecasting benchmarks surpass vanilla softmax and linear attention mechanisms and demonstrate competitive performance against seven strong baselines with linear complexity. Our code implementation is available at https://github.com/vireshpati/CAPS-Attention.

Keywords

Cite

@article{arxiv.2602.02729,
  title  = {CAPS: Unifying Attention, Recurrence, and Alignment in Transformer-based Time Series Forecasting},
  author = {Viresh Pati and Yubin Kim and Vinh Pham and Jevon Twitty and Shihao Yang and Jiecheng Lu},
  journal= {arXiv preprint arXiv:2602.02729},
  year   = {2026}
}
R2 v1 2026-07-01T09:32:54.839Z