How AI Designs, Ranks, and Tests New Antibodies

Peter Bubenik ยท Amazon Research Blog ยท ยท Source
Image for Advancing AI for biology: Teaching models to design and characterize antibodies

After studying this material, you should be able to:

  1. Explain why conventional antibody discovery is slow and costly
  2. Describe the three critical factors in antibody drug development
  3. Understand how AI models address each factor: binding affinity ranking, developability prediction, and de novo design
  4. Identify key technical innovations in MochiBind, CA-MAP, and the agent-guided design pipeline
  5. Recognize how experimental validation integrates with computational design in a lab-in-the-loop workflow

Step-by-Step Teaching

Step 1: The Problem โ€” Why Antibody Discovery Needs AI

What Are Monoclonal Antibodies?

Monoclonal antibodies (mAbs) are laboratory-made proteins that mimic the immune system's ability to target specific threats, such as cancer cells or pathogens. They are among the most successful drug types, with over 100 FDA-approved drugs.

The Core Problem

Despite their success, traditional discovery is:

  • โฑ๏ธ Slow: 6โ€“12 months from target to lead candidate
  • ๐Ÿ’ฐ Expensive: Requires extensive lab work and iteration
  • ๐Ÿ” Inflexible: Hard to adapt when targets change

The Three Questions Every Antibody Drug Must Answer

FactorQuestion
Binding SiteWhere on the target should the antibody attach?
Binding StrengthWhich candidates bind most tightly?
DevelopabilityCan the antibody survive manufacturing and clinical use?

๐Ÿ”‘ Key Insight: Existing AI models perform well on familiar targets but poorly on new ones. Benchmarks have hidden this gap, making it hard to measure and fix.


Step 2: Factor 1 โ€” Ranking Binders (MochiBind)

The Challenge

You have thousands of antibody candidates. Most don't bind at all. You need to find the ones worth testing in the lab โ€” without running expensive experiments on all of them.

Why Old Methods Fall Short

Previous models tried to predict absolute binding affinity (an exact number). This fails because:

  • Numbers aren't comparable across different labs or assays
  • Models trained on known targets perform poorly on new ones
  • Test sets rarely include non-binders, making evaluations misleadingly optimistic

MochiBind's Solution: Pairwise Ranking

Instead of predicting an exact number, MochiBind asks:

"Between antibody A and antibody B targeting the same antigen โ€” which binds more tightly?"

How it works, step by step:

Step 1: Represent antibody-antigen complexes
        โ†“
        Use ESM-2 (protein language model) to embed residues
        โ†“
Step 2: Compress each complex into one embedding
        โ†“
        Average all residue embeddings โ†’ single vector
        โ†“
Step 3: Project into lower-dimensional space
        โ†“
        Specialized network layer compares two candidates
        โ†“
Step 4: Aggregate pairwise comparisons
        โ†“
        TrueSkill algorithm (from video game ranking) โ†’ global ranking

๐ŸŽฎ Analogy: TrueSkill was originally used to rank Xbox players based on match wins/losses. Here it ranks antibodies based on pairwise binding comparisons.

Why This Is Better

FeatureOld MethodsMochiBind
Input neededOften requires 3D structureSequence only
OutputAbsolute affinity numberRanked list
Cross-antigen accuracyDegrades significantlyOutperforms all baselines by ~10%
SpeedSlow200,000 pairs in ~13 seconds on CPU

Evaluation Framework

The team used the AlphaBind dataset covering four antigens (TIGIT, PD-1, HER2, SARS-CoV-1 RBD) with ~30,000 variants each. The strict rule: train on some antigens, test on a completely different one โ€” simulating real-world use on novel targets.


Step 3: Factor 2 โ€” Predicting Developability (CA-MAP)

What Is Developability?

A drug candidate must not only bind its target โ€” it must also:

  • โœ… Stay stable in the bloodstream
  • โœ… Not clump together (aggregate)
  • โœ… Not trigger an immune response
  • โœ… Survive manufacturing processes

These properties are collectively called developability.

The Core Problem: Batch Effects

Batch effect: Systematic measurement differences between labs due to different equipment, protocols, or handling โ€” not biology.

When a model trains on one lab's data, it silently learns that lab's biases. When applied to a different lab's data, predictions become unreliable.

CA-MAP's Solution: Context-Aware Prediction

CA-MAP (Context-Aware Multi-property Antibody Predictor) uses a prompt-based approach:

Input Prompt Structure:
โ”Œโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”
โ”‚ Example antibody 1 + measured property  โ”‚ โ† From same lab as query
โ”‚ Example antibody 2 + measured property  โ”‚ โ† Captures batch offset
โ”‚ Example antibody 3 + measured property  โ”‚
โ”‚ ...                                     โ”‚
โ”‚ Query antibody + property name?         โ”‚ โ† What we want to predict
โ””โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”˜

The examples from the same lab carry the batch offset information. The model learns to read and correct for it automatically.

The Training Trick: AB-Context-Aware

The problem: a model might just ignore the examples and rely only on the query sequence.

Solution: During training, apply a random hidden transformation to both the context measurements and the expected answer. The model cannot recover the correct answer without reading the context โ€” so it's forced to use it.

Performance Comparison

ConditionStandard Fine-tuningCA-MAP
No batch effectsฯ = 0.99ฯ = 0.99
Simulated batch effects (0โ€“0.3)ฯ = 0.58 โŒฯ = 0.99 โœ…

ฯ = Spearman correlation; 1.0 = perfect prediction

Architecture Efficiency

ModelTrainable ParametersInference Speed
TxGemma (baseline)40 millionBaseline
CA-MAP182,000~200ร— faster

Bonus Capability: Predicting Unseen Properties

CA-MAP can estimate properties it was never directly trained on by leveraging correlations between developability properties:

  • Trained on 4 properties โ†’ tested on 2 new ones
  • Immunogenicity: ฯ improved from 0.25 โ†’ 0.73 when correlated properties were included in context
  • This means expensive assays could be estimated from cheaper ones

Step 4: Factor 3 โ€” Designing Antibodies From Scratch

The Campaign: A Real Cancer Target

The team targeted a protein on desmoplastic small round-cell tumors โ€” a rare, aggressive pediatric cancer. This target was chosen because:

  • It sits on tumor cell surfaces (accessible)
  • Driven by a specific genetic error
  • Largely absent from healthy tissue (selective)
  • No existing structure, no prior antibody data โ€” a true test of de novo design

Phase 1: Where to Attack? (Hotspot Recommendation Agent)

Before designing antibodies, you must decide which region of the target to aim for (the epitope/hotspot).

The agent orchestrates 7 bioinformatics tools analyzing:

Target Protein
     โ†“
โ”Œโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”
โ”‚ Tool 1: Solvent-accessible surface  โ”‚
โ”‚ Tool 2: Secondary structure         โ”‚
โ”‚ Tool 3: Hydrophobicity              โ”‚
โ”‚ Tool 4: Sequence uniqueness         โ”‚
โ”‚ Tool 5: Immune Epitope Database     โ”‚ โ† 500,000 entries
โ”‚ Tool 6: Protein family annotation   โ”‚
โ”‚ Tool 7: Negative target avoidance   โ”‚
โ””โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”˜
     โ†“
Agent synthesizes โ†’ Ranked hotspot recommendations
                    with biophysical rationale

Performance: Recovered at least one true epitope residue in top 5 recommendations ~80% of the time on diverse holdout targets.

Phase 2: Generate Antibody Designs

Three generative models with different approaches each produced 96,000 designs:

ModelDesign Principle
RFantibodyDiffusion over protein backbones
IgGMJoint sequence-structure diffusion
mBERBackpropagation through structure prediction

Phase 3: Score and Filter Candidates

Each design was scored on:

  • ๐Ÿ”ฌ Folding confidence: Will it fold into the right shape?
  • ๐Ÿ”— Complex quality: Will it form the correct binding interface?
  • โš ๏ธ Sequence liabilities: Any manufacturing/development red flags?
  • ๐Ÿ’ช Binding affinity: MochiBind's ranking score

A candidate selection agent applied multi-objective Pareto filtering โ€” keeping designs that excel on different combinations of metrics, not just one.

๐Ÿ“Š Pareto filtering: Retains candidates where no other candidate is better on all metrics simultaneously โ€” preserving diversity of strong solutions.

100,000 candidates were prioritized for experimental screening.

Phase 4: Experimental Validation

100,000 computational designs
         โ†“
    Yeast display screening
    (synthesized, displayed on yeast surface)
         โ†“
    Two rounds of sorting/filtering
         โ†“
    116 survivors โ€” none bound unrelated control protein
    (confirming target specificity, not general stickiness)
         โ†“
    Individual binding measurement
         โ†“
    46 confirmed strong binders โœ…

Phase 5: Lab-in-the-Loop

The 46 binders + all binder/non-binder labels become training data for the next design cycle:

Design โ†’ Score โ†’ Experiment โ†’ Learn โ†’ Better Design โ†’ ...

Each round of experiments sharpens the models for the next round.


Step 5: Putting It All Together

The Complete AI-Driven Antibody Discovery Pipeline

TARGET PROTEIN
      โ†“
[Hotspot Agent] โ†’ Identifies where to bind
      โ†“
[Generative Models] โ†’ Creates 288,000 candidate designs
      โ†“
[MochiBind] โ†’ Ranks by predicted binding strength
      โ†“
[CA-MAP] โ†’ Predicts developability properties
      โ†“
[Pareto Filtering] โ†’ Selects diverse top candidates
      โ†“
[Lab Validation] โ†’ Experimental confirmation
      โ†“
[Lab-in-the-Loop] โ†’ Data feeds back to improve models

Summary of Key Innovations

InnovationProblem SolvedKey Idea
MochiBindRanking binders on novel targetsPairwise comparison + TrueSkill aggregation
CA-MAPBatch effects in developability predictionContext-aware prompting with forced context use
Hotspot AgentChoosing where to target on novel antigensOrchestrating 7 deterministic bioinformatics tools
End-to-end pipelineIntegrating design and validationLab-in-the-loop iterative refinement

Key Takeaways

  1. Antibody drug discovery has three core challenges: finding the right binding site, ranking binders, and predicting developability
  2. Generalization to novel targets is the hardest problem โ€” models must be evaluated cross-antigen, not just cross-variant
  3. Relative prediction (ranking) is more robust than absolute prediction across different experimental contexts
  4. Batch effects can destroy model performance; context-aware architectures can correct for them at inference time
  5. Computational design + experimental validation in a feedback loop is more powerful than either alone
  6. Speed matters: 200,000 pairs in 13 seconds enables screening at scales impossible with wet-lab methods alone

More to study