只需访问模型输出的logits,就能高效适配闭源大模型生成特定内容。
Logits are All We Need to Adapt Closed Models
- 将文本生成视为分类问题,通过重加权logits实现精准控制。
- 仅用少量任务数据即可显著提升生成内容的相关性与准确性。
- 为闭源模型提供接口支持,适合需要定制化生成的开发者使用。
许多商业大语言模型为闭源,开发者只能通过提示工程进行内容对齐。尽管当前模型不开放token logits,我们提出若能获取logits,将可实现超越提示工程的更强适配能力。本文提出一种基于logits的逐标记概率重加权框架,结合少量任务数据,有效引导黑盒LLM生成应用相关的内容。方法将下一词预测建模为监督分类任务,证明模型对齐可转化为标签噪声修正问题,从而构建仅依赖logits的自回归重加权模型(Plugin model)。理论分析表明,仅重加权logits即足以完成任务适配。在多个数据集、模型及重加权策略上的实验验证了该方法的有效性,呼吁更广泛开放闭源模型的token logits访问权限。
原文摘要 · Abstract (English)
Many commercial Large Language Models (LLMs) are often closed-source, limiting developers to prompt tuning for aligning content generation with specific applications. While these models currently do not provide access to token logits, we argue that if such access were available, it would enable more powerful adaptation techniques beyond prompt engineering. In this paper, we propose a token-level probability reweighting framework that, given access to logits and a small amount of task-specific data, can effectively steer black-box LLMs toward application-specific content generation. Our approach views next-token prediction through the lens of supervised classification. We show that aligning black-box LLMs with task-specific data can be formulated as a label noise correction problem, leading to Plugin model -- an autoregressive probability reweighting model that operates solely on logits. We provide theoretical justification for why reweighting logits alone is sufficient for task adaptation. Extensive experiments with multiple datasets, LLMs, and reweighting models demonstrate the effectiveness of our method, advocating for broader access to token logits in closed-source models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。