Designing compact training sets for data-driven molecular property prediction through optimal exploitation and exploration

摘要

In this paper, we consider the problem of designing a compact training set comprising the most informa- tive molecules from a specified library to build data-driven molecular property models. Specifically, using (i) sparse generalized group additivity and (ii) kernel ridge regression as two representative classes of models, we propose a method combining rigorous model-based design of experiments and cheminformatics- based diversity-maximizing subset selection within the ε-greedy framework to systematically minimize the amount of data needed to train these models. We demonstrate the effectiveness of the algorithm on vari- ous databases, including QM7, NIST, and a dataset of surface intermediates for calculating thermodynamic properties (heat of atomization and enthalpy of formation). For sparse group additive models, a balance be- tween exploration (diversity-maximizing selection) and exploitation (D-optimality selection) leads to learn- ing with a fraction (sometimes as little as 15%) of the data to achieve similar accuracy to five-fold cross val- idation on the entire set. On the other hand, our results indicate that kernel methods prefer diversity- maximizing selection.