Retrained Models and Scripts for Aluminum at 298K and 933K

Retrained Models and Scripts for Aluminum at 298K and 933K

Authors

- Fiedler, Lenz (HZDR/CASUS)
- Cangi, Attila (HZDR/CASUS)

Affiliations:

HZDR - Helmholtz-Zentrum Dresden-Rossendorf
CASUS - Center for Advanced Systems Understanding

Data set description

This data sets contains models, scripts and inference results for aluminum at room temperature and the melting point. Training data, hyperparameters and general methodology follow Ref. [1]. The models here are retrained versions of the ones discussed in this publication, and therefore retrained versions of the models contained in Ref. [2]. As such, data from Ref. [2] has been used. Only a subset of models contained in Ref. [1] have been retrained, namely the room temperature model, one liquid and one solid melting point model with four training snapshot each, and the final melting point hybrid model (six training snapshots per phase). Furthermore, for both the hybrid melting temperature model and the room temperature model, multiple models with different initializations were trained.

All models were trained with the MALA code [3] version 1.2.1. They show better accuracy than their original counterparts, as they were trained using the inter-snapshot shuffling algorithm first discussed for the MALA code in Ref. [4].

[1] - "Accelerating finite-temperature Kohn-Sham density functional theory with deep neural networks", Physical Review B, doi.org/10.1103/PhysRevB.104.035120
[2] - "RODARE", doi.org/10.14278/rodare.2485 (v1.0.0)
[3] - "MALA", Zenodo, doi.org/10.5281/zenodo.5557254
[4] - "Machine learning the electronic structure of matter across temperatures", Physical Review B, doi.org/10.1103/PhysRevB.108.125146

Contents

- The models themselves, labeled as either Al298K or Al933K, given as one .zip file per model
    - For 933K, additionally "liquid", "solid" and "hybrid" denotes the training data set
    - For ensembles, a running index denotes the number in the ensemble
- Inference results, given as a single .zip file
    - For all models, band energy and total free energy results are given in the .csv format
        - The columns in these files correspond to "Calculated via DFT LDOS", "Calculated via ML-DFT LDOS", "Calculated via Kohn-Sham system", respectively
    - For some models, additionally the predicted electronic density and density of states on select snapshots is given
- Shuffling, training and testing scripts, given as a single .zip file
    - Scripts are ready-to-use with suitable MALA installation, however, correct data paths have to be filled in