Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

StretchTime: Adaptive Time Series Forecasting via Symplectic Attention

About

Transformer architectures have established strong baselines in time series forecasting, yet they typically rely on positional encodings that assume uniform, index-based temporal progression. However, real-world systems, from shifting financial cycles to elastic biological rhythms, frequently exhibit "time-warped" dynamics where the effective flow of time decouples from the sampling index. In this work, we first formalize this misalignment and prove that rotary position embedding (RoPE) is mathematically incapable of representing non-affine temporal warping. To address this, we propose Symplectic Positional Embeddings (SyPE), a learnable encoding framework derived from Hamiltonian mechanics. SyPE strictly generalizes RoPE by extending the rotation group $\mathrm{SO}(2)$ to the symplectic group $\mathrm{Sp}(2,\mathbb{R})$, modulated by a novel input-dependent adaptive warp module. By allowing the attention mechanism to adaptively dilate or contract temporal coordinates end-to-end, our approach captures locally varying periodicities without requiring pre-defined warping functions. We implement this mechanism in StretchTime, a multivariate forecasting architecture that achieves state-of-the-art performance on standard benchmarks, demonstrating superior robustness on datasets exhibiting non-stationary temporal dynamics.

Yubin Kim, Viresh Pati, Jevon Twitty, Vinh Pham, Shihao Yang, Jiecheng Lu• 2026

Related benchmarks

TaskDatasetResultRank
Multivariate Time-series ForecastingETTh2 (test)
MSE0.384
171
Multivariate Time-series ForecastingETTh1 (test)
MSE0.424
134
Multivariate Time-series ForecastingWeather (test)
MSE0.237
124
Multivariate Time-series ForecastingECL (test)
MSE0.174
77
Multivariate Time-series ForecastingETTm1 (test)
MSE0.367
67
Multivariate ForecastingPEMS03 (test)
MSE0.094
43
Multivariate ForecastingPEMS08 (test)
MSE0.118
43
Multivariate ForecastingPEMS04 (test)
MSE0.088
41
Multivariate Time-series ForecastingETTm2 (test)
MSE0.276
35
Multivariate Time-series ForecastingSolar (test)--
29
Showing 10 of 10 rows

Other info

Follow for update