arXiv:2506.07276cs.LGcs.AI2025-06ICML被引 5

提出新型令牌带子模型,解释大模型解码为何有效并提升对齐性能。

Tokenized Bandit for LLM Decoding and Alignment

  • 基于大模型解码设计新带子模型,引入共现距离递减假设
  • 算法实现近似最优后悔率,证明贪心解码在特定条件下有效
  • 适用于解码阶段的模型对齐,可验证于真实与合成数据

我们提出令牌化线性带子(TLB)和多臂带子(TMAB),两类受大模型解码与对齐启发的带子问题。每轮中,用户提交查询,决策者从词元集合中不可逆地逐个选择词元。序列完成后,观察到一个随机效用,其期望由依赖于查询的序列函数决定。我们首先证明,若序列函数无结构,则学习不可能。为此引入自然假设——共现距离递减(DDMC),并提出分别具有$ ilde{O}(Lar{ ho} ext{sqrt}{T})$和$ ilde{O}(Lar{ ho} ext{sqrt}{T^{2/3}})$后悔率的算法。作为副产品,我们证明了在DDMC下贪心解码近乎最优,解释了其在多个任务中的不凡表现。该结果可直接应用于解码时的模型对齐,当非对齐效用可表示为冻结大模型效用与线性可实现潜在函数之和时。最后,我们在合成与真实数据集上验证了算法性能并检验了假设的有效性。

原文摘要 · Abstract (English)

We introduce the tokenized linear bandit (TLB) and multi-armed bandit (TMAB), variants of linear and stochastic multi-armed bandit problems inspired by LLM decoding and alignment. In these problems, at each round $t \in [T]$, a user submits a query (context), and the decision maker (DM) sequentially selects a token irrevocably from a token set. Once the sequence is complete, the DM observes a random utility from the user, whose expectation is presented by a sequence function mapping the chosen token sequence to a nonnegative real value that depends on the query. In both problems, we first show that learning is impossible without any structure on the sequence function. We introduce a natural assumption, diminishing distance with more commons (DDMC), and propose algorithms with regret $\tilde{O}(L\sqrt{T})$ and $\tilde{O}(L\sqrt{T^{2/3}})$ for TLB and TMAB, respectively. As a side product, we obtain an (almost) optimality of the greedy decoding for LLM decoding algorithm under DDMC, which justifies the unresaonable effectiveness of greedy decoding in several tasks. This also has an immediate application to decoding-time LLM alignment, when the misaligned utility can be represented as the frozen LLM's utility and a linearly realizable latent function. We finally validate our algorithm's performance empirically as well as verify our assumptions using synthetic and real-world datasets.

大模型解码带子算法模型对齐

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。