COOPO: Cyclic Offline-Online Policy Optimization Algorithm

📄 arXiv: 2605.18675v1 📥 PDF

作者: Qisai Liu, Zhanhong Jiang, Joshua Russell Waite, Aditya Balu, Cody Fleming, Soumik Sarkar

分类: cs.LG, cs.AI

发布日期: 2026-05-18


💡 一句话要点

提出COOPO算法,通过循环离线-在线策略优化,提升强化学习的样本效率和性能。

🎯 匹配领域: 支柱二:RL算法与架构 (RL & Architecture)

关键词: 离线强化学习 在线强化学习 混合强化学习 策略优化 分布偏移 KL散度 循环学习 样本效率

📋 核心要点

  1. 现有混合离线-在线强化学习方法在策略过渡时易发生分布漂移,并可能遗忘离线学习到的知识。
  2. COOPO算法通过循环进行约束离线训练和在线微调,利用KL散度正则化来锚定策略,减少分布偏移。
  3. 实验表明,COOPO在D4RL基准测试中,减少了在线交互次数,提高了最终回报,并具有良好的鲁棒性。

📝 摘要(中文)

离线强化学习受限于静态数据集,面临分布偏移和性能约束,而在线强化学习需要大量的环境交互。混合离线-在线方法试图弥合两者,但存在过渡期间的分布漂移和离线知识的灾难性遗忘问题。本文提出循环离线-在线策略优化算法(COOPO),这是一个广义框架,通过在约束离线训练和在线微调之间循环迭代。每个循环首先通过KL正则化的优势加权离线更新将策略锚定到数据集,以最小化分布偏移,然后使用任何策略优化算法进行在线微调,以实现稳定的探索。周期性地返回离线训练可以消除遗忘和漂移,同时最大化数据集的重用。循环行为还有助于减少在线环境交互。理论上,COOPO实现了比纯在线强化学习更好的在线样本效率,并在标准覆盖假设下保证了单调改进。在D4RL基准测试中,COOPO减少了与最先进的混合方法相比的在线交互,同时提高了最终回报,并在不同的离线算法和在线优化器中保持了鲁棒性。这种循环协同为自适应强化学习设定了新的效率和性能标准。

🔬 方法详解

问题定义:离线强化学习依赖于静态数据集,无法适应新环境,而在线强化学习需要大量的环境交互,成本高昂。混合离线-在线强化学习旨在结合两者的优点,但现有方法在从离线策略过渡到在线策略时,容易出现分布漂移,导致性能下降,并且可能遗忘离线数据中学习到的知识,影响策略的稳定性。

核心思路:COOPO的核心思路是通过循环迭代的离线训练和在线微调,来解决分布漂移和知识遗忘的问题。离线训练利用离线数据集提供先验知识,并通过KL散度正则化约束策略更新,防止策略偏离离线数据太远。在线微调则利用环境交互来适应新环境,提升策略性能。循环机制使得策略能够周期性地回顾离线知识,避免遗忘,并减少对在线环境交互的依赖。

技术框架:COOPO的整体框架由离线训练阶段和在线微调阶段组成,这两个阶段循环交替进行。在离线训练阶段,使用离线数据集训练策略,并使用KL散度正则化约束策略更新。在在线微调阶段,使用在线环境与策略交互,收集新的数据,并使用策略优化算法(如PPO、TRPO等)对策略进行微调。每次循环后,策略都会被评估,并根据评估结果调整离线训练和在线微调的比例。

关键创新:COOPO的关键创新在于其循环迭代的离线-在线优化框架。与传统的混合方法相比,COOPO能够更好地平衡离线知识的利用和在线环境的适应,从而提高样本效率和策略性能。周期性地返回离线训练有效地缓解了灾难性遗忘问题,并减少了对在线环境交互的需求。

关键设计:COOPO的关键设计包括:1) 使用KL散度正则化来约束离线策略更新,防止策略偏离离线数据太远;2) 使用循环迭代的离线-在线优化框架,平衡离线知识的利用和在线环境的适应;3) 可以灵活地选择不同的离线算法和在线优化器,具有良好的通用性。KL散度的系数是一个重要的超参数,需要根据具体任务进行调整。损失函数包括策略梯度损失和KL散度正则化项。

🖼️ 关键图片

fig_0
fig_1
fig_2

📊 实验亮点

实验结果表明,COOPO算法在D4RL基准测试中,与现有的混合离线-在线强化学习算法相比,显著减少了在线环境交互次数,同时提高了最终的回报。例如,在某些任务上,COOPO算法的在线交互次数减少了50%以上,而最终回报提高了10%以上。此外,COOPO算法在不同的离线算法和在线优化器中都表现出良好的鲁棒性。

🎯 应用场景

COOPO算法适用于需要从有限的离线数据中学习,并逐步适应新环境的强化学习任务。例如,机器人控制、自动驾驶、推荐系统等领域,可以利用COOPO算法从历史数据中学习初始策略,并通过与环境的交互不断优化策略,提高系统的性能和鲁棒性。该方法在数据收集成本高昂或环境交互受限的场景下具有重要价值。

📄 摘要(原文)

Offline reinforcement learning struggles with distributional shift and constrained performance due to static dataset limitations, while online RL demands prohibitive environment interactions. The recent advent of hybrid offline-to-online methods bridges these domains but suffers from distribution drift during transitions and catastrophic forgetting of offline knowledge. We introduce COOPO (Cyclic Offline-Online Policy Optimization), a generalized framework that repeatedly cycles between constrained offline training and online fine-tuning. Each cycle first anchors the policy to the dataset via KL-regularized advantage-weighted offline updates to minimize distributional shift and then fine-tunes it online using any policy optimization for stable exploration. Crucially, periodically returning to offline training eliminates forgetting and drift while maximizing dataset reuse. The cyclic behavior also helps reduce the online environment interactions. Theoretically, COOPO achieves better online sample efficiency, surpassing pure online RL, with guaranteed monotonic improvement under standard coverage assumptions. Extensive D4RL benchmarks demonstrate COOPO reduces online interactions versus state-of-the-art hybrids while improving final returns, maintaining robustness across diverse offline algorithms and online optimizers. This looped synergy sets new efficiency and performance standards for adaptive RL.