Force sketching in MD: Can we replace multi-head committees with screening methods for faster force uncertainty compute?
Abstract
Machine-learned interatomic potentials simulate molecules far more cheaply than quantum chemistry, but an automated discovery loop only benefits if the model can say when it is guessing - that is what decides when to call the expensive oracle. Committee disagreement is the usual signal, and a multi-head committee makes it nearly free for energies: M heads share one trunk, so one forward pass gives all M energies. Forces are not free. Forces are gradients, and each head's forces need their own backward pass, so force disagreement - the quantity acquisition actually uses - costs about M times more. We ask whether a handful of cheap random probes can estimate it instead, on a pretrained eight-head committee across four molecular systems. The answer is no for replacement and yes for screening. Even with six of the seven probes the exact computation would use, the best estimator recovers only 72-80% of the genuinely most uncertain structures. But a calibrated filter built from the same score safely skips 84-87% of the expensive computations while still catching 96-98%, beating cheaper alternatives in 11 of 12 comparisons. The wall-clock benefit is regime-dependent: 1.85x in batch, only 1.11x one structure at a time. The decision-quality result is robust; the speed result is not.