不靠逐词生成,直接从初始得分选答案,提升效率与效果
Inferring from Logits: Exploring Best Practices for Decoding-Free Generative Candidate Selection
- 跳过传统逐词生成,直接从初始输出得分中选择候选答案
- 在多选题和临床决策任务中验证,支持数千个选项的高效筛选
- 覆盖多种大模型,为未来模型设计提供实证指导
生成式语言模型依赖自回归解码逐词生成输出序列。许多任务如偏好优化,需要模型直接从候选池中选出包含多个词的任务级输出作为预测。使用常规的词级别解码机制来确定任务级预测时,受限于耗时的解码过程以及离散词选择导致的梯度中断。现有方法采用无解码候选选择,通过初始词汇输出的得分估计候选概率。尽管这些估计方法广泛应用,但尚未系统评估,尤其在实际任务上的表现。本文对一组全面的无解码候选选择方法,在涵盖五项小候选池多选题任务和四项大规模候选临床决策任务(部分含10,000+选项)上进行了评估。方法与多种基础大模型结合,覆盖不同架构、规模和训练范式。分析结果与洞见将为未来模型设计提供依据。
原文摘要 · Abstract (English)
Generative Language Models rely on autoregressive decoding to produce the output sequence token by token. Many tasks such as preference optimization, require the model to produce task-level output consisting of multiple tokens directly by selecting candidates from a pool as predictions. Determining a task-level prediction from candidates using the ordinary token-level decoding mechanism is constrained by time-consuming decoding and interrupted gradients by discrete token selection. Existing works have been using decoding-free candidate selection methods to obtain candidate probability from initial output logits over vocabulary. Though these estimation methods are widely used, they are not systematically evaluated, especially on end tasks. We introduce an evaluation of a comprehensive collection of decoding-free candidate selection approaches on a comprehensive set of tasks, including five multiple-choice QA tasks with a small candidate pool and four clinical decision tasks with a massive amount of candidates, some with 10k+ options. We evaluate the estimation methods paired with a wide spectrum of foundation LMs covering different architectures, sizes and training paradigms. The results and insights from our analysis inform the future model design.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。