摘要
arXiv:2503.22600v1 公告类型: cross
摘要:自回归下一步预测模型已成为构建数据驱动神经求解器以预测时间相关偏微分方程(PDEs)的标准方法。与去噪扩散概率模型密切相关的去噪训练已被证明能够增强神经求解器的时间稳定性,同时其随机推理机制使集合预测和不确定性量化成为可能。原则上,这种训练涉及到在训练和推理过程中采样一系列离散的扩散时间步长,不可避免地增加了计算开销。此外,大多数扩散模型在结构化和均匀网格上施加各向同性的高斯噪声,这限制了它们对不规则域的适应性。我们提出了一种用于PDE模拟的潜在扩散模型,将PDE状态嵌入到一个低维潜在空间中,从而显著减少了计算成本。我们的框架使用自编码器将不同类型的网格映射到统一的结构化潜在网格,捕捉复杂几何形状。通过对常见的扩散路径进行分析,我们提出使用从流匹配中获得的粗采样噪声调度来进行训练和测试。数值实验表明,所提出的模型在准确性和长期稳定性方面都优于几种确定性基线,突显了基于扩散的方法在鲁棒数据驱动PDE学习中的潜在价值。