ABOPD

Antibodies recognize their targets through complementarity-determining regions (CDRs), six variable loops that form the antigen-binding interface. Diffusion models can now generate CDR sequences and backbones jointly against a fixed antigen, but they inherit a mismatch built into denoising training: the loss is computed on noisy states produced by perturbing a native structure, while sampling walks through states the model itself generated. For the long, flexible CDR-H3 loop, small backbone errors compound along that reverse trajectory, and the resulting geometry drifts away from anything the training distribution ever covered.

ABOPD (Antibody CDR Design via On-Policy Distillation) is a post-training framework that closes this gap. It first pretrains a hybrid base checkpoint, called H-DiffAb, that mixes single-CDR, multi-CDR, and all-six-CDR masking regimes in one model. It then builds a backbone-aware teacher from that base by granting it privileged access to native backbone geometry, and distills the teacher's per-step coordinate transitions into the student on states the student actually visits during its own reverse-denoising rollouts. The supervision is therefore dense, structural, and on-policy rather than a single sequence-level reward or a fixed offline dataset of teacher outputs.

The framework comes from a multi-institution Chinese academic collaboration led by the Shanghai Innovation Institute. It extends the DiffAb line of antigen-conditioned CDR diffusion models. Where antibody design systems such as the all-atom immune receptor model IgGM2, the retrieval-augmented binder generator RADiAnce, and the de novo nanobody pipeline Germinal each introduce a new generative architecture, ABOPD instead targets the post-training stage of an existing one.

#Key Features

• On-policy structural distillation: Supervision is applied to states sampled from the student's own denoising trajectory, correcting the model where its errors accumulate rather than on synthetically noised natives.
• Privileged-information teacher: Lightweight MLP adapters inject native backbone geometry into the base encoders during teacher adaptation, giving the teacher information the student never sees at inference.
• Hybrid multi-CDR pretraining: The H-DiffAb base mixes single-CDR, multi-CDR, and all-six-CDR masking in one checkpoint, serving both single-loop and full-paratope design.
• Coordinate-transition targets: Distillation matches Cα coordinate transitions step by step, which reduces to matching student and teacher noise predictions.
• Reusable released checkpoint: One trained model is applied to new antigen targets without per-target retraining, with Apache-2.0 code derived from DiffAb.

#Technical Details

H-DiffAb pretraining samples a target residue set at each update from a hybrid distribution weighted 0.25 single-CDR, 0.40 multi-CDR, and 0.35 all-six-CDR, noising and reconstructing only residues in that set, at roughly eight times the compute budget of standard DiffAb training on SAbDab-derived antibody-antigen complexes. The teacher's privileged descriptor comprises Cα-centered Cartesian coordinates and validity masks for the N, Cα, C, O, and observed Cβ atoms; explicit target residue labels and atoms beyond Cβ are withheld to limit information leakage. Training then combines an offline denoising anchor on forward-noised reference states with an on-policy coordinate-matching term computed on student rollouts, weighted uniformly across timesteps.

On RAbD CDR-H3 generation, ABOPD lowers backbone RMSD from 2.37 Å to 1.95 Å over the H-DiffAb base while raising amino-acid recovery from 36.35% to 37.31%, with marginal gains in lDDT and TM-score, and it outperforms both supervised fine-tuning and offline-distillation controls trained from the same base. In simultaneous six-CDR redesign, CDR-H3 RMSD drops from 2.94 Å to 2.48 Å, intra-CDR steric clashes fall from 0.93% to 0.42%, and CDR-antigen clashes fall from 1.58% to 0.67%. Timestep-resolved analysis shows the teacher's advantage emerging around the midpoint of denoising and peaking in the late refinement steps, consistent with the claim that trajectory drift is a geometry-refinement problem.

#Applications

ABOPD is aimed at antibody engineers who need higher-fidelity backbone geometry from a generative CDR designer. Because the released checkpoint conditions on a fixed antigen structure, it can redesign CDR-H3 against a chosen epitope or co-design all six loops in one pass, serving as the proposal stage of affinity maturation, humanization, or epitope-focused de novo campaigns. Its broader appeal is methodological: the recipe is a general post-training procedure for diffusion-based structure generators, applicable wherever recursive sampling drift degrades designed geometry. Inference runs from the released code and checkpoint; there is no hosted API.

#Impact

ABOPD imports on-policy distillation — a technique developed for post-training language models — into structural biomolecular diffusion, showing that dense per-step geometric supervision beats both supervised fine-tuning and offline teacher imitation on the same base model. The gains are concentrated in backbone accuracy: side-chain quality does not improve, with Jensen-Shannon divergence on side-chain dihedrals rising from 0.2451 to 0.2566 and the interface-improvement rate sitting marginally below baseline. The authors also note that dense trajectory supervision can concentrate generation on structurally reliable modes at the cost of exploration, a real tension for a task where novelty matters. As a preprint, the results are in-silico RAbD benchmarks awaiting peer review and experimental characterization of the designed antibodies.