arXiv Machine Learning

Do Neural PDE Solvers Learn the Right Dynamics?

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 AI
Jun 26

Error-Conditioned Neural Solvers

arXiv:2606. 27354v1 Announce Type: cross Abstract: Neural surrogate models offer fast approximate mappings from PDE parameters to solutions, but they typically treat solving as a purely statistical task: once trained, they struggle to correct their own constraint violations and extrapolate beyond the training distribution.

By Haina Jiang, Liam Wang, Peng-Chen Chen, Min Seop Kwak, Seungryong Kim, Brian Bell, Jeong Joon Park
arXiv Machine Learning
Jul 2

TRIE: An Evaluation Framework for Stochastic PDE Surrogates

arXiv:2607. 00196v1 Announce Type: new Abstract: Many scientific systems exhibit uncertainty from stochastic forcing, unresolved degrees of freedom, or imperfect observations, making reliable surrogate forecasting fundamentally distributional rather than pointwise.

By Bharat Srikishan, Javier E. Santos, Nikhil Muralidhar, Charles D. Young
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
arXiv Machine Learning
Sep 16

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.

By Junyi Liao, Johann Guilleminot, Vahid Tarokh