Design of experiments for the calibration of history-dependent models via deep reinforcement learning and an enhanced Kalman filter

• Published in: arXiv.org 2022-09
• Main Authors: Villarreal, Ruben, Vlassis, Nikolaos N, Phan, Nhon N, Catanach, Tommie A, Jones, Reese E, Trask, Nathaniel A, Kramer, Sharlotte L B, Sun, WaiChing
• Format: Article
• Language: English
• Subjects: Algorithms; Calibration; Data acquisition; Decision trees; Deep learning; Design of experiments; Kalman filters; Machine learning; Markov processes; Mechanical tests
• ISSN: 2331-8422

Experimental data is costly to obtain, which makes it difficult to calibrate complex models. For many models an experimental design that produces the best calibration given a limited experimental budget is not obvious. This paper introduces a deep reinforcement learning (RL) algorithm for design of experiments that maximizes the information gain measured by Kullback-Leibler (KL) divergence obtained via the Kalman filter (KF). This combination enables experimental design for rapid online experiments where traditional methods are too costly. We formulate possible configurations of experiments as a decision tree and a Markov decision process (MDP), where a finite choice of actions is available at each incremental step. Once an action is taken, a variety of measurements are used to update the state of the experiment. This new data leads to a Bayesian update of the parameters by the KF, which is used to enhance the state representation. In contrast to the Nash-Sutcliffe efficiency (NSE) index, which requires additional sampling to test hypotheses for forward predictions, the KF can lower the cost of experiments by directly estimating the values of new data acquired through additional actions. In this work our applications focus on mechanical testing of materials. Numerical experiments with complex, history-dependent models are used to verify the implementation and benchmark the performance of the RL-designed experiments.