Accurate structure prediction of biomolecular interactions with AlphaFold 3

Download PDF

Accurate structure prediction of biomolecular interactions with AlphaFold 3

Metadata

Compute Setup

The local extracted paper text does not report training hardware, accelerator model, chip count, wall-clock time, or FLOPs. The device row for 2024-alphafold3_2024 is therefore not reported. Under the project rule, the setup is inferred from the 2024 Google DeepMind/Isomorphic context and the local accelerator-era map as frontier science-compute infrastructure in the TPU v5p/v4-class range. That inference should not be quoted as a paper claim.

The paper does report training and inference scale. One optimizer step uses a minibatch of 256 input data samples. During initial training this becomes 256 x 48 = 12,288 diffusion samples; during fine-tuning it is reduced to 256 x 32 = 8,192 diffusion samples. The model trains in three stages with crop sizes 384, 640, and 768 tokens. Standard inference selects the top confidence sample from 5 model seeds with 5 diffusion samples per seed, for 25 samples. Protein-antibody scores are a special case ranked across 1,000 model seeds.

Bottleneck

The bottleneck is unified biomolecular structure prediction under memory and sampling constraints. AlphaFold 3 has to represent proteins, nucleic acids, small molecules, ions, covalent modifications, and complexes in one model. This stresses pair representations, MSA handling, ligand chemistry, stereochemistry, and ranking. The pair representation scales with token pairs, so crop size and maximum-token filters are core compute controls.

Diffusion changes the cost structure. AF2 produced structures through a non-generative structure module; AF3 predicts raw atom coordinates with a diffusion module. At training time, this creates thousands of diffusion samples per optimizer step. At inference time, random noise is recurrently denoised, and multiple seeds/samples may be ranked. Accuracy can therefore be improved by spending more inference compute, especially for difficult interfaces, but this creates a real cost knob.

Method Adaptation

The architecture is adapted to reduce special-case molecular machinery and to fit broad biomolecular inputs. AF3 replaces the AF2 Evoformer with a Pairformer. The paper says MSA processing is substantially de-emphasized: a smaller, simpler MSA embedding block uses inexpensive pair-weighted averaging, the MSA representation is not retained, and information flows through pair and single representations. The Pairformer has 48 blocks and operates on pair representation shape (n, n, c) and single representation shape (n, c), with c=128 for pair and c=384 for single.

The diffusion module operates directly on raw atom coordinates and coarse token representations. The paper explicitly says it omits global rotation/translation equivariance and avoids torsion-based parameterizations and violation losses, simplifying the model for arbitrary ligands and chemical graphs. That simplification trades handcrafted molecular constraints for learned denoising, confidence prediction, ranking penalties, and multiple samples.

The method also adapts to generative failure modes. Because diffusion can hallucinate plausible-looking structure in disordered regions, the authors use cross-distillation from AlphaFold-Multimer v2.3 predictions, where disordered regions tend to be extended loops. Confidence heads predict pLDDT, PAE, and a predicted distance error matrix.

Evidence

The paper reports broad benchmark evidence. Protein-ligand performance is evaluated on PoseBusters, 428 protein-ligand structures released to the PDB in 2021 or later, using pocket-aligned ligand RMSD under 2 Angstroms as the success metric. To avoid training leakage, the PoseBusters analysis uses a separate AF3 model with a 30 September 2019 training cutoff. The paper reports AF3 greatly outperforming AutoDock Vina even without structural inputs, with Fisher's exact test P = 2.27e-13, and outperforming true blind docking such as RoseTTAFold All-Atom with P = 4.45e-25.

For nucleic acids, AF3 predicts protein-nucleic complexes and RNA structures more accurately than RoseTTAFold2NA on the relevant recent PDB subsets, and evaluates ten publicly available CASP15 RNA targets. For proteins, protein-protein prediction success improves over AlphaFold-Multimer v2.3 with P = 1.8e-18; antibody-protein interaction prediction shows a marked improvement with P = 6.5e-5; protein monomer LDDT improvement is significant with P = 1.7e-34.

The compute-relevant evidence is in the seed analysis. Standard results rank 25 predictions, but antibody-antigen complexes improve as the number of ranked seeds rises up to 1,000. The paper says using one diffusion sample per model seed rather than five does not significantly change those antibody results, indicating that more model seeds, not merely more diffusion samples per seed, are necessary.

Historical Effect

AlphaFold 3 extends AlphaFold-style modeling from proteins and protein complexes toward a unified biomolecular interaction system. Historically, the compute structure changes from "predict one protein fold" to "generate, rank, and validate molecular complexes across chemistry types." The Pairformer and reduced MSA processing show one direction of adaptation: keep the pairwise representation that made AlphaFold powerful, but remove parts that do not generalize cleanly to ligands, nucleic acids, and arbitrary modifications.

The card also marks diffusion entering high-stakes scientific structure prediction. Sampling and ranking become part of the scientific result. For hard targets, more seeds can buy accuracy, placing AF3 in the search/simulation/science compute regime even though the core model is neural.

Limits

The limits are partly scientific and partly compute-shaped. The paper reports chirality violations, including a 4.4% chirality violation rate on PoseBusters despite a ranking penalty, and occasional clashing atoms that ranking penalties reduce but do not eliminate. Diffusion introduces hallucinated disorder; AF3 can produce compact plausible structure in regions that should remain disordered, even if confidence is low. Like earlier structure predictors, it predicts static structures rather than solution ensembles, and multiple random seeds do not approximate true biomolecular dynamics.

Some targets require many predictions and ranking to get the best result, which incurs extra computational cost. The paper also says code is not provided and AlphaFold 3 is available as a restricted non-commercial server, so independent hardware and runtime verification are limited from the local source.

Links

  • Compute regime: history/compute_regimes/search_simulation_science_compute/README.md
  • Source PDF and extracted text are listed in metadata above.
  • Queue status: read_complete.