arXiv Machine Learning

Stable by Construction: Variational Latent Markov Operators for Long-Horizon PDE Prediction

The paper introduces a variational framework called VAMO that incorporates latent Markov dynamics for neural PDE solvers, aiming to improve long‑horizon predictions by mitigating error accumulation. By representing physical states as latent distributions and evolving them through probabilistic transitions, the method aligns learned dynamics with a spectral geometry induced by structured Gaussian perturbations. Experiments on fluid‑dynamics benchmarks show that VAMO reduces error growth and enhances rollout stability compared to deterministic and noise‑injection baselines.

arXiv AI
Jun 8

Autoregression-Free Neural Operators for Time-Dependent PDEs

arXiv:2605. 25413v3 Announce Type: replace-cross Abstract: Neural operators learn mappings from function-dependent inputs to solutions, providing an effective framework for solving partial differential equations (PDEs).

By Jiaquan Zhang, Caiyan Qin, Haoyu Bian, Libin Cai, Yi Lu, Chaoning Zhang, Wei Dong, Yuanfang Guo, Yang Yang, Heng Tao Shen
arXiv Machine Learning
Oct 16

Same Target, Different Origins: Aligning Latent Rollout for Consistent PDE Emulation

The paper introduces Align Latent Rollout (ALR), a regularization technique for neural PDE solvers that encourages consistency across different rollout origins. ALR uses a second‑order Taylor analysis to define two objectives: zeroth‑order consistency aligning latent states at a shared target time, and first‑order consistency aligning local evolution responses to perturbations. The method can be applied to various neural PDE solvers without altering their architectures and demonstrates improved prediction accuracy and reduced discrepancies across rollout origins in multiple benchmarks.

By Xinying Li, Jing Xu, Wenjun Wang, Limei Han, Fenglei Cao, Ruoxi Jiang, Yuan Qi
arXiv Machine Learning
Oct 15

ReAl: Refine neural operators via aligning dynamical flow

ReAl: Refine neural operators via aligning dynamical flow introduces a model‑agnostic plug‑in that enhances pretrained PDE surrogates by adding a refinement stage with a conditional flow‑matching model to recover missing high‑frequency structures, and an alignment stage that generates multiple trajectories and uses trajectory‑level rewards to stabilize long‑horizon predictions. The method improves high‑frequency fidelity, forecasting accuracy, and long‑horizon stability across diverse surrogate architectures and PDE tasks.

By Jing Xu, Zenglin Xu, Li Yi, Ruoxi Jiang, Yuan Qi