Harmonic Torsional Diffusion for Protein-Ligand Flexible Docking

arXiv:2608.20366 · q-bio.BM, cs.LG · Submitted 2026-06-18 · Read on arXiv

q-bio.BM, cs.LG

Submitted: 2026-06-18

Updated: 2026-10-18

Comments: Accepted at NeurIPS 2026

License: http://creativecommons.org/licenses/by/4.0/

The gist: Molecular docking requires reasoning jointly about ligand pose and protein flexibility.

Abstract

Molecular docking requires reasoning jointly about ligand pose and protein flexibility. Most diffusion-based docking models predict torsional updates with generic Euclidean heads that ignore the periodic geometry of angular variables. This mismatch is especially limiting in flexible docking, where ligand conformations and pocket side chains co-adapt to form the bound complex. Here, we introduce Harmony, a harmonic torsional diffusion framework for flexible protein-ligand docking. Harmony parameterizes ligand and side-chain torsional score fields as derivatives of learned harmonic potentials on the circle, which are noised mode by mode in the Fourier basis owing to the properties of variance-exploding diffusion on the torus. This construction makes periodicity explicit and gives the model a frequency-aware inductive bias over rotameric motion. On the PDBBind benchmark, Harmony improves ligand pose accuracy and pocket all-atom reconstruction over recent flexible docking methods. On PoseBusters, it improves the physical validity of generated complexes. Case studies on EBNA1 and KRAS G12D illustrate the method's behavior on a polar and a shallow binding site, respectively. Together, these results indicate that aligning the score parameterization with the geometry of the diffusion process is a simple and effective lever for improving flexible docking.

Related papers