arXiv:2604.18701cs.LGcs.AI2026-04中稿 · ICML被引 1

用累积预测误差改进探索,让智能体更专注可学习的环境变化。

Curiosity-Critic: Cumulative Prediction Error Improvement as a Tractable Intrinsic Reward for World Model Training

论文配图:Curiosity-Critic: Cumulative Prediction Error Improvement as a Tractable Intrinsic Reward for World Model Training
图 1 · 摘自论文原文
  • 基于累积误差改进设计可计算的内在奖励
  • 在随机网格世界中训练速度与准确率均优于现有方法
  • 自动区分可学习与不可学习的环境变化,适合强化学习探索

传统基于局部预测误差的好奇心奖励只关注当前转移,忽略世界模型在所有访问转移上的累积预测误差。本文提出Curiosity-Critic,将内在奖励建立在累积目标的改进上,并推导出一个可计算的单步近似:当前预测误差与当前状态转移的渐近误差基线之差。该基线通过一个与世界模型协同训练的神经网络批评者在线估计;由于批评者只需学习转移的可预测难度,其对不可约噪声下限的估计在世界模型饱和前即已收敛,从而引导探索聚焦于可学习的转移。奖励在可学习转移处较高,在随机转移处趋于零,实现了在线区分认知性(可消除)与偶然性(不可消除)误差。从Schmidhuber(1991)到特征空间变体等早期方法均为该基线特定近似的特例。在随机网格世界实验中,Curiosity-Critic在训练速度和最终世界模型准确率上均优于预测误差、访问计数和随机网络蒸馏方法。

原文摘要 · Abstract (English)

Local prediction-error-based curiosity rewards focus on the current transition without considering the world model's cumulative prediction error across all visited transitions. We introduce Curiosity-Critic, which grounds its intrinsic reward in the improvement of this cumulative objective, and show that it admits a tractable per-step surrogate: the difference between the current prediction error and the asymptotic error baseline of the current state transition. We estimate this error baseline online with a learned critic co-trained alongside the world model; since the critic only has to learn how hard a transition is to predict, its estimate of the irreducible noise floor converges well before the world model saturates, redirecting exploration toward learnable transitions. The reward is higher for learnable transitions and collapses toward zero for stochastic ones, thereby separating epistemic (reducible) from aleatoric (irreducible) prediction error online. Prior prediction-error curiosity formulations, from Schmidhuber (1991) to learned-feature-space variants, emerge as special cases corresponding to specific approximations of this error baseline. Experiments on a stochastic grid world show that Curiosity-Critic outperforms prediction-error, visitation-count, and Random Network Distillation methods in training speed and final world model accuracy.

强化学习好奇心驱动世界模型探索策略

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