研究提出Nash解码框架,性能超18倍大自回归模型

论文AI 评分 68/100arXiv cs.AI待复核
AI 聚合

本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。

一项新研究将文本生成视为博弈,把词元位置当作参与者、词汇表作为动作、对数条件概率设为效用,证明其纳什均衡在序列增长时似然度呈指数级高于自回归输出。提出的Nash解码算法能在O(1/ε)时间达ε均衡。在CLAPNQ等三个问答基准测试中,基于掩码语言模型的均衡结果,其F1与ROUGE分数超越了体积达自身18倍的自回归模型,全程无需微调或重训,代价是增加测试期算力。

全文梳理

AI 摘要依据下方信源原文自动整理,非原文转载

把改写变成博弈

这篇论文把文本修订过程重新表述为一个博弈问题。在这个设定里,序列中的每一个 token 位置被看作一名参与者,词表中的候选词是这名参与者可以选择的动作,而每名参与者的收益就是语言模型给出的对数条件概率。这样一来,一段文本的修订结果就对应到博弈论中的纳什均衡:在均衡状态下,任何单个位置单独更换词元,都无法让自己的收益变得更好。作者给出的动机是,随着序列长度增加,纳什均衡所对应的似然度可以比自回归方式生成的输出高出指数级别。这一结论为后续算法设计提供了理论依据,也解释了为什么从均衡角度出发可能找到自回归解码难以触及的高概率文本。

Nash 解码的代价

在博弈表述的基础上,作者提出了 Nash 解码算法。该算法在能够获取以提示为条件的 token 联合概率的前提下,可以在 O(1/ε) 时间内达到一个 ε-纳什均衡。实际运行时,算法并不需要真实的联合概率,而是使用大型语言模型给出的条件概率估计来驱动。论文在问答类基准上评估了由此得到的均衡结果,涉及 CLAPNQ、PubMedQA 和 CoQA 三个数据集。结果显示,由掩码语言模型得到的纳什均衡在 F1 和 ROUGE 指标上超过了自回归模型,后者规模最多达到前者的 18 倍。这一优势是在没有做任何微调或重新训练的情况下取得的,代价则是测试阶段需要额外的计算量。

适用条件与限制

从适用范围看,这套方法依赖对 token 条件概率的估计,因此需要能够提供此类概率的模型,论文使用的是掩码语言模型。评估集中在问答任务上,指标为 F1 与 ROUGE,说明当前验证的场景偏向有参考答案的问答类生成。方法本身不涉及对模型参数的更新,既不需要微调也不需要重新训练,这意味着它可以叠加在已有模型之上使用。但论文也明确指出,性能提升伴随着测试时计算的增加,也就是说推理成本会上升。此外,算法达到 ε-纳什均衡的时间复杂度为 O(1/ε),ε 越小所需时间越长,精度与开销之间存在权衡。论文共 34 页,包含 6 幅图和 11 张表,并提供了代码链接。

对使用者的含义

对关注文本生成的人来说,这项工作提供了一条不同于自回归解码的路径:不再逐词从左到右生成,而是把整段文本的各个位置视为相互影响的参与者,通过求解均衡来得到最终输出。论文报告的结果表明,较小的掩码语言模型在问答基准上可以凭借这种方式超过规模大得多的自回归模型,且无需额外训练。这提示模型规模并非决定生成质量的唯一因素,解码策略本身也会带来明显差异。不过,由于需要额外的测试时计算,并且当前证据来自问答类任务,实际采用时需要在效果与推理开销之间做出取舍,也需要在自身任务上验证是否同样成立。

为什么值得看

无需重训即可用小模型击败18倍大模型,为测试期算力换性能提供新解法。

微调算力基准测试

信源1 家

  1. [1]arXiv cs.AI一手信源Nash Equilibrium Text: A Game-Theoretic Decoding Framework for Text Generation

关键事实

  • 该研究将文本修正建模为博弈,词元位置为参与者,词汇为动作,对数条件概率为效用,存在纳什均衡。[1]

  • 随序列变长,纳什均衡的似然度可比自回归输出呈指数级提高。[1]

  • Nash解码算法可在O(1/ε)时间达到ε-纳什均衡。[1]

  • 在CLAPNQ、PubMedQA和CoQA上,掩码语言模型的均衡结果F1和ROUGE超18倍大自回归模型。[1]

相关 · 论文

本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。