mlsanityData to 7 Oct 2026

Steering Without Breaking: Mechanistically Informed Interventions for Discrete Diffusion Language Models

arXiv 2605.10971 ↗First posted 8 May 2026cs.LG cs.AI cs.CLcited by 2

On the map: Diffusion models → Discrete and text diffusion

Abstract (arXiv)

Discrete diffusion language models (DLMs) generate text by iteratively denoising all positions in parallel, offering an alternative to autoregressive models. Controlled generation methods for DLMs, imported from autoregressive models, apply uniform intervention at every denoising step. We show this uniform schedule is inefficient and degrades quality, and the damage compounds when multiple attributes are steered jointly. To diagnose the failure, we train sparse autoencoders on four DLMs (124M-8B parameters) and find that different attributes commit on distinct schedules, varying in timing, sharpness, and magnitude. For instance, topic commits within the first 2% of denoising on MDLM, whereas sentiment emerges gradually over 20% of the process. Motivated by these profiles, we propose an adaptive scheduling mechanism that concentrates intervention where each attribute is actively forming. An idealized allocation analysis predicts that attributes with more sharply concentrated emergence benefit more from adaptive scheduling, a prediction we confirm empirically. Across seven single- and multi-attribute steering tasks on four DLMs, adaptive steering consistently improves the control-quality balance over uniform and interval-restricted baselines.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 32 of its 47 references that are in the map. First posted on arXiv 8 May 2026.

Builds on

32 of 47 references.

Closest ancestors · 6

Same direction, newest first

  1. 2026DLM-Scope: Mechanistic Interpretability of Diffusion Language Models via Sparse Autoencoders5 Feb
  2. 2026ILRR: Inference-Time Steering Method for Masked Diffusion Language Models29 Jan
  3. 2025Dream 7B: Diffusion Large Language Models21 Aug
  4. 2025Large Language Diffusion Models14 Feb
  5. 2024Simple and Effective Masked Diffusion Language Models11 Jun
  6. 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution25 Oct

Other references · 26

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  2. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  3. 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
  4. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
  5. 2015Character-level Convolutional Networks for Text Classificationcited by 2,293
  6. 2024Qwen2 Technical Reportcited by 2,076
  7. 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
  8. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
  9. 2023Representation Engineering: A Top-Down Approach to AI Transparencycited by 1,239
  10. 2023Steering Llama 2 via Contrastive Activation Additioncited by 1,009
  11. 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 831
  12. 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 579
14 more
  1. 2013k-Sparse Autoencoderscited by 209
  2. 2018Dear Sir or Madam, May I introduce the GYAFC Dataset: Corpus, Benchmarks and Metrics for Formality Style Transfercited by 209
  3. 2025SAEs Are Good for Steering -- If You Select the Right Featurescited by 74
  4. 2025Universal Sparse Autoencoders: Interpretable Cross-Model Concept Alignmentcited by 62
  5. 2023Don't lose the message while paraphrasing: A study on content preserving style transfercited by 25
  6. 2025Beyond Linear Steering: Unified Multi-Attribute Control for Language Modelscited by 23
  7. 2026Interpretability without actionability: mechanistic methods cannot correct language model errors despite near-perfect internal representationscited by 13
  8. 2025CorrSteer: Generation-Time LLM Steering via Correlated Sparse Autoencoder Featurescited by 10
  9. 2026From Directions to Regions: Decomposing Activations in Language Models via Local Geometrycited by 10
  10. 2025Breaking Bad Tokens: Detoxification of LLMs Using Sparse Autoencoderscited by 8
  11. 2025AbsTopK: Rethinking Sparse Autoencoders For Bidirectional Featurescited by 8
  12. 2026TDGNet: Hallucination Detection in Diffusion Language Models via Temporal Dynamic Graphscited by 7
  13. 2025Semantic Convergence: Investigating Shared Representations Across Scaled LLMscited by 3
  14. 2026When the Coffee Feature Activates on Coffins: An Analysis of Feature Extraction and Steering for Mechanistic Interpretabilitycited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026DLM-SWAI: Steering Diffusion Language Models Before They Unmask
  2. 2025A Survey on Diffusion Language Models
  3. 2026CoDAR: Continuous Diffusion Language Models are More Powerful Than You Think
  4. 2025Simple Denoising Diffusion Language Models
  5. 2024Scaling Diffusion Language Models via Adaptation from Autoregressive Models
  6. 2025Understanding the Quality-Diversity Trade-off in Diffusion Language Models
  7. 2023Diffusion Language Models Generation Can Be Halted Early
  8. 2026Not All Denoising Steps Are Equal: Model Scheduling for Faster Masked Diffusion Language Models