ICLR2022: Discovering Nonlinear PDEs from Scarce Data with Physics-encoded Learning
Paper link: [ArXiv]
By Chengping Rao, Pu Ren, Yang Liu, Hao Sun
Three stages of governing equation discovery process
Two types of physics-encoded recurrent network (a. partial physics known; c. physics completely known.)
In Stage-1, we use a physics-encoded recurrent network to reconstruct the high-fidelity data. This step uses the same rountine of https://github.com/Raocp/PeRCNN.
The sparse regression for two equation of u (PDE_FIND_u.py
), v (PDE_FIND_v.py
) is performed separately. We recommend you to run the sparse regression via IPython or Jupyter Notebook.
Based on the result from Stage-2, we perform coefficient finetuning using a physics-based recurrent network (i.e., the recurrent block mimics finite difference discretization of a governing PDE). Note that the finetuning is performed on the orginal sparse data (please refer the paper for explanation).
Set restart=True
(it is a bad arg name...) when invoking train() function to read from checkpoint and continue training.