Equivariant graph neural network surrogates for predicting the properties of relaxed atomic configurations

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Holber, Jamie, Srivastava, Siddhartha, Garikipati, Krishna
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910080256966656
author Holber, Jamie
Srivastava, Siddhartha
Garikipati, Krishna
author_facet Holber, Jamie
Srivastava, Siddhartha
Garikipati, Krishna
contents Density functional theory (DFT) calculations determine the relaxed atomic positions and lattice parameters that minimize the formation energy of a structure. We present an equivariant graph neural network (EGNN) model to predict the outcome of DFT calculations for structures of interest. Cluster expansions are a well established approach for representing the formation energies. However, traditional cluster expansions are limited in their ability to handle variations from a fixed lattice, including interstitial atoms, amorphous materials, and materials with multiple structures. EGNNs offer a more flexible framework that inherently respects the symmetry of the system without being reliant on a particular lattice. In this work, we present the mathematical framework and the results of training for lithium cobalt oxide (LCO) at various compositions of lithium and arrangements of the lithium atoms. Our results demonstrate that the EGNN can accurately predict quantities outside the training set including the largest atomic displacements, the strain tensor and energy, and the formation energy providing greater insight into the system being studied without the need for more DFT calculations.
format Preprint
id arxiv_https___arxiv_org_abs_2505_08121
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Equivariant graph neural network surrogates for predicting the properties of relaxed atomic configurations
Holber, Jamie
Srivastava, Siddhartha
Garikipati, Krishna
Materials Science
Computational Physics
Density functional theory (DFT) calculations determine the relaxed atomic positions and lattice parameters that minimize the formation energy of a structure. We present an equivariant graph neural network (EGNN) model to predict the outcome of DFT calculations for structures of interest. Cluster expansions are a well established approach for representing the formation energies. However, traditional cluster expansions are limited in their ability to handle variations from a fixed lattice, including interstitial atoms, amorphous materials, and materials with multiple structures. EGNNs offer a more flexible framework that inherently respects the symmetry of the system without being reliant on a particular lattice. In this work, we present the mathematical framework and the results of training for lithium cobalt oxide (LCO) at various compositions of lithium and arrangements of the lithium atoms. Our results demonstrate that the EGNN can accurately predict quantities outside the training set including the largest atomic displacements, the strain tensor and energy, and the formation energy providing greater insight into the system being studied without the need for more DFT calculations.
title Equivariant graph neural network surrogates for predicting the properties of relaxed atomic configurations
topic Materials Science
Computational Physics
url https://arxiv.org/abs/2505.08121