用轻量级重排序提升大模型生成质量,效率远超传统方法
Language Ranker: A Lightweight Ranking framework for LLM Decoding
- 将解码视为推荐系统的排序阶段,引入轻量模块重排候选文本
- 仅需不到0.5百万额外参数,性能接近大型奖励模型
- 适合追求高效推理的部署场景,尤其适合资源受限环境
传统大语言模型研究多聚焦于输出分布优化,对解码过程关注不足。近期基于奖励模型的推理扩展虽重要,但计算成本高且适用性有限。本文从推荐系统视角重新审视大模型生成,将解码类比为推荐管道中的排序阶段。观察发现,传统解码与奖励模型均存在冗余问题。为此,提出Language Ranker框架,通过基础模型提取特征,引入轻量级模块对候选响应进行重排序。跨多种任务的实验表明,该方法性能可媲美大规模奖励模型,仅需<0.5M额外参数,在训练与推理阶段均显著降低计算开销,充分释放大模型潜力。
原文摘要 · Abstract (English)
Conventional research on large language models (LLMs) has primarily focused on refining output distributions, while paying less attention to the decoding process that transforms these distributions into final responses. Recent advances, such as scaling the computation of inference time with reward models, have underscored the importance of decoding, but these methods often suffer from high computational costs and limited applicability. In this paper, we revisit LLM generation through the lens of recommender systems, conceptualizing the decoding process as analogous to the ranking stage in recommendation pipelines. From this perspective, we observe that both traditional decoding methods and reward models exhibit clear limitations such as redundancy. Motivated by this insight, we propose Language Ranker, a novel framework that introduces a lightweight module to rerank candidate responses using features extracted by the base model. Experiments across a wide range of tasks show that Language Ranker achieves performance comparable to large-scale reward models, while requiring only <0.5M additional parameters, significantly reducing the computational overhead during both training and inference stages. This highlights the efficiency and effectiveness of our method, showcasing its potential to fully unlock the capabilities of LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。