Загрузка видео...

Не удалось загрузить видео

На главную

Introducing 💫 Gumbel Straight Flow 💫: Distilling Autoregressive Models into One-step Flow Maps 🚀 Can autoregressive (language) models be used as teachers for flow map parallel samplers?

23,741 просмотров • 11 дней назад •via X (Twitter)

Комментарии: 6

Фото профиля David Ruhe
David Ruhe11 дней назад

Turns out they can! Sampling from an AR model can be done with the Gumbel-max trick, picking argmax(logits + noise) at each token. We define a flow matching process directly over this noise-token coupling. The resulting paths are straight and non-crossing.

Фото профиля David Ruhe
David Ruhe11 дней назад

Surprisingly, we can then distill them directly into a bidirectional Flow Map student: the AR teacher supervises the velocity along the path, and a consistency loss trains few-step jumps. At inference, GSF maps noise to a full token sequence in parallel in just a few steps.

Фото профиля David Ruhe
David Ruhe11 дней назад

Across OpenWebText, LM1B, TinyStories, and XSum, GSF outperforms several baselines in 1–4 steps, establishing a strong compute-quality Pareto frontier.

Фото профиля David Ruhe
David Ruhe11 дней назад

📄 Paper: Huge thanks to our amazing intern @k01047339268, and to co-authors @ArnaudDoucet1 , @AndrewC_ML , @ValentinDeBort1 , and Thomas Mensink!

Фото профиля Liran Ringel
Liran Ringel10 дней назад

Impressive research taste 👏🏻 Very interesting

Фото профиля Austin DeWolfe
Austin DeWolfe11 дней назад

200 years ago they would have burned you at the stake for this black magic.

Похожие видео