mlsanityData to 14 Oct 2026

Beyond Compression: Training Latent Representations for Stable Long-Horizon Rollout in Neural Surrogate Solvers

arXiv 2609.30198 ↗First posted 24 Sep 2026cs.LG cond-mat.mtrl-sci cs.CEcited by 0

Not on the map. Why?

Abstract (arXiv)

Latent neural surrogate solvers, or latent dynamics models, accelerate simulations of time-dependent physical systems by evolving a compressed latent space rather than resolving full-resolution fields directly. In principle this reduces computational cost and simplifies learning, but in practice errors often accumulate rapidly during long autoregressive rollouts, limiting predictive utility. We show that this instability does not stem from the latent representation itself, but arises when it is trained solely for reconstruction, producing representations poorly suited to long-horizon forecasting. We systematically evaluate training-level interventions that align latent representations with long-horizon rollout: Koopman operator learning and Hamming noise injection during autoencoder training to improve compression, together with noise injection and multi-step rollout fine-tuning to improve dynamics. Interventions that improve long-horizon rollout stability often degrade conventional training metrics, including reconstruction and one-step prediction accuracy. Collectively, these interventions reduce long-rollout error by approximately 40% and match or exceed the accuracy of full-resolution models on two physics benchmarks, while requiring 2 orders of magnitude fewer floating point operations and half the GPU memory. Applied to mesoscale crystal-plasticity simulations of high-cycle fatigue, the resulting surrogate achieves stable extrapolation over horizons orders of magnitude beyond those observed during training. More broadly, these results show that neural compression should be designed not merely to reduce dimensionality, but to restructure the solution space for stable dynamical evolution, a key requirement for reliable, efficient neural surrogates in scientific applications.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 28 of its 57 references that are in the map. First posted on arXiv 24 Sep 2026.

Builds on

28 of 57 references.

References · 28

  1. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  2. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
  3. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  4. 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,253
  5. 2020On Layer Normalization in the Transformer Architecturecited by 922
  6. 2020Learning Mesh-Based Simulation with Graph Networkscited by 868
  7. 2025Cosmos World Foundation Model Platform for Physical AIcited by 864
  8. 2022FourCastNet: A Global Data-driven High-resolution Weather Model using Adaptive Fourier Neural Operatorscited by 776
  9. 2024Genie: Generative Interactive Environmentscited by 747
  10. 2017Deep learning for universal linear embeddings of nonlinear dynamicscited by 494
  11. 2022Message Passing Neural PDE Solverscited by 386
  12. 2019Axial Attention in Multidimensional Transformerscited by 358
16 more
  1. 2023PDE-Refiner: Achieving Accurate Long Rollouts with Neural PDE Solverscited by 216
  2. 2021Primer: Searching for Efficient Transformers for Language Modelingcited by 164
  3. 2024The Well: a Large-Scale Collection of Diverse Physics Simulations for Machine Learningcited by 145
  4. 2020Transformers for Modeling Physical Systemscited by 106
  5. 2022Learning two-phase microstructure evolution using neural operators and autoencoder architecturescited by 72
  6. 2019Deep learning to discover and predict dynamics on an inertial manifoldcited by 43
  7. 2020Deep Neural Networks for Nonlinear Model Order Reduction of Unsteady Flowscited by 36
  8. 2021Deep Learning for Reduced Order Modelling and Efficient Temporal Evolution of Fluid Simulationscited by 36
  9. 2025Walrus: A Cross-Domain Foundation Model for Continuum Dynamicscited by 36
  10. 2023Rethinking materials simulations: Blending direct numerical simulations with neural operatorscited by 25
  11. 2024LiteVAE: Lightweight and Efficient Variational Autoencoders for Latent Diffusion Modelscited by 22
  12. 2025Lost in Latent Space: An Empirical Study of Latent Diffusion Models for Physics Emulationcited by 19
  13. 2025Revisiting Multimodal Positional Encoding in Vision-Language Modelscited by 16
  14. 2023Learning fast, accurate, and stable closures of a kinetic theory of an active fluidcited by 9
  15. 2026Tadpole: Autoencoders as Foundation Models for 3D PDEs with Online Learningcited by 3
  16. 2025Deep Learning-Driven Prediction of Microstructure Evolution via Latent Space Interpolationcited by 1

Built on by

Not cited yet.