Paper ID: 2411.02740

An information-matching approach to optimal experimental design and active learning

Yonatan Kurniawan (1), Tracianne B. Neilsen (1), Benjamin L. Francis (2), Alex M. Stankovic (3), Mingjian Wen (4), Ilia Nikiforov (5), Ellad B. Tadmor (5), Vasily V. Bulatov (6), Vincenzo Lordi (6), Mark K. Transtrum (1, 2, and 3) ((1) Brigham Young University, Provo, UT, USA, (2) Achilles Heel Technologies, Orem, UT, USA, (3) SLAC National Accelerator Laboratory, Menlo Park, CA, USA, (4) University of Houston, Houston, TX, USA, (5) University of Minnesota, Minneapolis, MN, USA, (6) Lawrence Livermore National Laboratory)

The efficacy of mathematical models heavily depends on the quality of the training data, yet collecting sufficient data is often expensive and challenging. Many modeling applications require inferring parameters only as a means to predict other quantities of interest (QoI). Because models often contain many unidentifiable (sloppy) parameters, QoIs often depend on a relatively small number of parameter combinations. Therefore, we introduce an information-matching criterion based on the Fisher Information Matrix to select the most informative training data from a candidate pool. This method ensures that the selected data contain sufficient information to learn only those parameters that are needed to constrain downstream QoIs. It is formulated as a convex optimization problem, making it scalable to large models and datasets. We demonstrate the effectiveness of this approach across various modeling problems in diverse scientific fields, including power systems and underwater acoustics. Finally, we use information-matching as a query function within an Active Learning loop for material science applications. In all these applications, we find that a relatively small set of optimal training data can provide the necessary information for achieving precise predictions. These results are encouraging for diverse future applications, particularly active learning in large machine learning models.

Submitted: Nov 5, 2024