Inference-Time Scaling of Diffusion Models via Progressive Seed Pruning Permalink
We propose Progressive Seed Pruning (PSP), an inference-time scaling method for diffusion and flow-matching models that scores intermediate denoised estimates and progressively narrows a large pool of candidate seeds, spending compute on promising trajectories instead of resampling a fixed-size pool. Across diffusion and flow-matching backbones, PSP consistently improves reward-guided selection and achieves higher GenEval scores and better human evaluation on prompt-alignment than best-of-N, importance-sampling, and tree-search baselines at matched compute.
