用强化学习优化口袋感知扩散模型,实现多属性药物分子设计
Fine-tuning Pocket-Aware Diffusion Models via Denoising Policy Optimization

- 将去噪过程建模为马尔可夫决策过程,通过奖励信号引导生成
- 在CrossDocked2020上实现-8.5 kcal/mol的结合亲和力,优于基线
- 适合需同时优化亲和力、类药性、可合成性和多样性的药物研发
基于结构的药物设计因口袋感知的3D生成模型而加速,但多数方法仅拟合训练分布,在真实药物发现中难以满足多重需求。近期研究关注结构基础分子优化(SBMO),旨在精细调控多个指定分子属性。本文提出DEPPA,一种基于去噪扩散策略优化的SBMO方法,通过强化学习微调预训练的口袋感知扩散模型。DEPPA支持对结合亲和力、类药性、可合成性和多样性等多属性的优化。将预训练模型的反向去噪过程建模为多步马尔可夫决策过程,以最终生成配体分子的属性作为奖励信号。微调过程中引入粗粒度去噪调度器,提升优化效率与效果。在CrossDocked2020基准上的实验表明,DEPPA在结合亲和力(Vina Score -8.5 kcal/mol)、类药性和多样性方面优于基线,可合成性表现具有竞争力。源代码见https://github.com/xy9485/DePPA。
原文摘要 · Abstract (English)
Structure-based drug design has been accelerated by pocket-aware 3D generative models, yet most methods primarily fit the training distribution and may fall short of satisfying multiple properties required in real-world therapeutic drug discovery. Recently, increasing attention has focused on structure-based molecule optimization (SBMO), which targets fine-grained control over multiple specified molecular properties. In this paper, we present DEPPA, a novel SBMO approach building upon Denoising Diffusion Policy Optimization for fine-tuning a pre-trained pocket-aware diffusion model via reinforcement learning. DEPPA enables optimization over multiple properties, including binding affinity, drug-likeness, synthesizability and diversity. We formulate the reverse denoising process of the pretrained pocket-aware diffusion model as a multi-step Markov Decision Process, where the desired properties that serve as reward signals are evaluated on the final generated ligand molecules. DEPPA incorporates a coarse denoising scheduler during the RL fine-tuning to achieve efficient and effective molecule optimization. Experimental results on the CrossDocked2020 benchmark demonstrate that DEPPA outperforms baselines in binding affinity (Vina Score -8.5 kcal/mol), drug-likeness and diversity while exhibiting competitive performance in synthesizability. The source code is available at https://github.com/xy9485/DePPA .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。