Proxima: Accelerating the Integration of Machine Learning in Atomistic Simulations (2021)

This paper shows how to use adaptive AI to maintain an accuracy constraint in scientific simulations.

In this case, the science simulation can dynamically report
its accuracy. We use a control system to determine when to use computationally expensive (but accurate) physics codes and when to use computationally inexpensive (but less accurate) AI surrogates (of the physics codes).

The control system uses AI to achieve large reductions in simulation time when the science result will not be adversely affected, and it uses the physics code when it determines AI would add too much error to the simulation.

This may be one of the first examples of controlling the accuracy of AI inference, which is only enabled by the application of self-aware computing to AI-based systems.

Read abstract

Atomistic-scale simulations are prominent scientific applications that require the repetitive execution of a computationally expensive routine to calculate a system’s potential energy. Prior work shows that these expensive routines can be replaced with a machine-learned surrogate approximation to accelerate the simulation at the expense of the overall accuracy. The exact balance of speed and accuracy depends on the specific configuration of the surrogate-modeling workflow and the science itself, and prior work leaves it up to the scientist to find a configuration that delivers the required accuracy for their science problem. Unfortunately, due to the underlying system dynamics, it is rare that a single surrogate configuration presents an optimal accuracy/latency trade-off for the entire simulation. In practice, scientists must choose conservative configurations so that accuracy is always acceptable, forgoing possible acceleration. As an alternative, we propose Proxima, a systematic and automated method for dynamically tuning a surrogate-modeling configuration in response to real-time feedback from the ongoing simulation. Proxima estimates the uncertainty of applying a surrogate approximation in each step of an iterative simulation. Using this information, the specific surrogate configuration can be adjusted dynamically to ensure maximum speedup while sustaining a required accuracy metric. We evaluate Proxima using a Monte Carlo sampling application and find that Proxima respects a wide range of user-defined accuracy goals while achieving speedups of 1.02–5.5X relative to a standard