Neural Electrostatics: A 3D Physics-Informed Boundary Element Poisson Equation Solver

27 Sept 2024 (modified: 05 Feb 2025)Submitted to ICLR 2025EveryoneRevisionsBibTeXCC BY 4.0
Keywords: PINN, physics informed, scientific machine learning, electrostatics, scientific computing
TL;DR: We propose a method for solving a 3D Poisson equation using a neural network basis.
Abstract: Electrostatics solvers relate an imposed voltage to a corresponding charge density. Current classical methods require fine discretization and scale poorly due to the construction of a large linear system of equations. We recast the problem using neural networks and introduce neural electrostatics, a hybrid 3D boundary element method (BEM). By using the boundary element form, we are able to overcome many shortcomings of previous neural solvers, such as learning trivial solutions and balancing loss terms between the domain and boundary, at the cost of introducing a large integral containing a singular kernel. We handle this singularity by locally transforming the integral into polar coordinates and applying a numerical quadrature. We also show that previous neural solver sampling methods are unable to minimize the PDE residual, and propose a variational adaptive sampling method. This technique is able to reduce mean absolute error by 5 times, while keeping training time constant. Extensive scaling and ablation studies are performed to justify our method. Results show that our method learns a charge distribution within 1.2 $pC/m^2$ of mean absolute error from a classical BEM solver, while using 25 times fewer rectangular elements.
Primary Area: applications to physical sciences (physics, chemistry, biology, etc.)
Code Of Ethics: I acknowledge that I and all co-authors of this work have read and commit to adhering to the ICLR Code of Ethics.
Submission Guidelines: I certify that this submission complies with the submission instructions as described on https://iclr.cc/Conferences/2025/AuthorGuide.
Anonymous Url: I certify that there is no URL (e.g., github page) that could be used to find authors’ identity.
No Acknowledgement Section: I certify that there is no acknowledgement section in this submission for double blind review.
Submission Number: 11610
Loading