arXiv:2505.23062cs.LGcs.AI2025-05NeurIPS被引 11

用流模型融合离线数据,解决在线学习时动态差异问题

Composite Flow Matching for Reinforcement Learning with Shifted-Dynamics Data

  • 构建复合流模型,将在线动态建模为离线流的条件输出
  • 利用Wasserstein距离稳定估算动态差距,避免传统方法失效
  • 主动探索高差距区域,提升样本效率,适合动态不一致场景

引入预收集的离线数据可显著提升强化学习的样本效率,但当离线数据与在线环境的转移动态不一致时,其优势可能消失。现有方法通常通过惩罚或过滤动态差距大的区域来缓解,但其动态差距估计依赖于KL散度或互信息,当离线与在线动态支持集不匹配时可能失效。为此,我们提出CompFlow,一个基于流匹配与最优传输理论的框架。具体地,将在线动态建模为预训练离线流输出分布的条件流,而非直接从高斯先验学习。该复合结构带来双重优势:(1) 在有限交互数据下提升在线动态学习的泛化能力;(2) 通过离线与在线转移间的Wasserstein距离获得良好定义且稳定的动态差距估计。基于此,我们进一步设计了一种乐观主动数据采集策略,优先探索高差距区域,并理论上证明其可缩小与最优策略的性能差距。实验表明,CompFlow在多种具有动态偏移的强化学习基准上持续优于强基线。

原文摘要 · Abstract (English)

Incorporating pre-collected offline data can substantially improve the sample efficiency of reinforcement learning (RL), but its benefits can break down when the transition dynamics in the offline dataset differ from those encountered online. Existing approaches typically mitigate this issue by penalizing or filtering offline transitions in regions with large dynamics gap. However, their dynamics-gap estimators often rely on KL divergence or mutual information, which can be ill-defined when offline and online dynamics have mismatched support. To address this challenge, we propose CompFlow, a principled framework built on the theoretical connection between flow matching and optimal transport. Specifically, we model the online dynamics as a conditional flow built upon the output distribution of a pretrained offline flow, rather than learning it directly from a Gaussian prior. This composite structure provides two advantages: (1) improved generalization when learning online dynamics under limited interaction data, and (2) a well-defined and stable estimate of the dynamics gap via the Wasserstein distance between offline and online transitions. Building on this dynamics-gap estimator, we further develop an optimistic active data collection strategy that prioritizes exploration in high-gap regions, and show theoretically that it reduces the performance gap to the optimal policy. Empirically, CompFlow consistently outperforms strong baselines across a range of RL benchmarks with shifted-dynamics data.

强化学习流模型离线学习动态对齐

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