English

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

Computation and Language 2026-05-11 v3 Machine Learning

Abstract

Autoregressive (AR) language models build representations incrementally via left-to-right prediction, while diffusion language models (dLLMs) are trained through full-sequence denoising. Although recent dLLMs match AR performance, whether diffusion objectives fundamentally reshape internal representations remains unclear. We perform the first layer- and token-wise representational analysis comparing native dLLMs (LLaDA), native AR models (Qwen2.5), and AR-initialized dLLMs (Dream-7B), using cosine similarity across layers and tokens alongside static inference-time layer-skipping as an analytical probe of redundancy. We find that diffusion objectives produce more global representations with substantial early-layer redundancy and reduced recency bias, while AR objectives yield tightly coupled, locally structured representations. AR-initialized dLLMs retain AR-like dynamics despite diffusion training, revealing persistent initialization bias. Leveraging this redundancy, native dLLMs absorb up to 18.75% FLOPs reduction while retaining over 90% performance on math-reasoning and coding benchmarks, whereas AR models collapse under identical skipping, revealing that diffusion objectives, rather than architecture alone, induce depth redundancy that enables principled compression.

Keywords

Cite

@article{arxiv.2603.07475,
  title  = {A Comparative analysis of Layer-wise Representational Capacity in AR and Diffusion LLMs},
  author = {Raghavv Goel and Risheek Garrepalli and Sudhanshu Agrawal and Chris Lott and Mingu Lee and Fatih Porikli},
  journal= {arXiv preprint arXiv:2603.07475},
  year   = {2026}
}

Comments

v3: improving writing with all v2 changes

R2 v1 2026-07-01T11:08:55.225Z