Transferable SCF-Acceleration through Solver-Aligned Initialization Learning

· Source: Machine Learning · Field: Science & Research — Physical Sciences & Chemistry, Mathematics & Computational Sciences · Depth: Expert, quick

Summary

Solver-Aligned Initialization Learning (SAIL) is a machine learning method designed to accelerate Self-Consistent Field (SCF) calculations in quantum chemistry by predicting improved initial guesses. Traditional matrix-prediction models often fail to extrapolate effectively to larger molecules, leading to degraded convergence. SAIL addresses this by differentiating through the SCF solver end-to-end, resolving a supervision problem in both Hamiltonian and density matrix models. The method introduces the Effective Relative Iteration Count (ERIC) to accurately account for Fock-build overhead. On the QM40 dataset, SAIL reduced ERIC by 37% for PBE, 33% for SCAN, and 27% for B3LYP, significantly outperforming prior methods. For QMugs molecules 10 times larger than the training data, SAIL achieved a 1.25x wall-time speedup at the hybrid level of theory, extending ML-based SCF acceleration to large, drug-like molecules.

Key takeaway

For computational chemists performing Self-Consistent Field (SCF) calculations on large molecules, adopting Solver-Aligned Initialization Learning (SAIL) can significantly reduce computation time. SAIL's ability to extrapolate effectively to molecules 4x to 10x larger than its training data means you can achieve substantial wall-time speedups, particularly for drug-like molecules and hybrid levels of theory, making complex simulations more feasible.

Key insights

SAIL accelerates SCF calculations by learning solver-aligned initial guesses, improving extrapolation to larger molecules.

Principles

Method

SAIL differentiates through the SCF solver end-to-end to optimize initial guess predictions, resolving supervision issues for Hamiltonian and density matrix models, and uses ERIC for accurate overhead accounting.

In practice

Topics

Best for: AI Scientist, Research Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Machine Learning.