chaoheng
发布于 2026-10-08 / 7 阅读
0
0

Learning Unmasking Policies for Diffusion Language Models

论文总结:Learning Unmasking Policies for Diffusion Language Models

核心思想: 不再人工规定扩散模型的生成顺序,而是通过强化学习自动决定每一步解除哪些 Mask。

具体方法:

  1. 建模: 将解掩码过程定义为 MDP,冻结原始扩散模型。

  2. 策略: 使用轻量级 Transformer,输入各位置置信度、Mask 状态和时间步,输出每个位置的解掩码概率。

  3. 训练: 使用 GRPO,以答案正确性和生成速度构建奖励,学习动态生成策略。

  4. 结果: Semi-AR 下接近 Fast-dLLM,Full-Diffusion 低计算预算下优势明显;专家引导可改善探索不足,但跨领域泛化有限。

对我们研究的启发

  1. 顺序学习: 冻结 DocID 生成模型,单独训练 Semantic ID 生成顺序策略。

  2. 奖励设计: 以 MRR@100 等文档级检索指标为目标,而非仅依赖置信度。

  3. 探索优化: 利用启发式顺序提供专家轨迹,缓解 RL 探索困难。

  4. 机制分析: 研究位置重要性、预测熵与最优生成顺序的关系。

  5. 泛化验证: 比较不同 ID 编码方式和检索数据集上的策略表现。


评论