Accelerating the calculation of electron-phonon coupling strength with machine learning

Yang Zhong,Shixu Liu,Binhua Zhang,Zhiguo Tao,Yuting Sun,Weibin Chu,Xin-Gao Gong,Ji-Hui Yang,Hongjun Xiang
DOI: https://doi.org/10.1038/s43588-024-00668-7
2024-08-08
Abstract:The calculation of electron-phonon couplings (EPCs) is essential for understanding various fundamental physical properties, including electrical transport, optical and superconducting behaviors in materials. However, obtaining EPCs through fully first-principles methods is notably challenging, particularly for large systems or when employing advanced functionals. Here we introduce a machine learning framework to accelerate EPC calculations by utilizing atomic orbital-based Hamiltonian matrices and gradients predicted by an equivariant graph neural network. We demonstrate that our method not only yields EPC values in close agreement with first-principles results but also enhances calculation efficiency by several orders of magnitude. Application to GaAs using the Heyd-Scuseria-Ernzerhof functional reveals the necessity of advanced functionals for accurate carrier mobility predictions, while for the large Kagome crystal CsV3Sb5, our framework reproduces the experimentally observed double domes in pressure-induced superconducting phase diagrams. This machine learning framework offers a powerful and efficient tool for the investigation of diverse EPC-related phenomena in complex materials.
What problem does this paper attempt to address?