跳到正文
arXiv cs.AI· Chengwei Zhou, Zhaoyan Jia, Massoud Pedram, Gourav Datta·· 14 小时前AI 评分31

Cord:面向掩码扩散语言模型的置信度锚定顺序残差解码

Cord: Confidence-Anchored Order Residual Decoding for Masked Diffusion Language Models

AI 导读

针对掩码扩散语言模型(dLLM)解码顺序问题,研究者提出 Cord——一种以置信度解码为稳定参照、通过几何残差学习何时偏离的轻量提示词条件策略。在 3 个 dLLM 和 5 个推理、代码与约束满足基准上,Cord 将平均任务成功率从置信度解码的 32.0% 提升至 48.2%,并优于各领域最佳固定残差。该方法仅增加 0.53M 可训练参数,无需额外主干前向,推理延迟开销为 5.39%。

正文 · 原文

View PDF HTML (experimental)

Abstract:The decoding order of masked diffusion language models (dLLMs) can substantially affect generation quality, yet common decoders typically apply the same reveal rule throughout generation. Confidence provides a strong default, but it measures how decisively a position can be predicted now, not how useful revealing that position will be for subsequent decoding. We propose Cord, a lightweight prompt-conditioned policy that retains confidence decoding as a stable reference and learns when and how to depart from it through geometric residuals over generation direction, proximity to revealed context, and two-sided infilling. The residuals adapt to both the prompt and the evolving partial response. Rather than sampling a new policy action at every decoding step, Cord samples a single low-dimensional coefficient matrix per rollout, which induces an entire state-dependent reveal trajectory. This yields a contextual-bandit formulation trained directly from sequence-level rewards, without step-level credit assignment or backbone updates. Across three dLLMs and five reasoning, code, and constraint-satisfaction benchmarks, Cord improves average task success from 32.0% under confidence decoding to 48.2%, and further improves over the best per-domain fixed residual. Behavioral audits show that the learned residuals induce substantial, task-dependent changes in reveal locality and two-sided infilling. Cord adds only $0.53$M trainable parameters, requires no additional backbone forward passes, and incurs only 5.39% inference latency overhead.
Subjects: Computation and Language (cs.CL); Artificial Intelligence (cs.AI)
Cite as: arXiv:2610.39372 [cs.CL]
  (or arXiv:2610.39372v1 [cs.CL] for this version)
  https://doi.org/10.48550/arXiv.2610.39372

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Chengwei Zhou [view email]
[v1] Thu, 29 Oct 2026 05:03:29 UTC (777 KB)

来源:arXiv cs.AI · arxiv.org