NeFut Logo NeFut
中 Admin Login

[CS.AI] Inference-Time Projection for Physically Valid Biomolecular Diffusion Models

Published at: 2026-10-07 22:00 Last updated: 2026-10-08 01:25
#algorithm #AI #Machine Learning

AlphaFold 3‑style co‑folding models can predict biomolecular complexes with high structural accuracy, yet many of their outputs violate physical constraints such as chain overlap at interfaces, distorted ligand bond lengths and angles, non‑planar rings, and inverted stereocenters. Existing solutions either steer the sampler with physics‑aware potentials—drastically increasing sampling cost and memory so that inference on large complexes becomes infeasible—or fine‑tune the model, which is time‑consuming and tied to a specific architecture. We observe that physical validity can be fully verified at inference time from quantities already available to the sampler. Consequently we treat it as a constrained inference problem and introduce two closed‑form projection operators applied to the diffusion model’s denoised coordinate estimate $\hat{x}_0$:

We evaluated the approach on two independently developed models, Boltz‑2 and OpenFold‑3, across five benchmarks (CASP15, CASP16, the PoseBusters monomer and complex sets, and the Boltz physical‑validity test set). The method recovers perfect physical validity while preserving structural‑accuracy and ligand‑placement metrics, with negligible runtime and memory overhead. This provides a practical, model‑agnostic route to physically valid all‑atom structure prediction.

Review

Original Source: https://arxiv.org/abs/2610.07037

[h] Back to Home