English

TIDES: Implicit Time-Awareness in Selective State Space Models

Machine Learning 2026-05-12 v1 Artificial Intelligence

Abstract

Selective state space models (SSMs), such as Mamba, achieve strong per-token expressivity by making the time discretization step \TildeΔ\Tilde{\Delta} a learned function of the input. However, in doing so, \TildeΔ\Tilde{\Delta} ceases to represent a physical sampling interval, limiting its irregular time series modeling capability. Continuous-time SSMs, such as S5, preserve the physical meaning of \TildeΔ\Tilde{\Delta} and handle irregular timestamps natively (\TildeΔΔ)\Tilde{\Delta}\equiv\Delta), but their dynamics remain linear time-invariant (LTI), limiting per-token expressivity. We propose \textbf{TIDES}, a selective SSM variant that reconciles selective and continuous architectures by moving input-dependence off the step size and onto the diagonal state matrix. As a result, \TildeΔ\Tilde{\Delta} retains its physical meaning, tied to the state discretization, allowing the model to handle irregular timestamps natively without sacrificing the per-token expressivity that makes selective SSMs effective. We show this on a novel \emph{Fading Flash} experimental benchmark, a compact controlled diagnostic for sequence models that jointly tests input-dependence and extrapolation to out-of-distribution Δ\Delta values, and isolates the distinct failure modes of current state-of-the-art architectures that TIDES avoids by construction. On large-scale benchmarks, TIDES sets the new state-of-the-art average rank on UEA time-series classification and the Physiome-ODE regression benchmark. Code available at: https://github.com/TaylanSoydan/TIDES.

Keywords

Cite

@article{arxiv.2605.09742,
  title  = {TIDES: Implicit Time-Awareness in Selective State Space Models},
  author = {Taylan Soydan and Miguel A. Bessa and Dirk Mohr and Rui Barreira},
  journal= {arXiv preprint arXiv:2605.09742},
  year   = {2026}
}

Comments

Preprint submitted for peer-review