用流匹配训练价值函数,让强化学习模型更高效地利用计算资源。
floq: Training Critics via Flow-Matching for Scaling Compute in Value-Based RL
- 通过流匹配构建可迭代计算的价值函数,提升模型表达能力。
- 在离线和在线任务中性能提升近1.8倍,优于传统方法。
- 适合追求高容量、高效率价值学习的强化学习研究者。
现代大规模机器学习常采用提供密集监督的训练目标,如语言模型中的教师强制或扩散模型中的逐步去噪,使模型能泛化地学习复杂函数。受此启发,我们研究了时序差分(TD)方法在强化学习中迭代计算的潜力。传统方法通常以整体形式表示价值函数,缺乏迭代计算机制。本文提出floq(流匹配Q函数),使用速度场参数化Q函数,并借助生成建模中的流匹配技术进行训练。该速度场通过TD学习目标训练,从由多步数值积分生成的目标速度场中自举。关键优势在于,通过调整积分步数,可灵活控制和扩展Q函数容量。在一系列挑战性离线强化学习基准和在线微调任务中,floq性能提升近1.8倍,显著优于标准TD架构,凸显了迭代计算在价值学习中的巨大潜力。
原文摘要 · Abstract (English)
A hallmark of modern large-scale machine learning techniques is the use of training objectives that provide dense supervision to intermediate computations, such as teacher forcing the next token in language models or denoising step-by-step in diffusion models. This enables models to learn complex functions in a generalizable manner. Motivated by this observation, we investigate the benefits of iterative computation for temporal difference (TD) methods in reinforcement learning (RL). Typically they represent value functions in a monolithic fashion, without iterative compute. We introduce floq (flow-matching Q-functions), an approach that parameterizes the Q-function using a velocity field and trains it using techniques from flow-matching, typically used in generative modeling. This velocity field underneath the flow is trained using a TD-learning objective, which bootstraps from values produced by a target velocity field, computed by running multiple steps of numerical integration. Crucially, floq allows for more fine-grained control and scaling of the Q-function capacity than monolithic architectures, by appropriately setting the number of integration steps. Across a suite of challenging offline RL benchmarks and online fine-tuning tasks, floq improves performance by nearly 1.8x. floq scales capacity far better than standard TD-learning architectures, highlighting the potential of iterative computation for value learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。