用监督微调思想让大模型学好多轮决策,无需重设参数
Q-SFT: Q-Learning for Language Models via Supervised Fine-Tuning
- 把Q值学习转为监督微调,直接用词概率表示Q值
- 在对话和视觉导航任务上优于传统方法,保留预训练优势
- 适合想用大模型做多步决策的开发者,免去复杂改造
基于值函数的强化学习理论上可解决多轮任务,如对话、机器人控制等,尤其适用于离线强化学习。然而,尽管策略梯度法已广泛用于大语言模型的单轮任务(如问答),将基于值函数的方法扩展到大模型的离线多轮设置仍面临巨大挑战:需有效利用预训练知识、适配数十亿参数的模型架构,并在大规模数据集上训练。本文提出Q-SFT算法,将Q-learning转化为一种改进的监督微调(SFT)问题,使词元概率直接对应Q值。该方法在理论上有类似顶尖Q-learning的性能保证,实践中又接近标准SFT目标。它无需在强化学习微调前重置权重,也无需新增价值或优势预测头。我们在预训练的语言模型和视觉语言模型上进行了评估,涵盖自然语言对话及图像驱动的机器人操控与导航任务,结果表明其在多个场景下表现优异。
原文摘要 · Abstract (English)
Value-based reinforcement learning (RL) can in principle learn effective policies for a wide range of multi-turn problems, from games to dialogue to robotic control, including via offline RL from static previously collected datasets. However, despite the widespread use of policy gradient methods to train large language models for single turn tasks (e.g., question answering), value-based methods for multi-turn RL in an off-policy or offline setting have proven particularly challenging to scale to the setting of large language models. This setting requires effectively leveraging pretraining, scaling to large architectures with billions of parameters, and training on large datasets, all of which represent major challenges for current value-based RL methods. In this work, we propose a novel offline RL algorithm that addresses these drawbacks, casting Q-learning as a modified supervised fine-tuning (SFT) problem where the probabilities of tokens directly translate to Q-values. In this way we obtain an algorithm that smoothly transitions from maximizing the likelihood of the data during pretraining to learning a near-optimal Q-function during finetuning. Our algorithm has strong theoretical foundations, enjoying performance bounds similar to state-of-the-art Q-learning methods, while in practice utilizing an objective that closely resembles SFT. Because of this, our approach can enjoy the full benefits of the pretraining of language models, without the need to reinitialize any weights before RL finetuning, and without the need to initialize new heads for predicting values or advantages. Empirically, we evaluate our method on both pretrained LLMs and VLMs, on a variety of tasks including both natural language dialogue and robotic manipulation and navigation from images.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。