Derivative Informed Learning of Exchange-Correlation Functionals
Eike S. Eberhard, Luca Anthony Thiede, Abdulrahman Aldossary, Andreas Burger, Nicholas Gao, Vignesh Bhethanabotla, Alan Aspuru-Guzik, Stephan Günnemann
Abstract
Machine-learned (ML) XC functionals aim to replace human-designed density functional approximations by learning directly from reference data, but they still do not consistently outperform traditional -scaling hybrid functionals. We therefore study a hybrid-distillation setting, where -scaling semilocal ML-XC functionals are trained to reproduce B3LYP/def2-SVP targets. We introduce Derivative Informed XC-Loss (DI-Loss), a loss that incorporates additional information from the reference hybrid functional by supervising first and second derivatives of the energy on the Grassmannian of admissible density matrices. Rather than only matching the self-consistent fixed point, DI-Loss aligns the local first- and second-order response of the learned functional with that of the target functional. Across four evaluated architectures, DI-Loss consistently improves the main energy metrics. Averaged uniformly across architectures, the total-energy MAE decreases by 66% relative to energy and density supervision alone. The density-sensitive mean-field energy metric improves from 1.2 to 0.8 mEh on average, while dipole and density errors do not improve uniformly. We further show that densities from the distilled functionals reduce hybrid-functional SCF iterations by up to 55%. In downstream TDDFT calculations, Hessian supervision improves excited-state predictions, with XCdiff reducing the mean excitation-energy MAE by 24-35% across molecule sizes on QM40.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext c3106fd1-edd3-4a5d-95d4-d57ed629e2f6Builds on4
- Stabilizing Equilibrium Models by Jacobian RegularizationShaojie Bai, Vladlen Koltun, J. Zico KolterICML 2021 · 80 citations
- Efficient and Equivariant Graph Networks for Predicting Quantum HamiltonianHaiyang Yu, Zhao Xu, Xiaofeng Qian, Xiaoning Qian et al.ICML 2023 · 51 citations
- Deep equilibrium networks are sensitive to initialization statisticsAtish Agarwala, Samuel S. SchoenholzICML 2022 · 12 citations
- Learning Equivariant Non-Local Electron Density FunctionalsNicholas Gao, Eike Eberhard, Stephan GünnemannICLR 2025
Related papers
- Towards Fast, Specialized Machine Learning Force Fields: Distilling Foundation Models via Energy HessiansIshan Amin, Sanjeev Raja, Aditi S. KrishnapriyanICLR 2025 · 5 citations
- Coupled Cluster con MoLe: Molecular Orbital Learning for Neural WavefunctionsLuca Anthony Thiede, Abdulrahman Aldossary, Andreas Burger, Jorge Campos-Gonzalez-Angulo et al.ICML 2026 · 2 citations
- Enhancing the Scalability and Applicability of Kohn-Sham Hamiltonians for Molecular SystemsYunyang Li, Zaishuo Xia, Lin Huang, Xinran Wei et al.ICLR 2025
- PFT: Phonon Fine-tuning for Machine Learned Interatomic PotentialsTeddy Koker, Abhijeet Gangan, Mit Kotak, Jaime Marian et al.ICML 2026
- Self-Consistency Training for Density-Functional-Theory Hamiltonian PredictionHe Zhang, Chang Liu, Zun Wang, Xinran Wei et al.ICML 2024 · 13 citations
