Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Lightweight Geometric Adaptation for Training Physics-Informed Neural Networks

About

Physics-Informed Neural Networks (PINNs) often suffer from slow convergence, training instability, and reduced accuracy on challenging partial differential equations due to the anisotropic and rapidly varying geometry of their loss landscapes. We propose a lightweight curvature-aware optimization framework that augments existing first-order optimizers with an adaptive predictive correction based on secant information. Consecutive gradient differences are used as a cheap proxy for local geometric change, together with a step-normalized secant curvature indicator to control the correction strength. The framework is plug-and-play, computationally efficient, and broadly compatible with existing optimizers, without explicitly forming second-order matrices. Experiments on diverse PDE benchmarks show consistent improvements in convergence speed, training stability, and solution accuracy over standard optimizers and strong baselines, including on the high-dimensional heat equation, Gray--Scott system, Belousov--Zhabotinsky system, and 2D Kuramoto--Sivashinsky system.

Kang An, Chenhao Si, Shiqian Ma, Ming Yan• 2026

Related benchmarks

TaskDatasetResultRank
PDE solving10D Heat Equation
Relative L2 Error (u)1.34
6
PDE solving2D Kuramoto-Sivashinsky equation
Relative L2 Error (u)1.79
6
PDE ModelingBelousov-Zhabotinsky system
Rel. L2 Error (u)4.7
6
Showing 3 of 3 rows

Other info

Follow for update