Accelerating Data Generation for Nonlinear temporal PDEs via homologous perturbation in solution space

TL;DR

The paper introduces HOPSS, a novel algorithm that accelerates data generation for training neural operators on nonlinear PDEs by using homologous perturbations, reducing computational time while maintaining data quality.

Contribution

HOPSS is a new data generation method that efficiently produces training datasets for nonlinear PDEs with fewer time steps, lowering computational costs.

Findings

HOPSS reduces data generation time to about 10% of traditional methods.

It maintains comparable accuracy in model training.

Effective on complex equations like Navier-Stokes.

Abstract

Data-driven deep learning methods like neural operators have advanced in solving nonlinear temporal partial differential equations (PDEs). However, these methods require large quantities of solution pairs\u2014the solution functions and right-hand sides (RHS) of the equations. These pairs are typically generated via traditional numerical methods, which need thousands of time steps iterations far more than the dozens required for training, creating heavy computational and temporal overheads. To address these challenges, we propose a novel data generation algorithm, called HOmologous Perturbation in Solution Space (HOPSS), which directly generates training datasets with fewer time steps rather than following the traditional approach of generating large time steps datasets. This algorithm simultaneously accelerates dataset generation and preserves the approximate precision required for model training. Specifically, we first obtain a set of base solution functions from a reliable solver, usually with thousands of time steps, and then align them in time steps with training datasets by downsampling. Subsequently, we propose a "homologous perturbation" approach: by combining two solution functions (one as the primary function, the other as a homologous perturbation term scaled by a small scalar) with random noise, we efficiently generate comparable-precision PDE data points. Finally, using these data points, we compute the variation in the original equation's RHS to form new solution pairs. Theoretical and experimental results show HOPSS lowers time complexity. For example, on the Navier-Stokes equation, it generates 10,000 samples in approximately 10% of traditional methods' time, with comparable model training performance.