Unified Sampling and Ranking for Protein Docking with DFMDock

Abstract

Recent diffusion-based approaches to protein-protein docking typically decouple structure generation from decoy ranking. We introduce DFMDock (Denoising Force Matching for Docking), a unified diffusion model that integrates generative sampling and energy-based ranking through physically motivated supervision. DFMDock predicts both denoising forces and a scalar energy, trained using force matching and energy contrastive objectives. The predicted forces guide the reverse diffusion process, while the energy enables decoy ranking without relying on a separately trained confidence model. On the Docking Benchmark 5, DFMDock achieves a 32.8% Oracle success rate and 5.3% Top-1 success rate, outperforming DiffDock-PP (16.2% and 4.3%, respectively). Unlike co-folding models, DFM-Dock does not require MSAs and generalizes to unseen targets. In decoy ranking, its learned energy function outperforms Rosetta energy and model-derived confidence scores, producing funnel-shaped energy landscapes enriched for near-native structures. These results suggest DFMDock as an efficient and physically grounded approach to diffusion-based protein docking.