中文

TDprop:Jacobi 预条件处理是否有助于时序差分学习?

机器学习 2020-07-07 v1 机器学习

摘要

我们研究了在时序差分(TD)学习中考虑 bootstrap 项的 Jacobi 预条件处理是否能帮助提升自适应优化器的性能。我们的方法 TDprop 基于 TD 更新规则的对角预条件计算逐参数学习率。我们展示了如何将其用于 nn 步回报与 TD(λ\lambda)。我们的理论发现表明,若通过超参数搜索为两者找到最优学习率,则包含这一额外预条件信息出人意料地与普通半梯度 TD 相当。在使用 Expected SARSA 的深度强化学习实验中,在近最优学习率下,TDprop 在所有测试游戏中的表现达到或超过 Adam,但调优良好的 SGD 也能产生类似改进——这与我们的理论一致。我们的发现表明,Jacobi 预条件处理可能改进深度强化学习中典型的自适应优化方法,但尽管融入了来自 TD bootstrap 项的额外信息,却未必总是优于 SGD。

关键词

引用

@article{arxiv.2007.02786,
  title  = {TDprop: Does Jacobi Preconditioning Help Temporal Difference Learning?},
  author = {Joshua Romoff and Peter Henderson and David Kanaa and Emmanuel Bengio and Ahmed Touati and Pierre-Luc Bacon and Joelle Pineau},
  journal= {arXiv preprint arXiv:2007.02786},
  year   = {2020}
}

备注

Presented at the Theoretical Foundations of Reinforcement Learning workshop at ICML 2020