Human-in-the-loop active learning for goal-oriented molecule generation

Beskrivning

Abstract Machine learning (ML) systems have enabled the modelling of quantitative structure–property relationships (QSPR) and structure-activity relationships (QSAR) using existing experimental data to predict target properties for new molecules. These property predictors hold significant potential in accelerating drug discovery by guiding generative artificial intelligence (AI) agents to explore desired chemical spaces. However, they often struggle to generalize due to the limited scope of the training data. When optimized by generative agents, this limitation can result in the generation of molecules with artificially high predicted probabilities of satisfying target properties, which subsequently fail experimental validation. To address this challenge, we propose an adaptive approach that integrates active learning (AL) and iterative feedback to refine property predictors, thereby improving the outcomes of their optimization by generative AI agents. Our method leverages the Expected Predictive Information Gain (EPIG) criterion to select additional molecules for evaluation by an oracle. This process aims to provide the greatest reduction in predictive uncertainty, enabling more accurate model evaluations of subsequently generated molecules. Recognizing the impracticality of immediate wet-lab or physics-based experiments due to time and logistical constraints, we propose leveraging human experts for their cost-effectiveness and domain knowledge to effectively augment property predictors, bridging gaps in the limited training data. Empirical evaluations through both simulated and real human-in-the-loop experiments demonstrate that our approach refines property predictors to better align with oracle assessments. Additionally, we observe improved accuracy of predicted properties as well as improved drug-likeness among the top-ranking generated molecules. Scientific contribution We present an adaptable framework that integrates AL and human expertise to refine property predictors for goal-oriented molecule generation. This approach is robust to noise in human feedback and ensures that navigating chemical space with human-refined predictors leverages human insights to identify molecules that not only satisfy predicted property profiles but also score highly on oracle models. Additionally, it prioritizes practical characteristics such as drug-likeness, synthetic accessibility, and a favorable balance between exploring diverse chemical space and exploiting similarity to existing training data.
Visa mer

Publiceringsår

2024

Typ av data

Upphovspersoner

Department of Computer Science

Eva Nittinger - Upphovsperson

Janosch Menke - Upphovsperson

Jon Paul Janet - Upphovsperson

Julien Martinelli - Upphovsperson

Markus Heinonen Orcid -palvelun logo - Upphovsperson

Mikhail Kabeshov - Upphovsperson

Ola Engkvist - Upphovsperson

Samuel Kaski Orcid -palvelun logo - Upphovsperson

Yasmine Nahal Orcid -palvelun logo - Upphovsperson

AstraZeneca Sweden - Medarbetare

Chalmers University of Technology - Medarbetare

Univ Manchester, Jodrell Bank Centre for Astrophysics, University of Manchester, Sch Phys & Astron, Jodrell Bank Ctr Astrophys - Medarbetare

Université de Bordeaux - Medarbetare

figshare - Utgivare

Projekt

Övriga uppgifter

Vetenskapsområden

Data- och informationsvetenskap

Språk

Öppen tillgång

Öppet

Licens

Creative Commons Attribution 4.0 International (CC BY 4.0)

Nyckelord

Ämnesord

Temporal täckning

undefined

Relaterade till denna forskningsdata