Assessment of Machine Learning Reliability Methods for Quantifying the Applicability Domain of QSAR Regression Models

作者:Toplak Marko; Mocnik Rok; Polajnar Matija; Bosnic Zoran; Carlsson Lars; Hasselgren Catrin; Demsar Janez; Boyer Scott; Zupan Blaz*; Stalring Jonna
来源:Journal of Chemical Information and Modeling, 2014, 54(2): 431-441.
DOI:10.1021/ci4006595

摘要

The vastness of chemical space and the relatively small coverage by experimental data recording molecular properties require us to identify subspaces, or domains, for which we can confidently apply QSAR models. The prediction of QSAR models in these domains is reliable, and potential subsequent investigations of such compounds would find that the predictions closely match the experimental values. Standard approaches in QSAR assume that predictions are more reliable for compounds that are "similar" to those in subspaces with denser experimental data. Here, we report on a study of an alternative set of techniques recently proposed in the machine learning community. These methods quantify prediction confidence through estimation of the prediction error at the point of interest. Our study includes 20 public QSAR data sets with continuous response and assesses the quality of 10 reliability scoring methods by observing their correlation with prediction error. We show that these new alternative approaches can outperform standard reliability scores that rely only on similarity to compounds in the training set. The results also indicate that the quality of reliability scoring methods in sensitive to data set characteristics and to the regression method used in QSAR. We demonstrate that at the cost of increased computational complexity these dependencies can be leveraged by integration of scores from various reliability estimation approaches. The reliability estimation techniques described in this paper have been implemented in an open source add-on package (https://bitbucket.org/biolab/orange-reliability) to the Orange data mining suite.

  • 出版日期2014-2