arXiv:2501.14604cs.LG2025-01被引 2

用逆演化方法高效生成训练数据,提升神经PDE求解器性能

Inverse Evolution Data Augmentation for Neural PDE Solvers

  • 基于方程逆过程生成新数据,结合原始数据增强训练集
  • 采用高阶显式方案,大时间步仍保持精度,计算成本更低
  • 在三种典型演化方程上验证,显著提升FNO和UNet的鲁棒性

神经网络在求解偏微分方程(PDE)方面展现出巨大潜力,尤其通过神经算子的应用。训练神经算子通常需要大量数据以保证精度与泛化能力。本文提出一种针对演化方程的新型数据增强方法。该方法利用方程逆过程的特性,从随机初值高效生成数据,并与原始数据融合。为进一步提升生成数据的准确性,引入高阶逆演化格式,仅需少量显式计算步骤,即可使生成的数据对满足对应的隐式数值格式。相比传统PDE求解器依赖小时间步或隐式格式以保证精度,本方法采用显式格式且允许较大时间步,显著降低计算开销。实验验证了方法的有效性。在Burgers方程、Allen-Cahn方程和Navier-Stokes方程上的测试表明,结合该数据增强方法后,Fourier Neural Operator和UNet均表现出显著提升的性能与鲁棒性。

原文摘要 · Abstract (English)

Neural networks have emerged as promising tools for solving partial differential equations (PDEs), particularly through the application of neural operators. Training neural operators typically requires a large amount of training data to ensure accuracy and generalization. In this paper, we propose a novel data augmentation method specifically designed for training neural operators on evolution equations. Our approach utilizes insights from inverse processes of these equations to efficiently generate data from random initialization that are combined with original data. To further enhance the accuracy of the augmented data, we introduce high-order inverse evolution schemes. These schemes consist of only a few explicit computation steps, yet the resulting data pairs can be proven to satisfy the corresponding implicit numerical schemes. In contrast to traditional PDE solvers that require small time steps or implicit schemes to guarantee accuracy, our data augmentation method employs explicit schemes with relatively large time steps, thereby significantly reducing computational costs. Accuracy and efficacy experiments confirm the effectiveness of our approach. Additionally, we validate our approach through experiments with the Fourier Neural Operator and UNet on three common evolution equations that are Burgers' equation, the Allen-Cahn equation and the Navier-Stokes equation. The results demonstrate a significant improvement in the performance and robustness of the Fourier Neural Operator when coupled with our inverse evolution data augmentation method.

神经算子PDE求解数据增强

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