arXiv:2410.04612cs.LGcs.AI2024-10ICLR被引 30

解决多轮对话中强化学习的分布偏移问题,让模型更稳定地优化对话策略。

Regressing the Relative Future: Efficient Policy Optimization for Multi-turn RLHF

  • 用单模型回归预测未来奖励,基于自生成数据训练,避免历史策略依赖
  • 在长对话任务中,80亿参数模型超越700亿参数基线模型表现
  • 方法简洁易实现,适合需要长期规划的对话系统优化

大型语言模型在单轮任务(如摘要)上已取得显著进展,但在需长期规划的多轮对话任务中仍存在挑战。以往方法将多轮对话视为长上下文处理,导致训练与实际运行时的分布偏移问题:训练数据中前序对话由参考策略生成,低误差未必代表真实对话中表现好。为此,我们提出一种名为REFUEL(Regressing the Relative Future)的高效策略优化方法,通过单模型估计$Q$值并使用自生成数据训练,缓解该问题。该方法将多轮强化学习从人类反馈(RLHF)转化为一系列迭代收集数据集上的回归任务,便于实现。理论上,我们证明了REFUEL可达到训练集中任意策略的性能。实验中,以Llama-3.1-70B-it模拟用户与模型对话,结果表明,REFUEL在多种设置下持续优于DPO和REBEL等先进方法。值得注意的是,仅80亿参数的Llama-3-8B-it经REFUEL微调后,在长对话中表现超越原版700亿参数模型。代码与模型见https://github.com/ZhaolinGao/REFUEL/ 和 https://huggingface.co/Cornell-AGI。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have achieved remarkable success at tasks like summarization that involve a single turn of interaction. However, they can still struggle with multi-turn tasks like dialogue that require long-term planning. Previous works on multi-turn dialogue extend single-turn reinforcement learning from human feedback (RLHF) methods to the multi-turn setting by treating all prior dialogue turns as a long context. Such approaches suffer from covariate shift: the conversations in the training set have previous turns generated by some reference policy, which means that low training error may not necessarily correspond to good performance when the learner is actually in the conversation loop. In response, we introduce REgressing the RELative FUture (REFUEL), an efficient policy optimization approach designed to address multi-turn RLHF in LLMs. REFUEL employs a single model to estimate $Q$-values and trains on self-generated data, addressing the covariate shift issue. REFUEL frames the multi-turn RLHF problem as a sequence of regression tasks on iteratively collected datasets, enabling ease of implementation. Theoretically, we prove that REFUEL can match the performance of any policy covered by the training set. Empirically, we evaluate our algorithm by using Llama-3.1-70B-it to simulate a user in conversation with our model. REFUEL consistently outperforms state-of-the-art methods such as DPO and REBEL across various settings. Furthermore, despite having only 8 billion parameters, Llama-3-8B-it fine-tuned with REFUEL outperforms Llama-3.1-70B-it on long multi-turn dialogues. Implementation of REFUEL can be found at https://github.com/ZhaolinGao/REFUEL/, and models trained by REFUEL can be found at https://huggingface.co/Cornell-AGI.

多轮对话强化学习策略优化高效训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。