arXiv:2501.15987math.NAcs.AI2025-01ICML被引 5

融合物理方程与多时间步学习,实现快速高精度流体模拟。

MultiPDENet: PDE-embedded Learning with Multi-time-stepping for Accelerated Flow Simulation

  • 用有限差分结构设计轻量卷积滤波器,粗网格上优化空间导数残差。
  • 引入四阶龙格-库塔物理模块与多尺度时间积分,抑制长期预测误差累积。
  • 仅需少量不完整数据即可准确预测长时间流场演化,适合工程仿真场景。

通过数值方法求解偏微分方程(PDE)面临计算成本高的挑战,因需精细网格和小时间步。机器学习虽可加速,但泛化性差、可解释性弱、依赖数据,且长期预测性能不佳。为此,我们提出一种嵌入物理方程的多时间步学习网络(MultiPDENet),融合数值方法与机器学习优势,用于加速流体模拟。设计基于有限差分模板结构的卷积滤波器,参数少,可在粗网格上估计空间导数的等效形式以最小化方程残差;构建包含四阶龙格-库塔积分器的物理块,在细时间尺度上嵌入PDE结构引导预测;为缓解长期预测中的时间误差累积,引入多尺度时间积分策略,用神经网络修正粗时间尺度上的预测误差。在多种PDE系统(包括纳维-斯托克斯方程)上的实验表明,即使训练数据稀疏不完整(如时空下采样数据),MultiPDENet仍能准确预测长期时空动态,性能优于其他神经基线模型,并显著快于传统数值方法。

原文摘要 · Abstract (English)

Solving partial differential equations (PDEs) by numerical methods meet computational cost challenge for getting the accurate solution since fine grids and small time steps are required. Machine learning can accelerate this process, but struggle with weak generalizability, interpretability, and data dependency, as well as suffer in long-term prediction. To this end, we propose a PDE-embedded network with multiscale time stepping (MultiPDENet), which fuses the scheme of numerical methods and machine learning, for accelerated simulation of flows. In particular, we design a convolutional filter based on the structure of finite difference stencils with a small number of parameters to optimize, which estimates the equivalent form of spatial derivative on a coarse grid to minimize the equation's residual. A Physics Block with a 4th-order Runge-Kutta integrator at the fine time scale is established that embeds the structure of PDEs to guide the prediction. To alleviate the curse of temporal error accumulation in long-term prediction, we introduce a multiscale time integration approach, where a neural network is used to correct the prediction error at a coarse time scale. Experiments across various PDE systems, including the Navier-Stokes equations, demonstrate that MultiPDENet can accurately predict long-term spatiotemporal dynamics, even given small and incomplete training data, e.g., spatiotemporally down-sampled datasets. MultiPDENet achieves the state-of-the-art performance compared with other neural baseline models, also with clear speedup compared to classical numerical methods.

流体模拟PDE网络多时间步物理信息

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。