中文

基于最优传输不确定性传播的蒙特卡洛树搜索

人工智能 2023-09-20 v1

摘要

本文针对高度随机且部分可观测的马尔可夫决策过程,提出了一种新颖的蒙特卡洛树搜索(MCTS)回溯策略。我们采用概率方法,将价值节点与动作价值节点均建模为高斯分布。我们引入一种新颖的回溯算子,将价值节点计算为其动作价值子节点的Wasserstein重心,从而将估计的不确定性沿树传播至根节点。我们研究了将 L1L^1-Wasserstein重心与 α\alpha-散度新颖结合时所用的该回溯算子,并指出了其与广义均值回溯算子的显著联系。我们以基于乐观选择与Thompson采样的两种采样策略补充该概率回溯算子,得到我们的Wasserstein MCTS算法。我们提供了渐近收敛至最优策略的理论保证,并在多个随机与部分可观测环境中进行实证评估,结果表明我们的方法优于知名相关基线。

关键词

引用

@article{arxiv.2309.10737,
  title  = {Monte-Carlo tree search with uncertainty propagation via optimal transport},
  author = {Tuan Dam and Pascal Stenger and Lukas Schneider and Joni Pajarinen and Carlo D'Eramo and Odalric-Ambrym Maillard},
  journal= {arXiv preprint arXiv:2309.10737},
  year   = {2023}
}