Efficient Learning of Accurate Surrogates for Simulations of Complex Systems

• DOI: 10.48550/arxiv.2207.12855
• Publication Date: 2024-05-17
• Authors: A. Diaw; M. McKerns; I. Sagert; L. G. Stanton; M. S. Murillo
• Keywords: FOS: Computer and information sciences; Computer Science - Machine Learning; Nuclear Theory; FOS: Physical sciences; Computational Physics (physics.comp-ph); 01 natural sciences; Physics - Plasma Physics; Machine Learning (cs.LG); Nuclear Theory (nucl-th); Plasma Physics (physics.plasm-ph); Physics - Data Analysis, Statistics and Probability; 0103 physical sciences; Physics - Computational Physics; Data Analysis, Statistics and Probability (physics.data-an)

13 pages, 6 figures, submitted to Nature Machine Intelligence<br/>Machine learning methods are increasingly used to build computationally inexpensive surrogates for complex physical models. The predictive capability of these surrogates suffers when data are noisy, sparse, or time-dependent. As we are interested in finding a surrogate that provides valid predictions of any potential future model evaluations, we introduce an online learning method empowered by optimizer-driven sampling. The method has two advantages over current approaches. First, it ensures that all turning points on the model response surface are included in the training data. Second, after any new model evaluations, surrogates are tested and "retrained" (updated) if the "score" drops below a validity threshold. Tests on benchmark functions reveal that optimizer-directed sampling generally outperforms traditional sampling methods in terms of accuracy around local extrema, even when the scoring metric favors overall accuracy. We apply our method to simulations of nuclear matter to demonstrate that highly accurate surrogates for the nuclear equation of state can be reliably auto-generated from expensive calculations using a few model evaluations.<br/>