LoGG next paper: Accelerated Diffusion Models via Speculative Sampling
1 view
Skip to first unread message
Hannes Stärk
unread,
Feb 1, 2025, 2:35:47 PMFeb 1
Reply to author
Sign in to reply to author
Forward
Sign in to forward
Delete
You do not have permission to delete messages in this group
Copy link
Report message
Show original message
Either email addresses are anonymous for this group or you need the view member email addresses permission to view the original message
to lo...@googlegroups.com
Hi together,
Next paper:
Accelerated Diffusion Models via Speculative Sampling https://arxiv.org/abs/2501.05370 (Valentin De Bortoli, Alexandre Galashov, Arthur Gretton, Arnaud Doucet) Speculative sampling is a popular technique for accelerating inference in Large Language Models by generating candidate tokens using a fast draft model and accepting or rejecting them based on the target model's distribution. While speculative sampling was previously limited to discrete sequences, we extend it to diffusion models, which generate samples via continuous, vector-valued Markov chains. In this context, the target model is a high-quality but computationally expensive diffusion model. We propose various drafting strategies, including a simple and effective approach that does not require training a draft model and is applicable out of the box to any diffusion model. Our experiments demonstrate significant generation speedup on various diffusion models, halving the number of function evaluations, while generating exact samples from the target model.
Speaker: The myth, the legend, Valentin De Bortoli from Google Deepmind.