A universal machine learning model for the electronic density of states

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: How, Wei Bin, Febrer, Pol, Chong, Sanggyu, Mazitov, Arslan, Bigi, Filippo, Kellner, Matthias, Pozdnyakov, Sergey, Ceriotti, Michele
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909984038584320
author How, Wei Bin
Febrer, Pol
Chong, Sanggyu
Mazitov, Arslan
Bigi, Filippo
Kellner, Matthias
Pozdnyakov, Sergey
Ceriotti, Michele
author_facet How, Wei Bin
Febrer, Pol
Chong, Sanggyu
Mazitov, Arslan
Bigi, Filippo
Kellner, Matthias
Pozdnyakov, Sergey
Ceriotti, Michele
contents In the last few years several ``universal'' interatomic potentials have appeared, using machine-learning approaches to predict energy and forces of atomic configurations with arbitrary composition and structure, with an accuracy often comparable with that of the electronic-structure calculations they are trained on. Here we demonstrate that these generally-applicable models can also be built to predict explicitly the electronic structure of materials and molecules. We focus on the electronic density of states (DOS), and develop PET-MAD-DOS, a rotationally unconstrained transformer model built on the Point Edge Transformer (PET) architecture, and trained on the Massive Atomistic Diversity (MAD) dataset. We demonstrate our model's predictive abilities on samples from diverse external datasets, showing also that the DOS can be further manipulated to obtain accurate band gap predictions. A fast evaluation of the DOS is especially useful in combination with molecular simulations probing matter in finite-temperature thermodynamic conditions. To assess the accuracy of PET-MAD-DOS in this context, we evaluate the ensemble-averaged DOS and the electronic heat capacity of three technologically relevant systems: lithium thiophosphate (LPS), gallium arsenide (GaAs), and a high entropy alloy (HEA). By comparing with bespoke models, trained exclusively on system-specific datasets, we show that our universal model achieves semi-quantitative agreement for all these tasks. Furthermore, we demonstrate that fine-tuning can be performed using a small fraction of the bespoke data, yielding models that are comparable to, and sometimes better than, fully-trained bespoke models.
format Preprint
id arxiv_https___arxiv_org_abs_2508_17418
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle A universal machine learning model for the electronic density of states
How, Wei Bin
Febrer, Pol
Chong, Sanggyu
Mazitov, Arslan
Bigi, Filippo
Kellner, Matthias
Pozdnyakov, Sergey
Ceriotti, Michele
Chemical Physics
Materials Science
In the last few years several ``universal'' interatomic potentials have appeared, using machine-learning approaches to predict energy and forces of atomic configurations with arbitrary composition and structure, with an accuracy often comparable with that of the electronic-structure calculations they are trained on. Here we demonstrate that these generally-applicable models can also be built to predict explicitly the electronic structure of materials and molecules. We focus on the electronic density of states (DOS), and develop PET-MAD-DOS, a rotationally unconstrained transformer model built on the Point Edge Transformer (PET) architecture, and trained on the Massive Atomistic Diversity (MAD) dataset. We demonstrate our model's predictive abilities on samples from diverse external datasets, showing also that the DOS can be further manipulated to obtain accurate band gap predictions. A fast evaluation of the DOS is especially useful in combination with molecular simulations probing matter in finite-temperature thermodynamic conditions. To assess the accuracy of PET-MAD-DOS in this context, we evaluate the ensemble-averaged DOS and the electronic heat capacity of three technologically relevant systems: lithium thiophosphate (LPS), gallium arsenide (GaAs), and a high entropy alloy (HEA). By comparing with bespoke models, trained exclusively on system-specific datasets, we show that our universal model achieves semi-quantitative agreement for all these tasks. Furthermore, we demonstrate that fine-tuning can be performed using a small fraction of the bespoke data, yielding models that are comparable to, and sometimes better than, fully-trained bespoke models.
title A universal machine learning model for the electronic density of states
topic Chemical Physics
Materials Science
url https://arxiv.org/abs/2508.17418