acceptodds
Under review as a conference paper at ICLR 2027

Guiding Where the Antibody Binds: Amortised Cross-Subsample Particle Steering for Antibody–Antigen Co-Folding

Abstract

Co-folding models have advanced antibody-antigen structure prediction, but a dominant failure mode persists. They often dock the antibody onto the wrong epitope. Docking has two problems. A sampler must produce an acceptable complex at the correct interface, and a verifier must then rank it above the alternatives. We address the first, site coverage, since a verifier cannot select an epitope the sampler never proposed. Drawing more seeds or samples adds variety around the pose the model already favours, but does not relocate it. We introduce CAST (Cross-subsample Amortised Site Tilting), which steers diffusion trajectories toward a binding site using a reward built from a candidate epitope map, training-free and with no reference structure. The sampler approximates an ideal target whose tilt provably changes the site marginal but not the pose at a site. Amortising alignment-subsample latents into one steered pool cuts the cost of exploring them from to for a pool of particles. At a matched structures and conditionings, CAST holds an acceptable pose on of the difficulty-balanced CFA50 targets against for Best-of-, and raises mean max DockQ from to , a gain of . On targets that a -seed run fails to reach CAPRI-medium quality, CAST reaches mean max DockQ at structures and brings of the to an acceptable pose against at the same budget. Best-of- flattens, spending the compute and trailing by . Raising the latent count alongside the samples carries this to , beyond every Best-of- budget. On CFA50 CAST also matches the best Best-of- allocation of structures, at less compute. Such steering is a reallocation, changing little where the base model already succeeds and much where it does not.

Then back it, or bet against it.

Related papers

Open the market on this paper to see 7 more related papers.