TRACESTEER: Trace Driven Activation Steering for Hallucination Mitigation in Diffusion Language Models
cs.CL
Submitted: 2026-10-13
Updated: 2026-10-13
Code: https://github.com/zhangxincode/TRACESTEER
License: http://creativecommons.org/licenses/by/4.0/
The gist: Diffusion large language models (dLLMs) have emerged as a promising alternative to autoregressive LLMs, yet remain susceptible to hallucination.
Terminology
Abstract
Diffusion large language models (dLLMs) have emerged as a promising alternative to autoregressive LLMs, yet remain susceptible to hallucination. Activation steering offers a promising mitigation approach by guiding parallel token predictions through internal representations. However, existing steering methods largely overlook two characteristics of dLLMs: dynamic denoising states and temporal hallucination risks, which challenge predefined steering vectors and uniform intervention strengths. We propose TRACESTEER, a trace-driven activation steering framework that uses preceding denoising predictions as feedback to jointly adapt steering direction and strength. Specifically, Denoising Feedback Steering (DF-Steering) constructs step-specific steering vectors from preceding predictions and current hidden states. Trajectory-Guided Steering Gate (TS-Gate) regulates their strength using prediction uncertainty and the denoising timestep. With the backbone frozen, factual supervision and magnitude regularization jointly guide the learning of adaptive steering interventions, promoting factual reliability and stability throughout denoising. Experiments on two dLLMs show improved factual reliability on TruthfulQA and generalization across five downstream benchmarks.
Sources
- GPT-4 Technical Report
- ContextFocus: Activation Steering for Contextual Faithfulness in Large Language Models
- ILRR: Inference-Time Steering Method for Masked Diffusion Language Models
- LLaDA2.0: Scaling Up Diffusion Language Models to 100B
- Retrieval-Augmented Generation for Large Language Models: A Survey
- Measuring Massive Multitask Language Understanding
- The Hidden Life of Tokens: Reducing Hallucination of Large Vision-Language Models via Visual Information Steering
- DeepSeek-V3 Technical Report
- DynHD: Hallucination Detection for Diffusion Large Language Models via Denoising Dynamics Deviation Learning
- OSCAR: Orchestrated Self-verification and Cross-path Refinement
- Activation Steering for Masked Diffusion Language Models
- TACG: Trajectory-Aware Commit Gating for Diffusion Language Model Decoding
- Corrective Retrieval Augmented Generation
- Qwen3 Technical Report
- Spherical Steering: Geometry-Aware Activation Rotation for Language Models
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering