Skip to yearly menu bar Skip to main content


Poster

Composing Unbalanced Flows for Flexible Docking and Relaxation

Gabriele Corso · Vignesh Ram Somnath · Noah Getz · Regina Barzilay · Tommi Jaakkola · Andreas Krause

Hall 3 + Hall 2B #7
[ ]
Fri 25 Apr midnight PDT — 2:30 a.m. PDT
 
Oral presentation: Oral Session 3B
Thu 24 Apr 7:30 p.m. PDT — 9 p.m. PDT

Abstract:

Diffusion models have emerged as a successful approach for molecular docking, but they often cannot model protein flexibility or generate nonphysical poses. We argue that both these challenges can be tackled by framing the problem as a transport between distributions. Still, existing paradigms lack the flexibility to define effective maps between such complex distributions. To address this limitation we propose Unbalanced Flow Matching, a generalization of Flow Matching (FM) that allows trading off sample efficiency with approximation accuracy and enables more accurate transport. Empirically, we apply Unbalanced FM on flexible docking and structure relaxation, demonstrating our ability to model protein flexibility and generate energetically favorable poses. On the PDBBind docking benchmark, our method FlexDock improves the docking performance while increasing the proportion of energetically favorable poses from 30% to 73%.

Live content is unavailable. Log in and register to view live content