Crystal Fractional Graph Neural Network for Energy Prediction of High-Entropy Alloys

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Kotama, Takanori, Huang, Yang
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914545675534336
author Kotama, Takanori
Huang, Yang
author_facet Kotama, Takanori
Huang, Yang
contents High-entropy alloys (HEAs) have attracted growing attention for their exceptional mechanical and thermal properties arising from complex atomic configurations. In this paper, we propose crystal fractional graph neural network for predicting the energy of high-entropy alloys by explicitly integrating both local atomic environments and global compositional information. The model consists of three components: a crystal graph neural network, which employs graph attention network layers to learn local interactions among 16 on-site atoms within the crystal lattice; fractional neural network, a fully connected network that embeds the global fraction of constituent elements; and feature fusion neural network, which fuses the outputs of the two submodels to predict the total crystal energy. We train the model on a dataset of 1,049 crystal structures and validate it on 198 quaternary structures, optimizing all hyperparameters via Optuna. Our results show that our model achieves an RMSE comparable to first-principles calculations and maintains high accuracy even for low-energy configurations. However, the model exhibits limitations in handling large crystal cells, which we aim to address in future work to extend its applicability to more complex systems.
format Preprint
id arxiv_https___arxiv_org_abs_2605_08103
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Crystal Fractional Graph Neural Network for Energy Prediction of High-Entropy Alloys
Kotama, Takanori
Huang, Yang
Computational Physics
Materials Science
Artificial Intelligence
High-entropy alloys (HEAs) have attracted growing attention for their exceptional mechanical and thermal properties arising from complex atomic configurations. In this paper, we propose crystal fractional graph neural network for predicting the energy of high-entropy alloys by explicitly integrating both local atomic environments and global compositional information. The model consists of three components: a crystal graph neural network, which employs graph attention network layers to learn local interactions among 16 on-site atoms within the crystal lattice; fractional neural network, a fully connected network that embeds the global fraction of constituent elements; and feature fusion neural network, which fuses the outputs of the two submodels to predict the total crystal energy. We train the model on a dataset of 1,049 crystal structures and validate it on 198 quaternary structures, optimizing all hyperparameters via Optuna. Our results show that our model achieves an RMSE comparable to first-principles calculations and maintains high accuracy even for low-energy configurations. However, the model exhibits limitations in handling large crystal cells, which we aim to address in future work to extend its applicability to more complex systems.
title Crystal Fractional Graph Neural Network for Energy Prediction of High-Entropy Alloys
topic Computational Physics
Materials Science
Artificial Intelligence
url https://arxiv.org/abs/2605.08103