VS-PINN: A fast and efficient training of physics-informed neural networks using variable-scaling methods for solving PDEs with stiff behavior

Seungchan Ko,Sang Hyeon Park
2024-07-12
Abstract:Physics-informed neural networks (PINNs) have recently emerged as a promising way to compute the solutions of partial differential equations (PDEs) using deep neural networks. However, despite their significant success in various fields, it remains unclear in many aspects how to effectively train PINNs if the solutions of PDEs exhibit stiff behaviors or high frequencies. In this paper, we propose a new method for training PINNs using variable-scaling techniques. This method is simple and it can be applied to a wide range of problems including PDEs with rapidly-varying solutions. Throughout various numerical experiments, we will demonstrate the effectiveness of the proposed method for these problems and confirm that it can significantly improve the training efficiency and performance of PINNs. Furthermore, based on the analysis of the neural tangent kernel (NTK), we will provide theoretical evidence for this phenomenon and show that our methods can indeed improve the performance of PINNs.
Numerical Analysis,Machine Learning
What problem does this paper attempt to address?
The paper attempts to address the issue of low training efficiency of Physics-Informed Neural Networks (PINNs) when solving partial differential equations (PDEs) with stiff behavior or high-frequency components. Specifically, traditional PINNs struggle to train effectively when the solutions of PDEs exhibit stiff behavior or high-frequency variations. To tackle this problem, the paper proposes a new training method—Variable Scaling Physics-Informed Neural Networks (VS-PINNs), which improves the training process through variable scaling techniques. This method is not only simple and feasible but also widely applicable to various problems, including solving PDEs with rapidly changing characteristics. Through a series of numerical experiments, the paper demonstrates the effectiveness of this method on these problems and provides theoretical justification through Neural Tangent Kernel (NTK) analysis, proving that this method indeed enhances the performance of PINNs.