Publication: A decision-theoretic approach to the evaluation of machine learning algorithms in computational drug discovery
Issued Date
2019-11-01
Resource Type
ISSN
14602059
13674803
13674803
Other identifier(s)
2-s2.0-85074962955
Rights
Mahidol University
Rights Holder(s)
SCOPUS
Bibliographic Citation
Bioinformatics. Vol.35, No.22 (2019), 4656-4663
Suggested Citation
Oliver P. Watson, Isidro Cortes-Ciriano, Aimee R. Taylor, James A. Watson A decision-theoretic approach to the evaluation of machine learning algorithms in computational drug discovery. Bioinformatics. Vol.35, No.22 (2019), 4656-4663. doi:10.1093/bioinformatics/btz293 Retrieved from: https://repository.li.mahidol.ac.th/handle/20.500.14594/50051
Research Projects
Organizational Units
Authors
Journal Issue
Thesis
Title
A decision-theoretic approach to the evaluation of machine learning algorithms in computational drug discovery
Abstract
© The Author(s) 2019. Published by Oxford University Press. 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.