通过随机分布蒸馏提升强化学习探索效率
Exploration by Random Distribution Distillation
- 从目标网络采样正态分布输出,显式构造内在奖励
- 结合伪计数与预测差异项,实现探索衰减与高维空间适应
- 适合需要高效在线探索的强化学习任务
探索仍是在线强化学习中的关键挑战,智能体需有效探索未知环境以获取高回报。现有主要探索方法为基于计数与基于好奇性的方法,其中预测误差方法尤为突出。本文提出一种新方法——随机分布蒸馏(RDD),通过从正态分布中采样目标网络输出,将预测网络与目标网络间的差异作为内在奖励。引入状态对应输出的随机性,并建模为正态分布样本,使内在奖励受两个关键成分约束:伪计数项确保探索衰减,差异项反映预测器收敛。实验表明,RDD能有效统一基于计数与预测误差的方法,在高维空间中保持预测误差方法的优势,同时具备类似伪计数的探索衰减机制。理论分析与实验结果均验证其在提升在线强化学习探索效率上的有效性。
原文摘要 · Abstract (English)
Exploration remains a critical challenge in online reinforcement learning, as an agent must effectively explore unknown environments to achieve high returns. Currently, the main exploration algorithms are primarily count-based methods and curiosity-based methods, with prediction-error methods being a prominent example. In this paper, we propose a novel method called \textbf{R}andom \textbf{D}istribution \textbf{D}istillation (RDD), which samples the output of a target network from a normal distribution. RDD facilitates a more extensive exploration by explicitly treating the difference between the prediction network and the target network as an intrinsic reward. Furthermore, by introducing randomness into the output of the target network for a given state and modeling it as a sample from a normal distribution, intrinsic rewards are bounded by two key components: a pseudo-count term ensuring proper exploration decay and a discrepancy term accounting for predictor convergence. We demonstrate that RDD effectively unifies both count-based and prediction-error approaches. It retains the advantages of prediction-error methods in high-dimensional spaces, while also implementing an intrinsic reward decay mode akin to the pseudo-count method. In the experimental section, RDD is compared with more advanced methods in a series of environments. Both theoretical analysis and experimental results confirm the effectiveness of our approach in improving online exploration for reinforcement learning tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。