中文

DiNADO:用于可控语言模型的范数解耦神经分解预言机

计算与语言 2024-06-10 v2

摘要

神经分解预言机(NADO)是一种用于大语言模型可控生成的强大方法。其旨在避免灾难性遗忘,同时以合理的建模能力实现对熵最大化的闭式最优解的保证收敛。尽管取得成功,将 NADO 应用于广泛场景时仍出现若干挑战。原始 NADO 对低概率控制信号存在梯度消失问题,且高度依赖正则化以满足贝尔曼方程的随机版本。此外,NADO 的原始实现引入了若干额外的 transformer 层,容量有限,尤其相较于 LoRA 等其他基于微调的模型适配方法。本文提出 NADO 算法的改进版本,即 DiNADO(范数解耦神经分解预言机,norm-Disentangled NeurAlly-Decomposed Oracles),通过对所有潜在下一词元的近似预言机 RR 值之逐步全局范数进行解耦,提升 NADO 算法性能,并使 DiNADO 可与 LoRA 等微调方法结合。我们从经验与理论结果深入讨论 DiNADO 如何实现更优的容量、稳定性与灵活性。在机器翻译的形式控制与词汇约束生成任务 CommonGen 上的实验表明了上述改进的重要意义。

关键词

引用

@article{arxiv.2306.11825,
  title  = {DiNADO: Norm-Disentangled Neurally-Decomposed Oracles for Controlling Language Models},
  author = {Sidi Lu and Wenbo Zhao and Chenyang Tao and Arpit Gupta and Shanchan Wu and Tagyoung Chung and Nanyun Peng},
  journal= {arXiv preprint arXiv:2306.11825},
  year   = {2024}
}

备注

Accepted to ICML 2024 (Poster). Work was done during an Amazon Internship Program