研究想象中训练的强化学习,如何平衡模型误差与采样成本。
On Training in Imagination
- 通过理论分析优化想象轨迹的采样比例,降低回报误差。
- 发现零均值奖励噪声不影响梯度估计,仅增加可下降的方差。
- 给出预算固定下的采样策略选择:多跑无噪还是少跑有噪?
基于模型的强化学习方法在想象的轨迹上训练策略,这些轨迹由学习到的动力学模型生成,并由学习到的奖励模型评分,而不需在策略更新时查询真实环境。本文通过量化学习到的动力学和奖励模型中的误差对回报及策略优化的影响,深入研究这一训练范式。首先,将Asadi等(2018)的分析扩展至具有学习奖励模型的MDP,推导出在幂律缩放假设下最小化回报误差界所需的最优样本分配比例(动力学样本与奖励样本之比)。识别出学习动力学、奖励和策略的低利普希茨常数作为表征设计目标,可收紧该误差界,并将其与Wang等(2026)的时间拉直目标相联系。其次,研究使用REINFORCE进行策略优化时对噪声奖励的容忍性。证明零均值奖励噪声使梯度估计无偏,并引入一个随轨迹数量减少的方差项。这带来实际权衡:在固定预算下,应选择更多但更便宜的噪声奖励轨迹,还是更少但更昂贵的低噪声轨迹?我们将此选择简化为一维优化问题,并刻画了最优解。
原文摘要 · Abstract (English)
State-of-the-art model-based reinforcement learning methods train policies on imagined rollouts. These rollouts are trajectories generated by a learned dynamics model and are scored by a learned reward model, but without querying the true environment during policy updates. We study this training paradigm by quantifying how errors in learned dynamics and reward models affect returns and policy optimization. First, we extend the analysis of Asadi et al. (2018) to MDPs with learned reward models, and derive the optimal sample allocation--the ratio of dynamics samples to reward samples that minimizes a bound on return error under power-law scaling assumptions. We identify lower Lipschitz constants of the learned dynamics, reward, and policy as a representation desideratum that tightens this bound, and we connect this perspective to the temporal-straightening objective of Wang et al. (2026). Second, we examine how policy optimization with REINFORCE tolerates noisy rewards, which are often cheaper to obtain. We show that zero-mean reward noise leaves the gradient estimator unbiased and adds at most a variance term that decreases with the number of rollouts. This introduces a practical tradeoff: given a fixed budget, should one buy more rollouts with cheaper but noisier rewards, or fewer rollouts with more expensive but less noisy rewards? We reduce this choice to a one-dimensional optimization problem and characterize the optimum.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。