NeurIPS 2026 · Poster

CORAL: Learning Amyloid Fibril Ligand Docking
with Cooperative Binding Rewards

Yasheng Sun1*†, Bohan Li2*, Youqi Tao3†, Jürgen Schmidhuber1
1Center of Excellence for Generative AI, KAUST
2Department of Biomedical Engineering, National University of Singapore
3Stern Laboratory, Brigham and Women's Hospital, Harvard University

*Equal contribution   ·   †Corresponding authors

Paper walkthrough

5:27 · English · Captioned

CORAL's two-stage training: geometric learning organizes ligands into periodic stacks along an amyloid groove; energetic optimization refines the interactions between the fibril and neighboring ligand copies.
Learning geometry, then cooperative interactions. Geometric warm-up teaches the model to place ligand copies along the cross-β groove. Online reinforcement learning then refines the pose distribution using both protein–ligand and inter-ligand interaction scores.

Abstract

A hallmark of neurodegenerative diseases such as Alzheimer's and Parkinson's is the aberrant aggregation of proteins into amyloid fibrils, and small molecules that selectively bind to these fibrils hold promise as diagnostics, imaging probes, and therapeutics. Predicting how such ligands bind to fibril targets, however, presents two fundamental challenges. First, resolved co-crystal structures of amyloid–ligand complexes are exceptionally scarce; even with recent advances in cryo-EM only a handful have been structurally characterized, making supervised training of docking models impractical for this target class. Second, amyloid fibrils present a binding mode fundamentally different from globular proteins: ligands intercalate into longitudinal cross-β grooves and stack cooperatively along the fibril axis, a geometry that existing docking models are not designed to capture.

To address these challenges, we present CORAL (COopeRative Amyloid Ligand docking), a reinforcement learning framework that trains a generative docking model to produce ligand pose distributions tailored to the cross-β groove geometry. Our reward explicitly incorporates cooperative ligand–ligand stacking energy alongside protein–ligand docking affinity, directly capturing the distinctive binding geometry of amyloid fibrils. We further introduce a curated evaluation set of amyloid–ligand complexes constructed from model-generated poses validated by domain experts. Experiments on both experimentally resolved structures and this evaluation set demonstrate improved pose quality and binding affinity correlation over existing docking baselines.

Method

Key observation. Amyloid grooves repeat along the fibril axis. Modeling the interactions between neighboring ligand copies gives the model a signal for this cooperative binding geometry.

CORAL architecture and training pipeline: a Pairformer–Diffusion model uses fibril-periodicity priors and is trained with geometric supervision followed by cooperative-reward reinforcement learning.
Overview of CORAL. A Pairformer–Diffusion structure model jointly predicts coordinates for peptide chains and ligand copies. Fibril-periodicity priors encode the repeating architecture, while two training stages shape the generated pose distribution.

1. Geometric warm-up

Approximately 2,000 curated stacking poses provide supervision for the model's diffusion, distogram, and confidence objectives, establishing a prior for groove-aligned ligand geometry.

2. Cooperative-reward refinement

Online DiffusionNFT training uses computed protein–ligand and inter-ligand energy terms. A frozen-reference velocity regularizer helps retain the warmed-up geometry during refinement.

Results

We evaluate docking on AmyDock-CryoEM (18 experimentally resolved complexes) and AmyDock-Synthetic (57 evaluation pairs with computational reference poses curated by experts). CORAL reaches a Hit@5 Å of 50.0% and 68.4%, respectively, with median RMSDs of 4.89 Å and 3.84 Å.

Docking accuracy and affinity rank correlation. Mean and median report best-pose heavy-atom RMSD in Å; Hit@5 Å is the fraction of pairs with RMSD ≤ 5 Å. αSyn-Bind contains 47 compounds with measured affinities; ρ is Spearman correlation with experimental log(Kd/M). Values are reported in the paper.

Main results for eight docking methods on AmyDock-CryoEM, AmyDock-Synthetic, and αSyn-Bind.
AmyDock-CryoEMAmyDock-SyntheticαSyn-Bind
MethodMean ↓Median ↓Hit@5 Å ↑Mean ↓Median ↓Hit@5 Å ↑ρ ↑p
DiffDock17.6713.680.0%19.1919.683.5%0.1910.197
DynamicBind26.6329.020.0%18.9017.931.8%0.1020.544
FABind26.5024.900.0%30.3426.870.0%0.1570.292
NeuralPLexer17.4319.005.6%19.7816.560.0%−0.2910.047
FlowDock19.9521.520.0%20.6919.880.0%0.1480.322
Boltz-228.2728.670.0%37.1333.870.0%0.2230.132
Chai-119.8519.920.0%19.4016.520.0%−0.0790.597
CORAL8.704.8950.0%7.353.8468.4%0.4190.004
Qualitative comparison on two AmyDock-CryoEM targets, showing ground-truth structures and predictions from seven baselines and CORAL in axial and oblique views.
Predicted fibril–ligand poses. Each target is shown in axial and oblique views, with close-ups of the ligand-binding region. CORAL models periodic ligand stacks along the groove. View the full-resolution figure.

Analysis

Both training stages contribute. Geometric warm-up supplies a useful stacking prior; cooperative-reward refinement improves the reported pose and energy metrics. Removing the stacking reward mainly degrades the computed interaction score, while removing the stability regularizer also worsens pose accuracy.

Ablation study on AmyDock-Synthetic. Each variant changes one component of the full model. RMSD is in Å. ΔGeff is the paper's computed effective binding score (kcal/mol); lower values are better.

Ablation study of the reward components and two-stage training recipe.
RMSD ↓Computed ΔGeff ↓
VariantMeanMedianMeanMedian
CORAL (full)7.353.84−2.24−2.44
No symmetry term7.924.31−2.07−2.21
No cooperative term8.214.46−0.78−0.95
No stability regularizer12.078.13−1.32−1.45
Warm-up only (no RL)8.675.18−1.41−1.58
RL only (no warm-up)13.459.27−0.72−0.88

Molecular Dynamics Visualizations

Simulation trajectories of candidate ligand stacks along amyloid fibrils. The gallery retains the original structure and system identifiers for each clip.

More simulations (6)

BibTeX

Download .bib
@inproceedings{sun2026coral,
  title     = {{CORAL}: Learning Amyloid Fibril Ligand Docking with Cooperative Binding Rewards},
  author    = {Sun, Yasheng and Li, Bohan and Tao, Youqi and Schmidhuber, J{\"u}rgen},
  booktitle = {Advances in Neural Information Processing Systems},
  year      = {2026},
  url       = {https://neurips.cc/virtual/2026/loc/sydney/poster/154426}
}