Graph Transformer Networks for Accurate Band Structure Prediction: An End-to-End Approach

Weiyi Gong, Tao Sun, Hexin Bai, Jeng-Yuan Tsai, Haibin Ling +1 more
11/25/2024
cond-mat.mtrl-scics.LG

No official code found. Best implementation: qmatyanlab/Bandformer (8 stars, MIT, last commit 2026-02-06).

Abstract

Predicting electronic band structures from crystal structures is crucial for understanding structure-property correlations in materials science. First-principles approaches are accurate but computationally intensive. Recent years, machine learning (ML) has been extensively applied to this field, while existing ML models predominantly focus on band gap predictions or indirect band structure estimation via solving predicted Hamiltonians. An end-to-end model to predict band structure accurately and efficiently is still lacking. Here, we introduce a graph Transformer-based end-to-end approach that directly predicts band structures from crystal structures with high accuracy. Our method leverages the continuity of the k-path and treat continuous bands as a sequence. We demonstrate that our model not only provides accurate band structure predictions but also can derive other properties (such as band gap, band center, and band dispersion) with high accuracy. We verify the model performance on large and diverse datasets.

View on arXivView PDF

Code Implementations(1)

qmatyanlab/BandformerCommunity58%
MIT

Implementation of Graph Transformer Networks for Accurate Band Structure Prediction: An End-to-End Approach

81PythonAug 28, 20258 months agoMIT

Similar Papers

Complete and Efficient Graph Transformers for Crystal Material Property Prediction

81% match3 repos8032024

Transformer-Enhanced Variational Autoencoder for Crystal Structure Prediction

81% match2025

Universal crystal material property prediction via multi-view geometric fusion in graph transformers

80% match2025

Universal Machine Learning Kohn-Sham Hamiltonian for Materials

80% match1 repo2172024

Band-gap regression with architecture-optimized message-passing neural networks

80% match1 repo12023

Enhancing material property prediction with ensemble deep graph convolutional networks

80% match2024

Cite this paper

@article{gong2024graph,
  title  = {Graph Transformer Networks for Accurate Band Structure Prediction: An End-to-End Approach},
  author = {Weiyi Gong and Tao Sun and Hexin Bai and Jeng-Yuan Tsai and Haibin Ling and Qimin Yan},
  year   = {2024},
  eprint = {2411.16483},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2411.16483v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2411.16483.svg)](https://codeofpaper.com/paper/2411.16483)

Discussion