EOSnet: Embedded Overlap Structures for Graph Neural Networks in Predicting Material Properties

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tao, Shuo, Zhu, Li
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916583528464384
author Tao, Shuo
Zhu, Li
author_facet Tao, Shuo
Zhu, Li
contents Graph Neural Networks (GNNs) have emerged as powerful tools for predicting material properties, yet they often struggle to capture many-body interactions and require extensive manual feature engineering. Here, we present EOSnet (Embedded Overlap Structures for Graph Neural Networks), a novel approach that addresses these limitations by incorporating Gaussian Overlap Matrix (GOM) fingerprints as node features within the GNN architecture. Unlike models that rely on explicit angular terms or human-engineered features, EOSnet efficiently encodes many-body interactions through orbital overlap matrices, providing a rotationally invariant and transferable representation of atomic environments. The model demonstrates superior performance across various materials property prediction tasks, achieving particularly notable results in properties sensitive to many-body interactions. For band gap prediction, EOSnet achieves a mean absolute error of 0.163 eV, surpassing previous state-of-the-art models. The model also excels in predicting mechanical properties and classifying materials, with 97.7\% accuracy in metal/non-metal classification. These results demonstrate that embedding GOM fingerprints into node features enhances the ability of GNNs to capture complex atomic interactions, making EOSnet a powerful tool for materials discovery and property prediction.
format Preprint
id arxiv_https___arxiv_org_abs_2411_02579
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle EOSnet: Embedded Overlap Structures for Graph Neural Networks in Predicting Material Properties
Tao, Shuo
Zhu, Li
Materials Science
Graph Neural Networks (GNNs) have emerged as powerful tools for predicting material properties, yet they often struggle to capture many-body interactions and require extensive manual feature engineering. Here, we present EOSnet (Embedded Overlap Structures for Graph Neural Networks), a novel approach that addresses these limitations by incorporating Gaussian Overlap Matrix (GOM) fingerprints as node features within the GNN architecture. Unlike models that rely on explicit angular terms or human-engineered features, EOSnet efficiently encodes many-body interactions through orbital overlap matrices, providing a rotationally invariant and transferable representation of atomic environments. The model demonstrates superior performance across various materials property prediction tasks, achieving particularly notable results in properties sensitive to many-body interactions. For band gap prediction, EOSnet achieves a mean absolute error of 0.163 eV, surpassing previous state-of-the-art models. The model also excels in predicting mechanical properties and classifying materials, with 97.7\% accuracy in metal/non-metal classification. These results demonstrate that embedding GOM fingerprints into node features enhances the ability of GNNs to capture complex atomic interactions, making EOSnet a powerful tool for materials discovery and property prediction.
title EOSnet: Embedded Overlap Structures for Graph Neural Networks in Predicting Material Properties
topic Materials Science
url https://arxiv.org/abs/2411.02579