Surrogate-Based Black-Box Optimization Method for Costly Molecular Properties

Abstract

AI-assisted molecular optimization is a very active research field as it is expected to provide the next-generation drugs and molecular materials. An important difficulty is that the properties to be optimized rely on costly evaluations. Machine learning methods are investigated with success to predict these properties, but show generalization issues on less known areas of the chemical space. We propose here a surrogate-based black box optimization method, to tackle jointly the optimization and machine learning problems. It consists in optimizing the expected improvement of the surrogate of a molecular property using an evolutionary algorithm. The surrogate is defined as a Gaussian Process Regression (GPR) model, learned on a relevant area of the search space with respect to the property to be optimized. We show that our approach can successfully optimize a costly property of interest much faster than a purely metaheuristic approach.

arXiv commentSubmitted to ICTAI 2021

References in corpus (10)

• Scikit-learn: Machine Learning in Python
• On representing chemical environments
• Fast and Accurate Modeling of Molecular Atomization Energies with Machine Learning
• DScribe: Library of Descriptors for Machine Learning in Materials Science
• Comparing molecules and solids across structural and alchemical space
• Machine Learning Unifies the Modelling of Materials and Molecules
• Deep learning for molecular design - a review of the state of the art
• COCO: A Platform for Comparing Continuous Optimizers in a Black-Box Setting
• Fast and Accurate Uncertainty Estimation in Chemical Machine Learning
• Machine learning of molecular properties: locality and active learning