Latent Representation and Simulation of Markov Processes via Time-Lagged Information Bottleneck
Marco Federici, Patrick Forré, Ryota Tomioka, Bastiaan S. Veeling
Abstract
Markov processes are widely used mathematical models for describing dynamic systems in various fields. However, accurately simulating large-scale systems at long time scales is computationally expensive due to the short time steps required for accurate integration. In this paper, we introduce an inference process that maps complex systems into a simplified representational space and models large jumps in time. To achieve this, we propose Time-lagged Information Bottleneck (T-IB), a principled objective rooted in information theory, which aims to capture relevant temporal features while discarding high-frequency information to simplify the simulation task and minimize the inference error. Our experiments demonstrate that T-IB learns information-optimal representations for accurately modeling the statistical properties and dynamics of the original process at a selected time lag, outperforming existing time-lagged dimensionality reduction methods.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers6
- Learning invariant representations of time-homogeneous stochastic dynamical systemsVladimir R. Kostic, Pietro Novelli, Riccardo Grazzi, Karim Lounici et al.ICLR 2024 · 17 citations
- Self-Supervised Evolution Operator Learning for High-Dimensional Dynamical SystemsGiacomo Turri, Luigi Bonati, Kai Zhu, Massimiliano Pontil et al.ICLR 2026 · 10 citations
- WeightFlow: Learning Stochastic Dynamics via Evolving Weight of Neural NetworkRuikun Li, Jiazhen Liu, Huandong Wang, Qingmin Liao et al.AAAI 2026 · 6 citations
- Information Shapes Koopman RepresentationXiaoyuan Cheng, Wenxuan Yuan, Yiming Yang, Yuanzhao Zhang et al.ICLR 2026 · 4 citations
- Predicting the Dynamics of Complex System via Multiscale Diffusion AutoencoderRuikun Li, Jingwen Cheng, Huandong Wang, Qingmin Liao et al.KDD 2025 · 1 citation
Builds on8
- A Simple Framework for Contrastive Learning of Visual RepresentationsTing Chen, Simon Kornblith, Mohammad Norouzi, Geoffrey E. HintonICML 2020 · 24,064 citations
- Earthformer: Exploring Space-Time Transformers for Earth System ForecastingZhihan Gao, Xingjian Shi, Hao Wang, Yi Zhu et al.NeurIPS 2022 · 410 citations
- Learning Robust Representations via Multi-View Information BottleneckMarco Federici, Anjan Dutta, Patrick Forré, Nate Kushman et al.ICLR 2020 · 330 citations
- Equivariant Transformers for Neural Network based Molecular PotentialsPhilipp Thölke, Gianni De FabritiisICLR 2022 · 277 citations
- Understanding the Limitations of Variational Mutual Information EstimatorsJiaming Song, Stefano ErmonICLR 2020 · 243 citations
Related papers
- Explaining A Black-box By Using A Deep Variational Information Bottleneck ApproachSeo-Jin Bang, Pengtao Xie, Heewook Lee, Wei Wu et al.AAAI 2021 · 33 citations
- Deep Bayesian Active Learning for Accelerating Stochastic SimulationDongxia Wu, Ruijia Niu, Matteo Chinazzi, Alessandro Vespignani et al.KDD 2023 · 3 citations
- Physics-aware, probabilistic model order reduction with guaranteed stabilitySebastian Kaltenbach, Phaedon-Stelios KoutsourelakisICLR 2021 · 16 citations
- Time-Aware World Model for Adaptive Prediction and ControlAnh N. Nhu, Sanghyun Son, Ming LinICML 2025
- Laplace Transform Based Low-Complexity Learning of Continuous Markov SemigroupsVladimir R. Kostic, Karim Lounici, Hélène Halconruy, Timothée Devergne et al.ICML 2025
