Blitz: Accelerating the Diffusion Modules of AlphaFold3 and Boltz-2
Presents novel distillation technique for computational biology researchers working with diffusion models.
AI Summary
Blitz framework accelerates protein structure prediction by reducing AlphaFold3 and Boltz-2 denoising steps from 200 to 8 while maintaining accuracy.
Excerpt
Diffusion-based protein structure prediction models produce all-atom structures in about 200 denoising steps, and that cost has driven a number of few-step samplers. We present Blitz, a few-step distillation framework that uses a geometric noise schedule to distribute steps more evenly across noise scales, then learns those updates through multistep consistency training. Physical steering and target-free candidate selection help reduce the remaining geometric defects at inference. We apply Blitz
