MEGA Hub

Subliminal Clocks: Latent Time Modelling in Diffusion Language Models

Authors

Do you know Maximo Eduardo Rulli?You can claim authorship or link another user.Do you know Thomas Vaitses Fontanari?You can claim authorship or link another user.Do you know Simone Petruzzi?You can claim authorship or link another user.Do you know Federico Alvetreti?You can claim authorship or link another user.Do you know Giorgio Strano?You can claim authorship or link another user.Do you know Donato Crisostomi?You can claim authorship or link another user.Do you know Giorgos Nikolaou?You can claim authorship or link another user.Do you know Tommaso Mencattini?You can claim authorship or link another user.Do you know Andrea Santilli?You can claim authorship or link another user.Do you know Emanuele Rodolà?You can claim authorship or link another user.Do you know Simone Scardapane?You can claim authorship or link another user.Do you know Alessio Devoto?You can claim authorship or link another user.

Abstract

Diffusion Language Models (DLMs) have recently emerged as a promising alternative to autoregressive models. Unlike standard diffusion-based approaches, DLMs are not explicitly conditioned on a timestep, raising a natural question: do these models internally represent denoising progress, and how is such information used downstream? In this work, we show that DLMs do in fact encode a latent representation related to the diffusion timestep within their residual streams. We find that this signal can be reliably extracted using probes across layers, indicating that denoising progress is decodable from internal activations. We further demonstrate that steering the model along a low-dimensional subspace associated with the inferred timestep allows us to systematically modulate its notion of denoising progress, leading to predictable changes in model confidence and entropy. Finally, we analyse the geometry of the identified representation, showing that it exhibits structured and interpretable properties in activation space, and shedding light on how such a signal is processed by these models.

Community

00