arXiv AI

SharedSAE: a unified sparse dictionary across u-net blocks and denoising time for diffusion models

arXiv Machine Learning
Aug 4

A Comparative analysis of Layer-wise Representational Capacity in AR and Diffusion LLMs

arXiv:2603. 07475v4 Announce Type: replace-cross Abstract: Autoregressive (AR) language models build representations incrementally via left-to-right prediction, while diffusion language models (dLLMs) are trained through full-sequence denoising.

By Raghavv Goel, Risheek Garrepalli, Sudhanshu Agrawal, Chris Lott, Mingu Lee, Fatih Porikli
arXiv Machine Learning
Aug 31

Learned Relay Representations for Forward-Thinking Discrete Diffusion Models

The paper introduces Learned Relay Representations (Relay), a technique for Masked Diffusion Models (MDMs) that preserves and forwards internal latent information across denoising steps via a differentiable per-token channel trained with truncated backpropagation through time. Relay enables MDMs to be forward‑thinking, avoiding costly recomputation of internal representations and improving efficiency. Applied to Fast‑dLLM v2, Relay outperforms standard supervised fine‑tuning on coding tasks and reduces inference latency by up to 32%, demonstrating a clear performance‑latency advantage for diffusion language models.

By Benjamin Rozonoyer, Jacopo Minniti, Dhruvesh Patel, Neil Band, Avishek Joey Bose, Tim G. J. Rudner, Andrew McCallum
arXiv Computer Vision
4d ago

Policy-Mask: Budget-Aware Adaptation of Diffusion Features to Discriminative Tasks

Policy‑Mask is a budget‑aware policy that jointly selects a timestep, a diffusion block, and a subset of channels from a frozen diffusion model to optimize downstream discriminative tasks. By normalizing channel scores and training on the downstream objective, it outperforms fixed‑extraction baselines on segmentation, classification, and semantic correspondence, achieving state‑of‑the‑art results on ADE20K, CIFAR‑100, and SPair‑71k. The method also extends to diffusion transformers by mapping U‑Net channels to image‑token hidden dimensions.

By Ivan Listopadov, Maksim Nakhodnov, Aibek Alanov
arXiv Computation and Language
Sep 1

Prefilling-dLLM: Predictive Prefilling for Long-Context Inference in Diffusion Language Models

arXiv:2606.10537v2 Announce Type: replace Abstract: Diffusion large language models (dLLMs) re-encode the entire prefix at every denoising step, causing recomputation that scales quadratically with...

By Jing Xiong, Qi Han, Shansan Gong, Yunta Hsieh, Boyuan Zheng, Chengyue Wu, Chaofan Tao, Chenyang Zhao, Ngai Wong