arXiv:2605.14497cs.LGcs.AI2026-05中稿 · IJCAI

提出自适应数据混合方法,让离线到在线强化学习更稳定高效。

ROAD: Adaptive Data Mixing for Offline-to-Online Reinforcement Learning via Bi-Level Optimization

论文配图:ROAD: Adaptive Data Mixing for Offline-to-Online Reinforcement Learning via Bi-Level Optimization
图 1 · 摘自论文原文
  • 用双层优化建模数据混合策略,自动调节离线与在线数据比例。
  • 在多个数据集上超越现有方法,提升训练稳定性和最终性能。
  • 无需人工调参,适合需要快速适配新环境的强化学习场景。

离线到在线强化学习结合了离线预训练的稳定性与在线微调的灵活性。关键挑战在于离线数据分布与不断演化的在线策略之间存在非平稳分布偏移。现有方法多依赖静态混合比例或启发式重放缓冲策略,缺乏对不同环境和训练动态的适应性,导致稳定性与最终性能之间的权衡不佳。本文提出面向优化自适应数据混合的强化学习框架(ROAD),通过双层优化形式化数据选择问题:将数据混合策略视为控制政策性能的元决策(外层),而传统的Q-learning更新作为内层。为实现可计算性,引入基于多臂赌博机机制的实用算法,其由代理目标近似双层梯度驱动,同时保持离线先验并防止价值过估计。实验表明,该方法在多个数据集上持续优于现有数据重放方法,无需人工、情境特定调整,实现了更优的稳定性和渐近性能。

原文摘要 · Abstract (English)

Offline-to-online reinforcement learning harnesses the stability of offline pretraining and the flexibility of online fine-tuning. A key challenge lies in the non-stationary distribution shift between offline datasets and the evolving online policy. Common approaches often rely on static mixing ratios or heuristic-based replay strategies, which lack adaptability to different environments and varying training dynamics, resulting in suboptimal tradeoff between stability and asymptotic performance. In this work, we propose Reinforcement Learning with Optimized Adaptive Data-mixing (ROAD), a dynamic plug-and-play framework that automates the data replay process. We identify a fundamental objective misalignment in existing approaches. To tackle this, we formulate the data selection problem as a bi-level optimization process, interpreting the data mixing strategy as a meta-decision governing the policy performance (outer-level) during online fine-tuning, while the conventional Q-learning updates operate at the inner level. To make it tractable, we propose a practical algorithm using a multi-armed bandit mechanism. This is guided by a surrogate objective approximating the bi-level gradient, which simultaneously maintains offline priors and prevents value overestimation. Our empirical results demonstrate that this approach consistently outperforms existing data replay methods across various datasets, eliminating the need for manual, context-specific adjustments while achieving superior stability and asymptotic performance.

强化学习自适应混合双层优化在线学习

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