Direct prediction of phonon density of states with Euclidean neural networks

Abstract

Machine learning has demonstrated great power in materials design, discovery, and property prediction. However, despite the success of machine learning in predicting discrete properties, challenges remain for continuous property prediction. The challenge is aggravated in crystalline solids due to crystallographic symmetry considerations and data scarcity. Here we demonstrate the direct prediction of phonon density of states using only atomic species and positions as input. We apply Euclidean neural networks, which by construction are equivariant to 3D rotations, translations, and inversion and thereby capture full crystal symmetry, and achieve high-quality prediction using a small training set of $\sim 10^3$ examples with over 64 atom types. Our predictive model reproduces key features of experimental data and even generalizes to materials with unseen elements,and is naturally suited to efficiently predict alloy systems without additional computational cost. We demonstrate the potential of our network by predicting a broad number of high phononic specific heat capacity materials. Our work indicates an efficient approach to explore materials' phonon structure, and can further enable rapid screening for high-performance thermal storage materials and phonon-mediated superconductors.

arXiv comment21 pages total, 5 main figures + 16 supplementary figures. To appear in Advanced Science (2021)

References in corpus (16)

• Phonons and related properties of extended systems from density-functional perturbation theory
• Automatic chemical design using a data-driven continuous representation of molecules
• Crystal Graph Convolutional Neural Networks for an Accurate and Interpretable Prediction of Material Properties
• A General-Purpose Machine Learning Framework for Predicting Properties of Inorganic Materials
• Electron-phonon interactions from first principles
• Graph Networks as a Universal Machine Learning Framework for Molecules and Crystals
• Machine Learning of Accurate Energy-Conserving Molecular Force Fields
• Universal Fragment Descriptors for Predicting Electronic Properties of Inorganic Crystals
• Active Learning of Uniformly Accurate Inter-atomic Potentials for Materials Simulation
• Machine learning modeling of superconducting critical temperature
• Accurate Interatomic Force Fields via Machine Learning with Covariant Kernels
• Unsupervised machine learning and band topology
• Origin of the large phonon band-gap in SrTiO3 and the vibrational signatures of ferroelectricity in ATiO3 perovskite: First principles lattice dynamics and inelastic neutron scattering of PbTiO3, BaTiO3 and SrTiO3
• Phonon Density of States and Anharmonicity of UO2
• Detection of Topological Materials with Machine Learning
• Finding Symmetry Breaking Order Parameters with Euclidean Neural Networks

Cited by in corpus (14)

• Recent Advances and Applications of Deep Learning Methods in Materials Science
• A Physics-Guided Neural Network Framework for Elastic Plates: Comparison of Governing Equations-Based and Energy-Based Approaches
• Machine Learning on Neutron and X-Ray Scattering
• Theory for Equivariant Quantum Neural Networks
• Density of States Prediction for Materials Discovery via Contrastive Learning from Probabilistic Embeddings
• Formula graph self-attention network for representation-domain independent materials discovery
• Robust Machine Learning Inference from X-ray Absorption Near Edge Spectra through Featurization
• Surrogate- and invariance-boosted contrastive learning for data-scarce applications in science
• Representations and Strategies for Transferable Machine Learning Models in Chemical Discovery
• Artificial Intelligence for Science in Quantum, Atomistic, and Continuum Systems
• Decoding Structure-Spectrum Relationships with Physically Organized Latent Spaces
• Mechanism of Local Lattice Distortion Effects on Vacancy Migration Barriers in FCC Alloys
• Accelerating superconductor discovery through tempered deep learning of the electron-phonon spectral function
• Crystalline Material Discovery in the Era of Artificial Intelligence