发现语言模型词汇键空间的相似性会引发错误预测偏倚,提出导航方法提升生成多样性。
Correlation and Navigation in the Vocabulary Key Representation Space of Language Models
- 通过上下文引导让模型避开已探索词键,实现更优采样路径
- 中间排名预测严重偏向与顶级词相似的无关词,影响长尾结果生成
- 适用于需要多样性和自我一致性验证的开放生成与推理任务
语言模型解码依赖于下一个词预测(NTP)概率分布。对于基于Transformer的神经语言模型,该分布本质上是编码输入上下文(查询)与固定词汇表示(键)之间的点积经Softmax正则化后的结果。本文研究键分布对NTP分布的影响,重点探讨键间相似性是否引发虚假相关性。知识探测实验表明,NTP中前几个高排名词通常准确,但中等排名预测高度偏向与这些顶级词在分布上相似(而非语义上)的词汇。例如,若"P"为第一预测词,则"A"至"Z"均会被高估,无论其是否有助于正确解码。这损害了采样多样性,使正确长尾结果难以被有效采样。为此,本文提出一种新颖的上下文内方法(ICN),通过迭代将查询表示从已探索区域推开:将已探索结果加入上下文并提示模型生成其他内容,促使模型产生与已探索键点积较小的查询表示。知识探测实验显示,该方法能高效导航至未探索的正确键。进一步扩展至开放式及链式思维推理生成,结果表明ICN提升了生成多样性与自一致投票性能。最后讨论了固定键空间可能带来的训练问题,以及未来研究的挑战与应对方向。
原文摘要 · Abstract (English)
Language model (LM) decoding is based on the next-token prediction (NTP) probability distribution. For neural LMs (e.g., Transformer-based), NTP distribution is essentially a softmax-regularized dot product between an encoded input context (query) and fixed vocabulary representations (keys). In this paper, we study the effect of the key distribution on the NTP distribution, with a focus on whether the similarity between keys will trigger spurious correlations in NTP. Through knowledge-probing tasks, we show that in the NTP distribution, the few top-ranked tokens are typically accurate. However, the middle-ranked prediction is highly biased towards the tokens that are distributionally (not necessarily semantically) similar to these top ones. For instance, if "P" is predicted as the top-1 token, "A"-"Z" will all be ranked high in NTP, no matter whether they can lead to correct decoding results. This hurts the sampling diversity and makes the sampling of correct, long-tail results hopeless and noisy. We attempt to alleviate this issue via a novel in-context method that iteratively pushes the query representation away from explored regions. Specifically, we include the explored decoding results in the context and prompt the LM to generate something else, which encourages the LM to produce a query representation that has small dot products with explored keys. Experiments on knowledge-probing tasks show that our method leads to efficient navigation away from explored keys to correct new keys. We further extend our method to open-ended and chain-of-thought (for reasoning) generation. Experiment results show that ICN contributes to better generation diversity and improved self-consistency voting performance. Finally, we discuss potential training issues caused by the fixed key space together with the challenges and possible ways to address them in future research.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。