用记忆机制提升生成式偏微分方程求解的长期稳定性
Memory-Conditioned Flow-Matching for Stable Autoregressive PDE Rollouts
- 引入记忆条件化的扩散/流匹配,通过隐空间注入在线状态
- 在含激波和多尺度混合的压缩流中实现更稳定、更精确的长时滚动
- 适合需要高保真长期模拟的物理建模与科学计算研究者
自回归生成式偏微分方程求解器在单步预测中准确,但在长时滚动中易发散,尤其在从粗到细的演化阶段,每一步需重新生成未解析的精细尺度。这正是扩散与流匹配生成器的典型场景:尽管其内部动态为马尔可夫过程,但滚动稳定性取决于每步的条件分布误差。基于Mori--Zwanzig投影形式,我们证明消除未解析变量会导出一个精确的解析演化方程,包含马尔可夫项、记忆项和正交扰动项,揭示了无记忆闭包的根本局限性。受此启发,我们提出记忆条件化的扩散/流匹配,通过潜空间特征在线注入紧凑状态以指导去噪。利用分解理论,记忆构建了未解析尺度的结构化条件尾部先验,减少对缺失频率的传输需求。我们证明了所得条件核的Wasserstein稳定性,并推导出离散Grönwall滚动边界,将记忆近似误差与条件生成误差分离。在含激波和多尺度混合的压缩流实验中,模型显著提升精度,长期滚动更加稳定,且在频谱与统计特性上更贴近真实数据。
原文摘要 · Abstract (English)
Autoregressive generative PDE solvers can be accurate one step ahead yet drift over long rollouts, especially in coarse-to-fine regimes where each step must regenerate unresolved fine scales. This is the regime of diffusion and flow-matching generators: although their internal dynamics are Markovian, rollout stability is governed by per-step \emph{conditional law} errors. Using the Mori--Zwanzig projection formalism, we show that eliminating unresolved variables yields an exact resolved evolution with a Markov term, a memory term, and an orthogonal forcing, exposing a structural limitation of memoryless closures. Motivated by this, we introduce memory-conditioned diffusion/flow-matching with a compact online state injected into denoising via latent features. Via disintegration, memory induces a structured conditional tail prior for unresolved scales and reduces the transport needed to populate missing frequencies. We prove Wasserstein stability of the resulting conditional kernel. We then derive discrete Grönwall rollout bounds that separate memory approximation from conditional generation error. Experiments on compressible flows with shocks and multiscale mixing show improved accuracy and markedly more stable long-horizon rollouts, with better fine-scale spectral and statistical fidelity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。