Accelerated Training of Physics-Informed Neural Networks (PINNs) using Meshless Discretizations
Authors: Ramansh Sharma, Varun Shankar
AI-generated Key Points
- The paper presents a new technique called discretely-trained PINNs (DT-PINNs) for accelerating the training of physics-informed neural networks (PINNs).
- DT-PINNs are trained using high-order accurate numerical discretizations computed using meshless radial basis function-finite differences (RBF-FD) and applied via sparse-matrix vector multiplication.
- This approach allows for DT-PINNs to be trained even on point cloud samples placed on irregular domain geometries.
- Using fp64 on the GPU leads to significantly faster training times than fp32 vanilla-PINNs with comparable accuracy.
- RBF-FD approximations of third order accuracy and above are more efficient while being sufficiently accurate for network depth.
- DT-PINNs achieve similar losses with 2–4x faster training times compared to vanilla-PINNs on both linear and nonlinear Poisson equations on a consumer GPU.
- Similar results can be obtained for the PINN solution to the heat equation by discretizing spatial derivatives using RBF-FD and using automatic differentiation for temporal derivatives.
- Fp64 DT-PINNs offer a superior cost–accuracy profile compared to fp32 vanilla–PINNs, which could have implications for accelerating scientific computing tasks that rely on PINNs such as fluid dynamics simulations and inverse problems in medical imaging.
Abstract: We present a new technique for the accelerated training of physics-informed neural networks (PINNs): discretely-trained PINNs (DT-PINNs). The repeated computation of partial derivative terms in the PINN loss functions via automatic differentiation during training is known to be computationally expensive, especially for higher-order derivatives. DT-PINNs are trained by replacing these exact spatial derivatives with high-order accurate numerical discretizations computed using meshless radial basis function-finite differences (RBF-FD) and applied via sparse-matrix vector multiplication. The use of RBF-FD allows for DT-PINNs to be trained even on point cloud samples placed on irregular domain geometries. Additionally, though traditional PINNs (vanilla-PINNs) are typically stored and trained in 32-bit floating-point (fp32) on the GPU, we show that for DT-PINNs, using fp64 on the GPU leads to significantly faster training times than fp32 vanilla-PINNs with comparable accuracy. We demonstrate the efficiency and accuracy of DT-PINNs via a series of experiments. First, we explore the effect of network depth on both numerical and automatic differentiation of a neural network with random weights and show that RBF-FD approximations of third-order accuracy and above are more efficient while being sufficiently accurate. We then compare the DT-PINNs to vanilla-PINNs on both linear and nonlinear Poisson equations and show that DT-PINNs achieve similar losses with 2-4x faster training times on a consumer GPU. Finally, we also demonstrate that similar results can be obtained for the PINN solution to the heat equation (a space-time problem) by discretizing the spatial derivatives using RBF-FD and using automatic differentiation for the temporal derivative. Our results show that fp64 DT-PINNs offer a superior cost-accuracy profile to fp32 vanilla-PINNs.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
⚠The license of the paper does not allow us to build upon its content and the AI assistant only knows about the paper metadata rather than the full article.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
⚠The license of this specific paper does not allow us to build upon its content and the summarizing tools will be run using the paper metadata rather than the full article. However, it still does a good job, and you can also try our tools on papers with more open licenses.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.