arXiv:2411.00040math.NAcs.AI2024-11NeurIPS被引 22

用粗网格+少量数据高效预测时空动力学,精度远超现有方法。

P$^2$C$^2$Net: PDE-Preserved Coarse Correction Network for efficient prediction of spatiotemporal dynamics

  • 设计双模块网络,融合物理约束与神经校正,实现粗网格求解。
  • 仅需3-5条训练轨迹,相对误差降低超50%。
  • 适合数据稀缺、追求计算效率的物理模拟场景。

求解偏微分方程(PDE)时,传统数值方法需细网格和小时间步以保证稳定性、一致性和收敛性,导致计算成本高。近年来机器学习被用于求解PDE,但面临可解释性差、泛化能力弱及对大量标注数据依赖的问题。为此,本文提出一种PDE保留的粗网格校正网络(P²C²Net),在少量数据条件下高效求解时空PDE问题。模型包含两个协同模块:(1) 可训练的PDE块,基于高阶数值格式并编码边界条件,更新粗网格系统状态;(2) 神经网络块,在线持续修正解。特别地,提出一种可学习的对称卷积滤波器,权重全局共享,准确估计基于神经校正状态的PDE空间导数。所提物理编码模型可在有限训练数据(如3–5条轨迹)下运行,显著加速粗时空网格上的解预测,同时保持高精度。P²C²Net在涵盖复杂反应-扩散过程与湍流的四个数据集上均实现一致的最先进性能,相对预测误差提升超过50%。

原文摘要 · Abstract (English)

When solving partial differential equations (PDEs), classical numerical methods often require fine mesh grids and small time stepping to meet stability, consistency, and convergence conditions, leading to high computational cost. Recently, machine learning has been increasingly utilized to solve PDE problems, but they often encounter challenges related to interpretability, generalizability, and strong dependency on rich labeled data. Hence, we introduce a new PDE-Preserved Coarse Correction Network (P$^2$C$^2$Net) to efficiently solve spatiotemporal PDE problems on coarse mesh grids in small data regimes. The model consists of two synergistic modules: (1) a trainable PDE block that learns to update the coarse solution (i.e., the system state), based on a high-order numerical scheme with boundary condition encoding, and (2) a neural network block that consistently corrects the solution on the fly. In particular, we propose a learnable symmetric Conv filter, with weights shared over the entire model, to accurately estimate the spatial derivatives of PDE based on the neural-corrected system state. The resulting physics-encoded model is capable of handling limited training data (e.g., 3--5 trajectories) and accelerates the prediction of PDE solutions on coarse spatiotemporal grids while maintaining a high accuracy. P$^2$C$^2$Net achieves consistent state-of-the-art performance with over 50\% gain (e.g., in terms of relative prediction error) across four datasets covering complex reaction-diffusion processes and turbulent flows.

PDE求解神经网络粗网格物理建模

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