arXiv:2503.07453cs.LGcs.AI2025-03被引 33

用预训练模型提升语言模型强化学习探索效率,关键在运行时计算与覆盖度。

Is a Good Foundation Necessary for Efficient Reinforcement Learning? The Computational Role of the Base Model in Exploration

  • 通过采样查询接口,设计新框架实现高效探索。
  • 覆盖度决定算法下限,单轮探索需序列级覆盖,多轮可降为词级。
  • 运行时计算能显著提速,训练干预无效于多项式时间。

基于语言模型对齐(或强化学习)的主动探索方法,通过鼓励模型生成多样且信息丰富的响应,有望实现超人类能力。然而,当前对语言模型中计算高效的探索算法设计原理理解有限。为更好利用强大预训练生成模型提升探索效率,我们提出一种新的语言模型强化学习计算框架,学习者通过采样预言机与模型交互。聚焦线性softmax参数化,揭示了高效探索的计算-统计权衡:1. 必要性:覆盖度指预训练模型覆盖近优响应的程度,虽非数据效率所必需,但下界约束任意算法的运行时间;2. 推理时探索:提出新算法SpannerSampling,当预训练模型具备足够覆盖度时,可实现最优数据效率与计算效率,匹配理论下界;该方法利用推理时计算缩小探索有效搜索空间;3. 训练干预不足:反例表明,训练时构建合理策略无法在多项式时间内获得类似保证;4. 多轮探索优势:在额外表示假设下,通过多轮探索可实现更优运行时间,将序列级覆盖需求降至词级覆盖。

原文摘要 · Abstract (English)

Language model alignment (or, reinforcement learning) techniques that leverage active exploration -- deliberately encouraging the model to produce diverse, informative responses -- offer the promise of super-human capabilities. However, current understanding of algorithm design primitives for computationally efficient exploration with language models is limited. To better understand how to leverage access to powerful pre-trained generative models to improve the efficiency of exploration, we introduce a new computational framework for RL with language models, in which the learner interacts with the model through a sampling oracle. Focusing on the linear softmax model parameterization, we provide new results that reveal the computational-statistical tradeoffs of efficient exploration: 1. Necessity of coverage: Coverage refers to the extent to which the pre-trained model covers near-optimal responses -- a form of hidden knowledge. We show that coverage, while not necessary for data efficiency, lower bounds the runtime of any algorithm in our framework. 2. Inference-time exploration: We introduce a new algorithm, SpannerSampling, which obtains optimal data efficiency and is computationally efficient whenever the pre-trained model enjoys sufficient coverage, matching our lower bound. SpannerSampling leverages inference-time computation with the pre-trained model to reduce the effective search space for exploration. 3. Insufficiency of training-time interventions: We contrast the result above by showing that training-time interventions that produce proper policies cannot achieve similar guarantees in polynomial time. 4. Computational benefits of multi-turn exploration: Finally, we show that under additional representational assumptions, one can achieve improved runtime (replacing sequence-level coverage with token-level coverage) through multi-turn exploration.

强化学习语言模型探索效率推理优化

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