LLM2D
关于用于偏微分方程模拟的条件扩散模型
On conditional diffusion models for PDE simulations
作者: Aliaksandra Shysheya, Cristiana Diaconu, Federico Bergamin, Paris Perdikaris, Jos\'e Miguel Hern\'andez-Lobato, Richard E. Turner, Emile Mathieu
发布日期: 10/23/2024
arXiv ID: oai:arXiv.org:2410.16415v1

摘要

对偏微分方程(PDE)的建模在科学和工程领域至关重要,它涵盖了从预测到反问题(如数据同化)等各种任务。然而,大多数以前针对预测的数值和机器学习方法都不能直接应用于数据同化。最近,扩散模型作为一种强大的条件生成工具出现,能够灵活地整合观测数据而无需重新训练。在这项工作中,我们对基于分数的扩散模型在稀疏观测的预测和同化中的应用进行了比较研究。特别是,我们关注以条件方式训练或在无条件训练后进行条件化的扩散模型。我们通过提出以下改进来解决现有模型的不足:1)一种自回归采样方法,显著提高了预测性能;2)一种针对条件基于分数的模型的新训练策略,可在一定的历史长度范围内实现稳定的性能;3)一种混合模型,它采用灵活的预训练条件来处理初始条件,并采用灵活的后训练条件来处理数据同化。我们通过实验证明,这些改进对于成功解决预测和数据同化相结合的任务至关重要,而这项任务在现实世界中经常遇到。