| ▲ | DiffusionGemma: 4x Faster Text Generation(blog.google) |
| 122 points by meetpateltech 2 hours ago | 25 comments |
| |
|
| ▲ | vineyardmike an hour ago | parent | next [-] |
| Recently I had switched to OpenCode to try out many of the Non-US-Frontier-Labs models. My unexpected favorite model to use was Mercury (a diffusion model). Not because it was “smart” but because it was stupid fast. It was more of a pair-programming experience instead of the SOTA agentic experience of prompting and waiting. Honestly, it was also way more fun and brought back some of the pre-AI coding experience while still getting some benefits of AI. It felt less of a slot machine where you prompt, wait, and hope it went in the right direction. It made me even use the tiny models like Gemini Flash Lite and GPT Mini/Nano more too. Anyways, so excited for an open-weight model and I hope it performs well. I’ll be testing this ASAP. |
| |
| ▲ | onlyrealcuzzo 8 minutes ago | parent | next [-] | | If you can run your tests fast and cheaply, and have metrics that show what bad/sloppy code is that are cheap & fast to generate, a worse fast model can outperform a far better far slower model if you value time... I've had pretty good success with LLMs after putting in place metrics to measure true complexity (not cyclomatic). | |
| ▲ | skybrian 7 minutes ago | parent | prev [-] | | Could you say more about how you use it? What does your workflow look like? |
|
|
| ▲ | samuelknight 31 minutes ago | parent | prev | next [-] |
| Some of these comments miss the advantage of diffusion. This is will have a big impact on edge devices, such as your phone or the GPU in your computer. An LLM's decoder computes tokens one-at-a-time because attention has to account for each previous token. The existing LLM decoders scale well when you have enough load to batch many inferences together. Diffusion of limited benefit there. On edge you have a different problem: your inference accelerator is starved while sloshing GB of weights back and forth from RAM. That's because the consumer RAM like LPDDRx/GDDRx is lower bandwidth than HBM, and the requests are serial so you can't batch compute common weights. Diffusion can compute tokens in parallel which relieves the memory bandwidth bottle neck. |
|
| ▲ | beklein an hour ago | parent | prev | next [-] |
| A good visual explanation of how text diffusion models like DiffusionGemma work: https://newsletter.maartengrootendorst.com/p/a-visual-guide-... |
|
| ▲ | schmorptron 29 minutes ago | parent | prev | next [-] |
| What would a diffusing reasoning model look like? have a pre-defined length [thinking] block that gets diffused over a long time, and then the final output block uses what is in that thinking block as part of its input?
And how do diffusion models decide the output length in the first place, is it a pre-set parameter? or does it diffuse an [end] token into the middle somewhere? |
| |
|
| ▲ | minimaxir 2 hours ago | parent | prev | next [-] |
| A few days ago I was just thinking that Google never talked about their diffusion text generation model after demoing it at I/O a year ago. The rumor is that it was too expensive to run, but with the provided chart using the same 1x H100 hardware and comparing DiffusionGemma to regular Gemma, that shouldn't be the case. I'm curious what the downside for this speed is here aside from being slightly weaker than Gemma. |
| |
| ▲ | ac29 2 hours ago | parent | next [-] | | > I'm curious what the downside for this speed is here "DiffusionGemma's speedup is designed for local and low-concurrency inference. In high-QPS cloud serving, autoregressive models can be deployed to saturate compute efficiently, so DiffusionGemma's parallel decoding offers diminishing returns and can result in higher serving costs" | |
| ▲ | GaggiX an hour ago | parent | prev [-] | | Well with a standard autoregressive model you can generate for example 256 tokens at once if you have 256 users, with this approach you can generate 256 tokens for a single user but you need several forward steps. So the diffusion process takes more GFLOPs, if you have enough users you can already balance memory and compute. | | |
|
|
| ▲ | roosgit 34 minutes ago | parent | prev | next [-] |
| Can LoRAs be used to increase the quality of these diffusion models? Nvidia mentions something about this https://huggingface.co/nvidia/Nemotron-Labs-Diffusion-8B#inf... |
|
| ▲ | SkitterKherpi an hour ago | parent | prev | next [-] |
| It is cool but local models while okay already feel noticeably worse than even the cheapest APIs so I can't see myself sacrificing even a little bit of their quality for speed. I'm sure it's worth it for some usecases, curious to hear specific ones that people are already planning to deploy to production. |
| |
| ▲ | Mashimo 35 minutes ago | parent [-] | | Maybe writing / bootstraping unit tests? Does not need opus level to write, and easy to iterate on. | | |
| ▲ | SkitterKherpi 31 minutes ago | parent [-] | | I can see it but even if I do that for something like tests I'd still eat the time cost of the normal Gemma for 10% extra performance. And further, if you switch between the fast and normal Gemma for different tasks you eat the big time cost of loading the other model (and maintaining both in the first place). |
|
|
|
| ▲ | kkukshtel an hour ago | parent | prev | next [-] |
| I think this is the future. The sort of left-field rumble that turns into a quake in 5 years. |
| |
| ▲ | lambda an hour ago | parent | next [-] | | This may be the future of local models. The thing is, diffusion models perform somewhat worse than autoregressive on text. So you lose some performance. Speed is the big advantage. Autoregressive when doing local inference is mostly memory bound; you're doing one token at a time, for each token you need to load all weights. MTP helps a bit by allowing you to draft tokens in a smaller model and then verify them in parallel with the larger model, allowing you to do a few computations for every memory load, but because you're still doing tokens sequentially and need to discard invalid drafted tokens, you can only get so much speedup. For hosted models, however, you can batch many token generations together, fully utilizing all of the compute while no longer being bottlenecked on memory bandwidth. So they are already operating at close to max efficiency. So, diffusion kind of loses its beneifit in hosted models. Sure, maybe you could pay more to have slightly lower latency responses by doing diffusion for one user at a time instead of autoregressive for many in parallel. But given that it also reduces accuracy, it's hard to see where you'd really want that. Unless they're able to bring it up to par with autoregressive, it seems like it's a bit of a dead out outside of local models where you're generally just doing one thing at a time. | | |
| ▲ | horsawlarway 7 minutes ago | parent [-] | | I'm particularly curious to know how this plays out, and I seriously hope that more labs focus on diffusion models for text usage. My immediate thought - this performs slightly worse than the autoregressive gemma equivalent, but it may also let me functionally run better models in diffusion variants. Ex - I can run 70b-120b autoregressive models locally right now, but I get ~5-15t/s, which just isn't fast enough for serious work. Which caps me down in the 20-36b models (ex - gemma4) where I can get 100+t/s on the same hardware. So the question becomes - does the quality drop from a diffusion model outweigh the quality bump from using a larger model? Because if not... sounds like diffusion models have a lot of space to thrive. --- Sadly - if they can't be hosted profitably, I question whether this space will actually be explored. |
| |
| ▲ | famouswaffles an hour ago | parent | prev [-] | | Almost certainly not if things remain as they are. The reason there's been little traction is the quality gap between diffusion and autoregressive models is pretty stark. I mean just look at the benchmarks here. Large dropoffs, with the hardest benchmarks seeing the largest drops. On top of that, almost all the speed benefits of diffusion models become negated at scale. So this is only attractive for local model development and almost everyone training local models still care about pound for pound quality and inference efficiency at scale. |
|
|
| ▲ | xnx an hour ago | parent | prev | next [-] |
| Is the diffusion approach any use in Multi-Token Prediction (MTP) drafters? https://blog.google/innovation-and-ai/technology/developers-... |
| |
| ▲ | fcanesin an hour ago | parent | next [-] | | Yes, DFlash is currently a SOTA speculative decoding method that Xiaomi just used in their MiMo model for >1000tkps | |
| ▲ | doctorpangloss an hour ago | parent | prev [-] | | MTP is a training optimization. Drafting requires verification, and verification is the full model inference. Speculative decoders are the name for the inference time optimization, that is more like a verifier that is a smaller model. |
|
|
| ▲ | incognito124 23 minutes ago | parent | prev | next [-] |
| I just *love* the commit message on Github: "Make TPUs go brr" |
|
| ▲ | rvz 2 hours ago | parent | prev [-] |
| We need more local open weight models that are performant and just as good (or good enough) as the best frontier ones. Then you will be able to achieve Jevons Paradox and enjoy the same “productivity gains” without paying for these extortionate token prices by closed model providers or have it as cheap as possible. And especially, no silent nerfing of the model. |
| |
| ▲ | _fw an hour ago | parent [-] | | We have this though, right? Compare SOTA local models to where the frontier was last year. There weren't many people complaining that last year's frontier models were incapable. Next year, and the year after, Fable, GPT 5.5 and Gemini 3.5 will feel quite ordinary. And perhaps even within reach of a prosumer running models locally. |
|