arXiv:2505.05701cs.AIcs.LG2025-05NeurIPS被引 2

用预训练共享Q网络提升离线强化学习的数据效率,小数据也能跑出好效果。

Pretraining a Shared Q-Network for Data-Efficient Offline Reinforcement Learning

  • 设计共享Q网络,同时预测下一状态和Q值,实现双任务预训练
  • 仅用10%数据就超越标准算法在全量数据下的表现
  • 适配性强,对不同数据质量与分布均有效,适合数据稀缺场景

离线强化学习旨在不与环境交互的情况下,仅从静态数据集中学习策略。然而,收集足够大的数据集耗时巨大,尤其在环境交互受限时尤为困难。因此,如何以最小的静态数据集高效学习最优策略,是离线RL的核心挑战,类似于在线RL中的样本效率问题。本文提出一种简单有效的即插即用预训练方法,通过监督回归任务初始化Q网络特征,提升数据效率。具体地,引入共享Q网络结构,同时输出下一状态预测与Q值;利用多种离线RL方法对共享网络进行预训练。大量实验表明,该方法显著提升D4RL、Robomimic和V-D4RL等基准上现有主流离线RL方法的性能。进一步验证显示,该方法在不同数据质量与分布下(涵盖D4RL和ExoRL)均显著提升数据效率。值得注意的是,仅使用10%数据的方法,在全量数据下仍优于标准算法。

原文摘要 · Abstract (English)

Offline reinforcement learning (RL) aims to learn a policy from a static dataset without further interactions with the environment. Collecting sufficiently large datasets for offline RL is exhausting since this data collection requires colossus interactions with environments and becomes tricky when the interaction with the environment is restricted. Hence, how an agent learns the best policy with a minimal static dataset is a crucial issue in offline RL, similar to the sample efficiency problem in online RL. In this paper, we propose a simple yet effective plug-and-play pretraining method to initialize a feature of a Q-network to enhance data efficiency in offline RL. Specifically, we introduce a shared Q-network structure that outputs predictions of the next state and Q-value. We pretrain the shared Q-network through a supervised regression task that predicts a next state and trains the shared Q-network using diverse offline RL methods. Through extensive experiments, we empirically demonstrate that our method enhances the performance of existing popular offline RL methods on the D4RL, Robomimic and V-D4RL benchmarks. Furthermore, we show that our method significantly boosts data-efficient offline RL across various data qualities and data distributions trough D4RL and ExoRL benchmarks. Notably, our method adapted with only 10% of the dataset outperforms standard algorithms even with full datasets.

离线RL数据效率预训练Q网络

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