📄
Abstract - Beyond Fully Random Masking: Attention-Guided Denoising and Optimization for Diffusion Language Models
Diffusion large language models (dLLMs) offer an efficient alternative to autoregressive models through parallel decoding, yet existing post-training methods largely rely on random masking strategies that overlook intrinsic token dependencies. In this work, we present an empirical analysis of attention in dLLMs and show that tokens attending more strongly to unmasked context exhibit greater generation stability and play a critical role in reasoning. Motivated by these findings, we propose AGDO, an attention-guided denoising and optimization framework that aligns both training and optimization with attention-derived dependencies. AGDO determines the denoising order based on attention structure and emphasizes attention-critical tokens during supervised fine-tuning and reinforcement learning. Experiments on mathematical and coding benchmarks demonstrate that AGDO consistently improves reasoning performance, outperforming state-of-the-art post-training methods for dLLMs.
超越完全随机掩码:面向扩散语言模型的注意力引导去噪与优化方法 /
Beyond Fully Random Masking: Attention-Guided Denoising and Optimization for Diffusion Language Models
1️⃣ 一句话总结
本文提出了一种名为AGDO的新框架,通过分析扩散语言模型中注意力机制的作用,不再盲目随机决定哪些词先被生成,而是利用模型内部的注意力信号来智能规划去噪顺序,并重点优化关键词语,从而显著提升了模型在数学推理和编程等任务上的表现。