用强化学习改进大模型快速排序,提升延迟下排名精度。
Reinforcement Speculative Decoding for Fast Ranking
- 提出自上而下的解码策略,通过智能体迭代优化排序序列。
- 在有限预算内实现更高精度的多轮预测,尾部排名误差显著降低。
- 适合对延迟敏感的检索与推荐系统,尤其擅长处理未采纳项信息。
大语言模型(LLMs)已广泛应用于信息检索(IR)和推荐系统(RS)中。为缓解自回归解码的延迟问题,现有研究尝试仅解码首个词进行近似排名,但导致尾部位置性能严重下降。尽管推测解码(SD)方法可通过不同位置验证缓解此问题,其左到右的解码范式在排名系统中面临挑战:一是验证轮次无法适应严格的延迟约束;二是前轮未被采纳项的列表级排名知识常被丢弃,影响后续多词预测,尤其当候选词为未采纳项时。本文提出一种面向快速排序的强化推测解码方法。为满足排名系统的延迟要求,我们设计了自上而下的解码范式,通过智能体在受限预算下迭代修改排序序列。具体地,我们构建了针对排名任务的策略优化机制,利用强化学习(RL)主动探索最优多轮修改策略,并由LLM验证。同时,我们充分利用各轮中所有已被验证项的列表级排名信息,在RL中增强智能体的修改策略。更重要的是,我们证明了该范式与实现的理论鲁棒性与优势。在IR和RS任务上的实验表明,所提方法有效提升排序精度与推理效率。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have been widely adopted in ranking systems such as information retrieval (IR) systems and recommender systems (RSs). To alleviate the latency of auto-regressive decoding, some studies explore the single (first) token decoding for ranking approximation, but they suffer from severe degradation in tail positions. Although speculative decoding (SD) methods can be a remedy with verification at different positions, they face challenges in ranking systems due to their left-to-right decoding paradigm. Firstly, ranking systems require strict latency constraints, but verification rounds in SD methods remain agnostic; Secondly, SD methods usually discard listwise ranking knowledge about unaccepted items in previous rounds, hindering future multi-token prediction, especially when candidate tokens are the unaccepted items. In this paper, we propose a Reinforcement Speculative Decoding method for fast ranking inference of LLMs. To meet the ranking systems' latency requirement, we propose an up-to-down decoding paradigm that employs an agent to iteratively modify the ranking sequence under a constrained budget. Specifically, we design a ranking-tailored policy optimization, actively exploring optimal multi-round ranking modification policy verified by LLMs via reinforcement learning (RL). To better approximate the target LLM under the constrained budget, we trigger the agent fully utilizing the listwise ranking knowledge about all items verified by LLMs across different rounds in RL, enhancing the modification policy of the agent. More importantly, we demonstrate the theoretical robustness and advantages of our paradigm and implementation. Experiments on both IR and RS tasks show the effectiveness of our proposed method.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。