用奖励模型实现跨模型可复用微调,降低推理开销。
Portable Reward Tuning: Towards Reusable Fine-Tuning across Different Pretrained Models
- 将微调重定义为奖励最大化,训练独立奖励模型
- 在多模态任务中达到与现有方法相当的准确率
- 适合需频繁更换基础模型的部署场景
尽管通过微调可让基础模型胜任各类专家任务,但其知识过时或能力受限终将导致模型更新。每次更换模型都需重新微调,成本高昂。现有工作采用推理时微调,即利用旧模型及其微调版本的输出来调整新模型的输出概率,但引入额外推理开销。本文提出一种新微调范式——可移植奖励微调(Portable Reward Tuning, PRT),基于微调即奖励最大化的重构,不直接微调基础模型参数,而是通过相同损失函数显式训练奖励模型。推理时,该奖励模型可与任意同词汇表的基础模型结合使用。实验表明,涵盖视觉与语言模型的任务中,PRT实现的精度与现有推理时微调方法相当,且推理成本更低。
原文摘要 · Abstract (English)
While foundation models have been exploited for various expert tasks through fine-tuning, any foundation model will become outdated due to its old knowledge or limited capability. Thus the underlying foundation model should be eventually replaced by new ones, which leads to repeated cost of fine-tuning these new models. Existing work addresses this problem by inference-time tuning, i.e., modifying the output probabilities from the new foundation model with the outputs from the old foundation model and its fine-tuned model, which involves an additional overhead in inference by the latter two models. In this paper, we propose a new fine-tuning principle, Portable Reward Tuning (PRT), that reduces the inference overhead by its nature, based on the reformulation of fine-tuning as the reward maximization. Specifically, instead of fine-tuning parameters of the foundation models, PRT trains the reward model explicitly through the same loss function as in fine-tuning. During inference, the reward model can be used with any foundation model (with the same set of vocabularies or labels) through the formulation of reward maximization. Experimental results, covering both vision and language models, demonstrate that the PRT-trained model can achieve comparable accuracy to the existing work of inference-time tuning, with less inference cost.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。