The Mechanics of Delta Learning: Target Design for Generalizable Scientific Machine Learning
Organizations: Department of Physical and Environmental Sciences, University of Toronto Scarborough Alliance for AI-Accelerated Materials Discovery (A3MD) Toronto, Ontario, Canada · Alliance for AI-Accelerated Materials Discovery (A3MD) Toronto, Ontario, Canada · Department of Chemistry, University of Toronto Department of Physical and Environmental Sciences, University of Toronto Scarborough Alliance for AI-Accelerated Materials Discovery (A3MD) Toronto, Ontario, Canada
Abstract
In scientific machine learning, -learning trains models on residual errors relative to physical baselines, assuming that more accurate baselines with smaller residual scales inherently improve downstream performance. Here, we demonstrate that residual scale alone is an insufficient heuristic for learnability. Evaluating molecular graph neural networks on total energy targets, we show that complex local descriptor baselines can yield small residual targets that are disproportionately rough within architecture-informed proxy spaces and harder to learn relative to their scale. Conversely, semi-empirical baseline reduces both scale and normalized roughness, improving in-domain and out-of-domain prediction. We introduce scale-normalized graph Dirichlet roughness () as a pre-training diagnostic for residual learnability and establish baseline complementarity as a core target-design principle, elevating target space formulation alongside model architecture as a key axis for scientific machine learning.
Figures & tables
| Baseline model | Input features | Fitting approach | Physical/representation target |
|---|---|---|---|
| BoB | Typed-bond counts ( RDKit ) | Linear / MLP | Compositional and functional-group variation |
| PRE | Pairwise distances ( ) | Linear / MLP | Local two-body geometry (SchNet proxy) |
| PRE+A | Distances and angles ( ) | Linear / MLP | Local three-body geometry (GemNet proxy) |
| xTB | Atomic identities and molecular coordinates | Semiempirical QM | Non-local electronic and higher-order interactions |
| DFTB | Atomic identities and molecular coordinates | Semiempirical QM | Molecule-wide electronic structure; supporting comparison with xTB |