“Does DiffusionGemma do latent reasoning?” by Jan Bauer, Neel Nanda

16/08/2026 25 min
“Does DiffusionGemma do latent reasoning?” by Jan Bauer, Neel Nanda

Listen "“Does DiffusionGemma do latent reasoning?” by Jan Bauer, Neel Nanda"

Episode Synopsis

TL;DR
Google DeepMind's recent model DiffusionGemma (DG) generates text via diffusion, meaning many diffusion steps happen before generating the final output. In particular, these diffusion steps carry vectors in addition to tokens. If we cannot interpret these tokens and vectors, the model has significant opaque serial depth, potentially harming monitorability. Recently, Engels et al. found that DG nevertheless maintains high monitorability, for instance by showing that projecting the distribution to its top-k items largely retains performance. We strengthen these results by showing that this performance degradation is largely a sampler artifact and good performance can be maintained with only the top item, supporting the case for high monitorability. Still, we also find some rare case studies where the distribution vector is load-bearing computationally, i.e. where top-1 projection would be detrimental. However even in these cases, it just encodes superposition, remaining interpretable.
Apart from model behavior, we also examined how interpretability techniques carry over to DiffusionGemma, including probes, steering, and J-lens. We find that performance is largely retained. This is a positive update on the interpretability of diffusion models that are derived from text-pretrained LLMs (an efficient training method more likely to be deployed), but might not apply [...] ---Outline:(00:10) TL;DR(01:51) Introduction(02:49) Background on DiffusionGemma(04:39) Performance degradation from top-k truncation largely is a sampler artifact(06:24) A case study for using the distribution computationally: letter arithmetic(09:13) Parallel computation(11:09) Autonomous computational usage of(13:01) Transfer of interpretability techniques(13:16) Representation similarity(14:26) Probe retention(15:45) DiffusionGemma's representation is more linearly separable(16:08) Steering retention(17:31) J-Lens retention(18:50) DiffusionGemma represents tokens non-causally(19:27) Conclusion(20:30) Appendix(20:46) Post-hoc rationalization(23:11) Load-bearing problems commit the answer only after the CoT(24:04) How bidirectional are DiffusionGemma's generations? ---
First published:
August 15th, 2026

Source:
https://www.lesswrong.com/posts/QBuJ3suRZxrrxSTtv/does-diffusiongemma-do-latent-reasoning
---
Narrated by TYPE III AUDIO.
---Images from the article:Apple Podcasts and Spotify do not show images in the episode description. Try Pocket Casts, or another podcast app.

More episodes of the podcast LessWrong (30+ Karma)