General Binding Affinity Guidance for Diffusion Models in Structure-Based Drug Design

Abstract

Structure-based drug design (SBDD) aims to generate ligands that bind strongly and specifically to target protein pockets. Recent diffusion models have advanced SBDD by capturing the distributions of atomic positions and types, yet they often underemphasize binding affinity control during generation. To address this limitation, we introduce \textbf{\textnormal{\textbf{BADGER}}}, a general \textbf{binding-affinity guidance framework for diffusion models in SBDD}. \textnormal{\textbf{BADGER} }incorporates binding affinity awareness through two complementary strategies: (1) \textit{classifier guidance}, which applies gradient-based affinity signals during sampling in a plug-and-play fashion, and (2) \textit{classifier-free guidance}, which integrates affinity conditioning directly into diffusion model training. Together, these approaches enable controllable ligand generation guided by binding affinity. \textnormal{\textbf{BADGER} } can be added to any diffusion model and achieves up to a \textbf{60\% improvement in ligand--protein binding affinity} of sampled molecules over prior methods. Furthermore, we extend the framework to \textbf{multi-constraint diffusion guidance}, jointly optimizing for binding affinity, drug-likeness (QED), and synthetic accessibility (SA) to design realistic and synthesizable drug candidates.

References in corpus (9)

• Score-Based Generative Modeling through Stochastic Differential Equations
• GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generation
• Structure-based Drug Design with Equivariant Diffusion Models
• 3D Equivariant Diffusion for Target-Aware Molecule Generation and Affinity Prediction
• Generating 3D Molecules for Target Protein Binding
• DecompDiff: Diffusion Models with Decomposed Priors for Structure-Based Drug Design
• Benchmarking Generated Poses: How Rational is Structure-based Drug Design with Generative Models?
• Rethinking Specificity in SBDD: Leveraging Delta Score and Energy-Guided Diffusion
• Binding-Adaptive Diffusion Models for Structure-Based Drug Design