A decision-theoretic approach to the evaluation of machine learning algorithms in computational drug discovery

Watson, O; Cortes-Ciriano, I; Taylor, A; Watson, J

Journal article

A decision-theoretic approach to the evaluation of machine learning algorithms in computational drug discovery

Abstract:: Motivation: Artificial intelligence, trained via machine learning (e.g. neural nets, random forests) or computational statistical algorithms (e.g. support vector machines, ridge regression), holds much promise for the improvement of small-molecule drug discovery. However, small-molecule structure-activity data are high dimensional with low signal-to-noise ratios and proper validation of predictive methods is difficult. It is poorly understood which, if any, of the currently available machine learning algorithms will best predict new candidate drugs.

Results: The quantile-activity bootstrap is proposed as a new model validation framework using quantile splits on the activity distribution function to construct training and testing sets. In addition, we propose two novel rank-based loss functions which penalize only the out-of-sample predicted ranks of high-activity molecules. The combination of these methods was used to assess the performance of neural nets, random forests, support vector machines (regression) and ridge regression applied to 25 diverse high-quality structure-activity datasets publicly available on ChEMBL. Model validation based on random partitioning of available data favours models that overfit and ‘memorize’ the training set, namely random forests and deep neural nets. Partitioning based on quantiles of the activity distribution correctly penalizes extrapolation of models onto structurally different molecules outside of the training data. Simpler, traditional statistical methods such as ridge regression can outperform state-of-the-art machine learning methods in this setting. In addition, our new rank-based loss functions give considerably different results from mean squared error highlighting the necessity to define model optimality with respect to the decision task at hand.

Publication status:: Published

Peer review status:: Peer reviewed

Actions

Email

Email this record

Send the bibliographic details of this record to your email address.

Your Email
Please enter the email address that the record information will be sent to.

-
Your message (optional)
Please add any additional information to be included within the email.
Cite

Cite this record

APA Style

Watson, O., Cortes-Ciriano, I., Taylor, A., & Watson, J. (2019). A decision-theoretic approach to the evaluation of machine learning algorithms in computational drug discovery . Bioinformatics, 35(22), 4656–4663.

MLA Style

Watson, O., et al. “A Decision-Theoretic Approach to the Evaluation of Machine Learning Algorithms in Computational Drug Discovery .” Bioinformatics, vol. 35, no. 22, Oxford University Press, 2019, pp. 4656–63.

Chicago Style

Watson, O, I Cortes-Ciriano, A Taylor, and J Watson. 2019. “A Decision-Theoretic Approach to the Evaluation of Machine Learning Algorithms in Computational Drug Discovery .” Bioinformatics 35 (22): 4656–63.
Share
Print