arXiv:2606.06712cs.CLcs.AI2026-06被引 1

用自蒸馏方法将自回归模型高效转为扩散语言模型,大幅降低训练成本。

Data-Efficient Autoregressive-to-Diffusion Language Models via On-Policy Distillation

论文配图:Data-Efficient Autoregressive-to-Diffusion Language Models via On-Policy Distillation
图 1 · 摘自论文原文
  • 让自回归模型自己生成推理轨迹,教师模型在这些轨迹上进行知识蒸馏。
  • 仅需15到7000倍少的训练数据,性能仍优于传统方法。
  • 适合想低成本改造现有大模型的开发者和研究者。

我们研究了将自回归语言模型(ARLM)转换为扩散语言模型(DLM)的方法。以往工作通过用双向注意力替换ARLM的因果注意力,并以DLM目标重新训练,但存在双重分布偏移:一是从下一个词预测目标转向DLM目标会丢失原模型积累的知识;二是标准DLM存在训练-推理不一致问题,因训练损失基于随机掩码序列,而非推理时信心解码产生的实际轨迹。为此,我们提出一种基于策略的扩散语言模型(OPDLM),采用自蒸馏方式实现转换。具体地,学生模型(带双向注意力的ARLM)生成自身推理轨迹,教师模型(原始冻结的ARLM)在这些轨迹上提供目标logits。由于直接在策略内训练,消除了训练-推理偏差,同时来自原始模型的蒸馏增强了知识保留。实验表明,OPDLM仅需15至7000倍更少的训练标记,在多种任务上表现优异。该方法避免了昂贵的DLM预训练,使模型转换成为一种高效的ARLM后训练范式。

原文摘要 · Abstract (English)

We study the transformation of autoregressive models (ARLMs) into diffusion language models (DLMs). Rather than pretraining from scratch, prior work replaces the causal attention in ARLMs with bidirectional attention and then trains the resulting model using a DLM objective. However, these approaches incur two distribution shifts. First, transitioning from a next-token prediction objective to a DLM objective can discard knowledge acquired by the ARLM during training. Second, standard DLMs suffer from a train-inference mismatch, as the training loss is defined on randomly masked sequences rather than the trajectories encountered at inference produced by confidence-based decoding. To address both challenges, we introduce an On-Policy Diffusion Language Model (OPDLM) in which On-Policy Distillation (OPD) is employed for ARLM-to-DLM transformation. Specifically, OPDLM is trained via self-OPD, where the student, an ARLM with bidirectional attention, generates its own trajectories, and the teacher, the original frozen ARLM, distills its knowledge by providing target logits on these trajectories. By training directly in an on-policy manner, OPDLM eliminates the train-inference mismatch in DLMs, while distillation from the original model enhances knowledge retention from the ARLM. Empirical results demonstrate that OPDLM requires 15x to 7,000x fewer training tokens with strong performance across a wide variety of tasks. OPDLM avoids the prohibitive cost of DLM pretraining and positions DLM transformation as a form of ARLM post-training.

扩散模型自回归知识蒸馏高效训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。