<p>Generating novel molecules with higher properties than the training space, namely the out-of-distribution generation, is important for <i>de novo</i> drug design. However, it is not easy for distribution learning-based models, for example diffusion models, to solve this challenge as these methods are designed to fit the distribution of training data as close as possible. In this paper, we show that Bayesian flow network, especially ChemBFN model, is capable of intrinsically generating high quality out-of-distribution samples that meet several scenarios. A reinforcement learning strategy is added to the ChemBFN and a controllable ordinary differential equation solver-like generating process is employed that accelerate the sampling processes. Most importantly, we introduce a semi-autoregressive strategy during training and inference that enhances the model performance and surpass the state-of-the-art models. A theoretical analysis of out-of-distribution generation in ChemBFN with semi-autoregressive approach is included as well.</p><p><b>Scientific contribution</b></p><p>Benchmarked on both small molecule and protein generative tasks, ChemBFN method showed outstanding out-of-distribution performance without complex modification, proving its feasibility of exploring chemical spaces outside the training data and its value as a tool to accelerate drug design and materials discovery.</p> Graphic Abstract <p></p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Sampling out-of-distribution chemical spaces via Bayesian flow

  • Nianze Tao,
  • Minori Abe

摘要

Generating novel molecules with higher properties than the training space, namely the out-of-distribution generation, is important for de novo drug design. However, it is not easy for distribution learning-based models, for example diffusion models, to solve this challenge as these methods are designed to fit the distribution of training data as close as possible. In this paper, we show that Bayesian flow network, especially ChemBFN model, is capable of intrinsically generating high quality out-of-distribution samples that meet several scenarios. A reinforcement learning strategy is added to the ChemBFN and a controllable ordinary differential equation solver-like generating process is employed that accelerate the sampling processes. Most importantly, we introduce a semi-autoregressive strategy during training and inference that enhances the model performance and surpass the state-of-the-art models. A theoretical analysis of out-of-distribution generation in ChemBFN with semi-autoregressive approach is included as well.

Scientific contribution

Benchmarked on both small molecule and protein generative tasks, ChemBFN method showed outstanding out-of-distribution performance without complex modification, proving its feasibility of exploring chemical spaces outside the training data and its value as a tool to accelerate drug design and materials discovery.

Graphic Abstract