用解耦数据训练扩散模型,推理时组合生成耦合方程长时序解。
Compositional Generation for Long-Horizon Coupled PDEs
- 仅用解耦数据训练扩散模型,推理时组合恢复耦合场。
- 在反应-扩散与修正伯格斯方程上,误差低且可扩展至长时序。
- 适合缺乏耦合数据但需高效建模的科学计算场景。
模拟耦合偏微分方程系统计算成本高,以往方法多依赖大量耦合数据训练代理模型。本文研究基于组合扩散的方法:仅在解耦的PDE数据上训练扩散模型,并在推理阶段组合以恢复耦合场。特别关注长时域、大量时间步下的可行性。对比了基础扩散模型与采用v-参数化策略的模型,还提出基于欧拉格式的对称组合方案。在反应-扩散和修改伯格斯方程上评估,使用更长时间网格,与在耦合数据上训练的傅里叶神经算子(FNO)对比。尽管仅见解耦训练数据,组合扩散模型仍能以低误差恢复耦合轨迹。v-参数化优于基础模型,但FNO因使用耦合数据仍最优。结果表明,组合扩散是实现高效长时域耦合PDE建模的可行策略。
原文摘要 · Abstract (English)
Simulating coupled PDE systems is computationally intensive, and prior efforts have largely focused on training surrogates on the joint (coupled) data, which requires a large amount of data. In the paper, we study compositional diffusion approaches where diffusion models are only trained on the decoupled PDE data and are composed at inference time to recover the coupled field. Specifically, we investigate whether the compositional strategy can be feasible under long time horizons involving a large number of time steps. In addition, we compare a baseline diffusion model with that trained using the v-parameterization strategy. We also introduce a symmetric compositional scheme for the coupled fields based on the Euler scheme. We evaluate on Reaction-Diffusion and modified Burgers with longer time grids, and benchmark against a Fourier Neural Operator trained on coupled data. Despite seeing only decoupled training data, the compositional diffusion models recover coupled trajectories with low error. v-parameterization can improve accuracy over a baseline diffusion model, while the neural operator surrogate remains strongest given that it is trained on the coupled data. These results show that compositional diffusion is a viable strategy towards efficient, long-horizon modeling of coupled PDEs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。