Mitigating Gradient Pathology in PINNs through Aligned Constraint
Yichen Luo, Peiyu Zhu, Dongxiao Hu, Jia Wang, Tailin Wu, Dapeng Lan, Yu Liu, Zhibo Pang
Abstract
While Physics-Informed Neural Networks (PINNs) are powerful for solving Partial Differential Equations (PDEs), their training is often paralyzed by gradient pathology. The gradients from the PDE residuals and boundary constraints oppose each other, trapping the model in local minima. Current solutions, such as adaptive weighting or hard constraints, either fail to fundamentally resolve this ill-conditioning or are limited to simple geometries. In this study, we systematically analyze the possible causes of this gradient pathology from the perspectives of loss landscapes and optimization dynamics. Based on the obtained conclusion, we propose Constraint-Aligned loss with Manifold Lifting (CAML). By reformulating all zeroth-order terms into aligned constraints, our method effectively mitigates gradient conflicts. In addition, we introduce a delay factor to help the optimizer skip the high-curvature area. Experiments demonstrate that our CAML significantly enhances numerical stability and efficiency in highly complex PINN problems. Our code is open-sourced on CAML .
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 22a768d4-9b77-4465-972e-e079ef25aff3Builds on11
- Fourier Neural Operator for Parametric Partial Differential EquationsZongyi Li, Nikola Borislavov Kovachki, Kamyar Azizzadenesheli, Burigede Liu et al.ICLR 2021 · 3,911 citations
- Sharpness-aware Minimization for Efficiently Improving GeneralizationPierre Foret, Ariel Kleiner, Hossein Mobahi, Behnam NeyshaburICLR 2021 · 1,861 citations
- Characterizing possible failure modes in physics-informed neural networksAditi S. Krishnapriyan, Amir Gholami, Shandian Zhe, Robert M. Kirby et al.NeurIPS 2021 · 1,421 citations
- Challenges in Training PINNs: A Loss Landscape PerspectivePratik Rathore, Weimu Lei, Zachary Frangella, Lu Lu et al.ICML 2024 · 137 citations
- A Unified Hard-Constraint Framework for Solving Geometrically Complex PDEsSongming Liu, Zhongkai Hao, Chengyang Ying, Hang Su et al.NeurIPS 2022 · 47 citations
Related papers
- Dual Cone Gradient Descent for Training Physics-Informed Neural NetworksYoungsik Hwang, Dong-Young LimNeurIPS 2024 · 34 citations
- CoPINN: Cognitive Physics-Informed Neural NetworksSiyuan Duan, Wenyuan Wu, Peng Hu, Zhenwen Ren et al.ICML 2025
- RoPINN: Region Optimized Physics-Informed Neural NetworksHaixu Wu, Huakun Luo, Yuezhou Ma, Jianmin Wang et al.NeurIPS 2024 · 50 citations
- Enhancing Stability of Physics-Informed Neural Network Training Through Saddle-Point ReformulationDmitry Bylinkin, Mikhail Aleksandrov, Savelii Chezhegov, Aleksandr BeznosikovICLR 2026 · 1 citation
- An operator preconditioning perspective on training in physics-informed machine learningTim De Ryck, Florent Bonnet, Siddhartha Mishra, Emmanuel de BézenacICLR 2024 · 28 citations
