Salvato in:
Dettagli Bibliografici
Autori principali: Shen, Yuchen, Zhang, Chenhao, Fu, Sijie, Zhou, Chenghui, Washburn, Newell, Póczos, Barnabás
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:https://arxiv.org/abs/2410.06502
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929752092180480
author Shen, Yuchen
Zhang, Chenhao
Fu, Sijie
Zhou, Chenghui
Washburn, Newell
Póczos, Barnabás
author_facet Shen, Yuchen
Zhang, Chenhao
Fu, Sijie
Zhou, Chenghui
Washburn, Newell
Póczos, Barnabás
contents Recent advances in diffusion models have shown remarkable potential in the conditional generation of novel molecules. These models can be guided in two ways: (i) explicitly, through additional features representing the condition, or (ii) implicitly, using a property predictor. However, training property predictors or conditional diffusion models requires an abundance of labeled data and is inherently challenging in real-world applications. We propose a novel approach that attenuates the limitations of acquiring large labeled datasets by leveraging domain knowledge from quantum chemistry as a non-differentiable oracle to guide an unconditional diffusion model. Instead of relying on neural networks, the oracle provides accurate guidance in the form of estimated gradients, allowing the diffusion process to sample from a conditional distribution specified by quantum chemistry. We show that this results in more precise conditional generation of novel and stable molecular structures. Our experiments demonstrate that our method: (1) significantly reduces atomic forces, enhancing the validity of generated molecules when used for stability optimization; (2) is compatible with both explicit and implicit guidance in diffusion models, enabling joint optimization of molecular properties and stability; and (3) generalizes effectively to molecular optimization tasks beyond stability optimization.
format Preprint
id arxiv_https___arxiv_org_abs_2410_06502
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Chemistry-Inspired Diffusion with Non-Differentiable Guidance
Shen, Yuchen
Zhang, Chenhao
Fu, Sijie
Zhou, Chenghui
Washburn, Newell
Póczos, Barnabás
Machine Learning
Artificial Intelligence
Recent advances in diffusion models have shown remarkable potential in the conditional generation of novel molecules. These models can be guided in two ways: (i) explicitly, through additional features representing the condition, or (ii) implicitly, using a property predictor. However, training property predictors or conditional diffusion models requires an abundance of labeled data and is inherently challenging in real-world applications. We propose a novel approach that attenuates the limitations of acquiring large labeled datasets by leveraging domain knowledge from quantum chemistry as a non-differentiable oracle to guide an unconditional diffusion model. Instead of relying on neural networks, the oracle provides accurate guidance in the form of estimated gradients, allowing the diffusion process to sample from a conditional distribution specified by quantum chemistry. We show that this results in more precise conditional generation of novel and stable molecular structures. Our experiments demonstrate that our method: (1) significantly reduces atomic forces, enhancing the validity of generated molecules when used for stability optimization; (2) is compatible with both explicit and implicit guidance in diffusion models, enabling joint optimization of molecular properties and stability; and (3) generalizes effectively to molecular optimization tasks beyond stability optimization.
title Chemistry-Inspired Diffusion with Non-Differentiable Guidance
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2410.06502