Site-Net: using global self-attention and real-space supercells to capture long-range interactions in crystal structures



Moran, Michael, Gaultois, Michael W ORCID: 0000-0003-2172-2507, Gusev, Vladimir V ORCID: 0000-0002-2815-607X and Rosseinsky, Matthew J ORCID: 0000-0002-1910-2483
(2023) Site-Net: using global self-attention and real-space supercells to capture long-range interactions in crystal structures DIGITAL DISCOVERY, 2 (5). pp. 1297-1310. ISSN 2635-098X, 2635-098X

Access the full-text of this item by clicking on the Open Access link.

Abstract

Site-Net is a transformer architecture that models the periodic crystal structures of inorganic materials as a labelled point set of atoms and relies entirely on global self-attention and geometric information to guide learning. Site-Net processes standard crystallographic information files to generate a large real-space supercell, and the importance of interactions between all atomic sites is flexibly learned by the model for the prediction task presented. The attention mechanism is probed to reveal Site-Net can learn long-range interactions in crystal structures, and that specific attention heads become specialised to deal with primarily short- or long-range interactions. We perform a preliminary hyperparameter search and train Site-Net using a single graphics processing unit (GPU), and show Site-Net achieves state-of-the-art performance on a standard band gap regression task.

Item Type: Article
Uncontrolled Keywords: 46 Information and Computing Sciences, 4611 Machine Learning
Depositing User: Symplectic Admin
Date Deposited: 07 Aug 2024 14:00
Last Modified: 16 Jun 2026 20:14
DOI: 10.1039/d3dd00005b
Open Access URL: https://doi.org/10.1039/D3DD00005B
Related Websites:
URI: https://livrepository.liverpool.ac.uk/id/eprint/3183427
Disclaimer: The University of Liverpool is not responsible for content contained on other websites from links within repository metadata. Please contact us if you notice anything that appears incorrect or inappropriate.