Saved in:
Bibliographic Details
Main Authors: Tom, Gary, Lo, Stanley, Corapi, Samantha, Aspuru-Guzik, Alan, Sanchez-Lengeling, Benjamin
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.09290
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913998848393216
author Tom, Gary
Lo, Stanley
Corapi, Samantha
Aspuru-Guzik, Alan
Sanchez-Lengeling, Benjamin
author_facet Tom, Gary
Lo, Stanley
Corapi, Samantha
Aspuru-Guzik, Alan
Sanchez-Lengeling, Benjamin
contents Bayesian optimization (BO) has become an indispensable tool for autonomous decision-making across diverse applications from autonomous vehicle control to accelerated drug and materials discovery. With the growing interest in self-driving laboratories, BO of chemical systems is crucial for machine learning (ML) guided experimental planning. Typically, BO employs a regression surrogate model to predict the distribution of unseen parts of the search space. However, for the selection of molecules, picking the top candidates with respect to a distribution, the relative ordering of their properties may be more important than their exact values. In this paper, we introduce Rank-based Bayesian Optimization (RBO), which utilizes a ranking model as the surrogate. We present a comprehensive investigation of RBO's optimization performance compared to conventional BO on various chemical datasets. Our results demonstrate similar or improved optimization performance using ranking models, particularly for datasets with rough structure-property landscapes and activity cliffs. Furthermore, we observe a high correlation between the surrogate ranking ability and BO performance, and this ability is maintained even at early iterations of BO optimization when using ranking surrogate models. We conclude that RBO is an effective alternative to regression-based BO, especially for optimizing novel chemical compounds.
format Preprint
id arxiv_https___arxiv_org_abs_2410_09290
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Ranking over Regression for Bayesian Optimization and Molecule Selection
Tom, Gary
Lo, Stanley
Corapi, Samantha
Aspuru-Guzik, Alan
Sanchez-Lengeling, Benjamin
Machine Learning
Artificial Intelligence
Bayesian optimization (BO) has become an indispensable tool for autonomous decision-making across diverse applications from autonomous vehicle control to accelerated drug and materials discovery. With the growing interest in self-driving laboratories, BO of chemical systems is crucial for machine learning (ML) guided experimental planning. Typically, BO employs a regression surrogate model to predict the distribution of unseen parts of the search space. However, for the selection of molecules, picking the top candidates with respect to a distribution, the relative ordering of their properties may be more important than their exact values. In this paper, we introduce Rank-based Bayesian Optimization (RBO), which utilizes a ranking model as the surrogate. We present a comprehensive investigation of RBO's optimization performance compared to conventional BO on various chemical datasets. Our results demonstrate similar or improved optimization performance using ranking models, particularly for datasets with rough structure-property landscapes and activity cliffs. Furthermore, we observe a high correlation between the surrogate ranking ability and BO performance, and this ability is maintained even at early iterations of BO optimization when using ranking surrogate models. We conclude that RBO is an effective alternative to regression-based BO, especially for optimizing novel chemical compounds.
title Ranking over Regression for Bayesian Optimization and Molecule Selection
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2410.09290