Strong Stochastic Flow Maps
Diffusion Models, such as Alphafold 3, are powerful step-wise generative models. Flow Maps are a new, powerful class of models that aim to improve them by speeding up the sampling process by taking fewer steps. However, this has usually come at the cost of reduced stochasticity during sampling. Here we introduce strong stochastic flow maps, an approach that allows us to maintain the pathwise stochasticity of the diffusion SDE. For example, Strong Stochastic Flow Maps allow us to sample diverse, high quality conformers of the protein Chignolin with just 8 steps, while comparable diffusion models require up to 100 steps for similar fidelity.