Electrostatics from Laplacian Eigenbasis for Neural Network Interatomic Potentials

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhdanov, Maksim, Kurenkov, Vladislav
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917017607471104
author Zhdanov, Maksim
Kurenkov, Vladislav
author_facet Zhdanov, Maksim
Kurenkov, Vladislav
contents In this work, we introduce Phi-Module, a universal plugin module that enforces Poisson's equation within the message-passing framework to learn electrostatic interactions in a self-supervised manner. Specifically, each atom-wise representation is encouraged to satisfy a discretized Poisson's equation, making it possible to acquire a potential ϕ and corresponding charges \r{ho} linked to the learnable Laplacian eigenbasis coefficients of a given molecular graph. We then derive an electrostatic energy term, crucial for improved total energy predictions. This approach integrates seamlessly into any existing neural potential with insignificant computational overhead. Our results underscore how embedding a first-principles constraint in neural interatomic potentials can significantly improve performance while remaining hyperparameter-friendly, memory-efficient, and lightweight in training. Code will be available at https://github.com/dunnolab/phi-module.
format Preprint
id arxiv_https___arxiv_org_abs_2505_14606
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Electrostatics from Laplacian Eigenbasis for Neural Network Interatomic Potentials
Zhdanov, Maksim
Kurenkov, Vladislav
Machine Learning
Computational Physics
In this work, we introduce Phi-Module, a universal plugin module that enforces Poisson's equation within the message-passing framework to learn electrostatic interactions in a self-supervised manner. Specifically, each atom-wise representation is encouraged to satisfy a discretized Poisson's equation, making it possible to acquire a potential ϕ and corresponding charges \r{ho} linked to the learnable Laplacian eigenbasis coefficients of a given molecular graph. We then derive an electrostatic energy term, crucial for improved total energy predictions. This approach integrates seamlessly into any existing neural potential with insignificant computational overhead. Our results underscore how embedding a first-principles constraint in neural interatomic potentials can significantly improve performance while remaining hyperparameter-friendly, memory-efficient, and lightweight in training. Code will be available at https://github.com/dunnolab/phi-module.
title Electrostatics from Laplacian Eigenbasis for Neural Network Interatomic Potentials
topic Machine Learning
Computational Physics
url https://arxiv.org/abs/2505.14606