ReAl: Refine neural operators via aligning dynamical flow
On the map: Diffusion models → Flow matching, time series and science
Abstract (arXiv)
Neural partial differential equation (PDE) surrogates trained with mean squared error (MSE) can capture dominant dynamics but often lose high-frequency information and fine-scale structures. Generative models (e.g., Flow Matching) can capture the distribution of fine-scale details, yet stochastic errors may accumulate during autoregressive rollouts and compromise long-term stability. To address these limitations, we introduce Refine and Align(ReAl), a model-agnostic plug-in that augments pretrained PDE surrogates to improve the reconstruction of high-frequency structures associated with complex physical phenomena. In the refinement stage, a conditional flow-matching model learns the residual distribution around the deterministic base prediction, recovering missing fine-scale structures while retaining the dominant dynamics. In the alignment stage, to mitigate the generative discrepancies amplified in the autoregressive rollout, particularly in chaotic systems, we generate multiple trajectories from the same input and reinforce the refiner with trajectory-level rewards, thereby achieving long-horizon dynamical alignment. Experiments across diverse surrogate architectures and PDE tasks demonstrate that ReAl consistently improves high-frequency fidelity, forecasting accuracy, and long-horizon stability.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. Builds on 29 of its 36 references that are in the map. First posted on arXiv 14 Oct 2026.
Builds on
29 of 36 references.
Closest ancestors · 10
Same direction, newest first
- 2026FlowRefiner: Flow Matching-Based Iterative Refinement for 3D Turbulent Flow Simulation18 Apr
- 2026DiffusionRollout: Uncertainty-Aware Rollout Planning in Long-Horizon PDE Solving14 Feb
- 2025Physics-Constrained Flow Matching: Sampling Generative Models with Hard Constraints4 Jun
- 2024Wavelet Diffusion Neural Operator6 Dec
- 2024On conditional diffusion models for PDE simulations21 Oct
- 2024DiffusionPDE: Generative PDE-Solving Under Partial Observation25 Jun
- 2023Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation4 Sep
- 2023PDE-Refiner: Achieving Accurate Long Rollouts with Neural PDE Solvers10 Aug
- 2023DYffusion: A Dynamics-informed Diffusion Model for Spatiotemporal Forecasting3 Jun
- 2022Flow Matching for Generative Modeling6 Oct
Other references · 19
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,908
- 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 2,018
- 2023Training Diffusion Models with Reinforcement Learningcited by 982
- 2025Flow-GRPO: Training Flow Matching Models via Online RLcited by 661
- 2023DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Modelscited by 450
- 2024Transolver: A Fast Transformer Solver for PDEs on General Geometriescited by 326
- 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Processcited by 230
- 2024Poseidon: Efficient Foundation Models for PDEscited by 200
- 2024DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Trainingcited by 151
7 more
- 2022Physics-Informed Deep Neural Operator Networkscited by 113
- 2023Solving High-Dimensional PDEs with Latent Spectral Modelscited by 89
- 2025Guided Diffusion Sampling on Function Spaces with Applications to PDEscited by 36
- 2026RealPDEBench: A Benchmark for Complex Physical Systems with Real-World Datacited by 16
- 2025Hierarchical Implicit Neural Emulatorscited by 10
- 2025PDESpectralRefiner: Achieving More Accurate Long Rollouts with Spectral Adjustmentcited by 2
- 2026Di-BiLPS: Denoising induced Bidirectional Latent-PDE-Solver under Sparse Observationscited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Flow marching for a generative PDE foundation model
- 2026Latent Generative Solvers for Generalizable Long-Term Physics Simulation
- 2026Learning When to Refine: Long-Horizon Reinforcement Learning for Budgeted Neural-Operator PDE Solvers
- 2026Towards Infinitely Long Neural Simulations: Self-Refining Neural Surrogate Models for Dynamical Systems
- 2025CFO: Learning Continuous-Time PDE Dynamics via Flow-Matched Neural Operators
- 2025Residual-augmented flow matching operators for probabilistic partial differential equations
- 2025Operator Flow Matching for Timeseries Forecasting
- 2025Physics-Constrained Fine-Tuning of Flow-Matching Models for Generation and Inverse Problems