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

Abstract

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.

References in corpus (25)

• A Comprehensive Survey on Graph Neural Networks
• Gaussian Approximation Potentials: the accuracy of quantum mechanics, without the electrons
• Crystal Graph Convolutional Neural Networks for an Accurate and Interpretable Prediction of Material Properties
• On representing chemical environments
• SchNet - a deep learning architecture for molecules and materials
• E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentials
• Graph Networks as a Universal Machine Learning Framework for Molecules and Crystals
• A Spectral Analysis Method for Automated Generation of Quantum-Accurate Interatomic Potentials
• PhysNet: A Neural Network for Predicting Energies, Forces, Dipole Moments and Partial Charges
• A Universal Graph Deep Learning Interatomic Potential for the Periodic Table
• A Performance and Cost Assessment of Machine Learning Interatomic Potentials
• Atomistic Line Graph Neural Network for Improved Materials Property Predictions
• Automatic Selection of Atomic Fingerprints and Reference Configurations for Machine-Learning Potentials
• WACSF - Weighted Atom-Centered Symmetry Functions as Descriptors in Machine Learning Potentials
• MACE: Higher Order Equivariant Message Passing Neural Networks for Fast and Accurate Force Fields
• Global Attention based Graph Convolutional Neural Networks for Improved Materials Property Prediction
• Evaluation of the MACE Force Field Architecture: from Medicinal Chemistry to Materials Science
• Optimizing many-body atomic descriptors for enhanced computational performance of machine learning based interatomic potentials
• Metrics for measuring distances in configuration spaces
• Representing Long-Range Context for Graph Neural Networks with Global Attention
• A fingerprint based metric for measuring similarities of crystalline structures
• e3nn: Euclidean Neural Networks
• SchNetPack 2.0: A neural network toolbox for atomistic machine learning
• Manifolds of quasi-constant SOAP and ACSF fingerprints and the resulting failure to machine learn four-body interactions
• Capturing long-range interaction with reciprocal space neural network