Rapid prediction of phonon structure and properties using the atomistic line graph neural network (ALIGNN)

The phonon density of states (DOS) summarizes the lattice vibrational modes supported by a structure and gives access to rich information about the material's stability, thermodynamic constants, and thermal transport coefficients. Here, we present an atomistic line graph neural network (ALIGNN) model for the prediction of the phonon density of states and the derived thermal and thermodynamic properties. The model is trained on a database of over 14 000 phonon spectra included in the joint automated repository for various integrated simulations: density functional theory (jarvis-dft) database. The model predictions are shown to capture the spectral features of the phonon density of states, effectively categorize dynamical stability, and lead to accurate predictions of DOS-derived thermal and thermodynamic properties, including heat-capacity ${C}_{\mathrm{V}}$, vibrational entropy ${S}_{\mathrm{vib}}$, and the isotopic phonon-scattering rate ${\ensuremath{\tau}}_{\mathrm{i}}^{\ensuremath{-}1}$. A comparison of room temperature thermodynamic property predictions reveals that the DOS-mediated ALIGNN model provides superior predictions when compared to a direct deep-learning prediction of these material properties as well as predictions based on analytic simplifications of the phonon DOS, including the Debye or Born--von Karman models. Finally, the ALIGNN model is used to predict the phonon spectra and properties for about 40 000 additional materials listed in the jarvis-dft database, which are validated as far as possible against other open-sourced high-throughput DFT phonon databases.