arXiv:2601.01301cs.AIcs.LG2026-01

用批量推理加速蒙特卡洛树搜索,速度提升超40倍。

Accelerating Monte-Carlo Tree Search with Optimized Posterior Policies

  • 采用广度优先遍历与后验策略递归优化,实现网络推理批量处理。
  • 单状态搜索快40倍以上,批量搜索快3倍,训练时间缩短至1/3。
  • 适合追求推理速度的强化学习部署场景,尤其游戏AI应用。

我们提出一种递归式AlphaZero风格的蒙特卡洛树搜索算法RMCTS。与AlphaZero的MCTS-UCB相比,RMCTS通过广度优先方式探索搜索树,使网络推理自然形成大批次,显著降低GPU延迟成本。实验表明,在单个根状态搜索中,RMCTS比MCTS-UCB快40倍以上;在大规模根状态搜索中,速度提升约3倍。其递归机制基于从叶节点向根节点逐层计算优化后的后验策略,该策略源自“Monte--Carlo tree search as regularized policy optimization”(Grill, et al.),即在估计动作奖励基础上,最大化期望回报并惩罚偏离先验策略。与MCTS-UCB自适应构建树不同,RMCTS的树结构由每个节点的先验网络策略决定,虽牺牲一定灵活性,但提速优势更显著。实践中,用RMCTS训练的网络在约1/3的训练时间内即可达到与MCTS-UCB相当的性能。我们在连接四子、点连线和奥赛罗三个游戏中进行了时间和质量对比。

原文摘要 · Abstract (English)

We introduce a recursive AlphaZero-style Monte--Carlo tree search algorithm, "RMCTS". The advantage of RMCTS over AlphaZero's MCTS-UCB is speed. In RMCTS, the search tree is explored in a breadth-first manner, so that network inferences naturally occur in large batches. This significantly reduces the GPU latency cost. We find that RMCTS is often more than 40 times faster than MCTS-UCB when searching a single root state, and about 3 times faster when searching a large batch of root states. The recursion in RMCTS is based on computing optimized posterior policies at each game state in the search tree, starting from the leaves and working back up to the root. Here we use the posterior policy explored in "Monte--Carlo tree search as regularized policy optimization" (Grill, et al.) Their posterior policy is the unique policy which maximizes the expected reward given estimated action rewards minus a penalty for diverging from the prior policy. The tree explored by RMCTS is not defined in an adaptive manner, as it is in MCTS-UCB. Instead, the RMCTS tree is defined by following prior network policies at each node. This is a disadvantage, but the speedup advantage is more significant, and in practice we find that RMCTS-trained networks match the quality of MCTS-UCB-trained networks in roughly one-third of the training time. We include timing and quality comparisons of RMCTS vs. MCTS-UCB for three games: Connect-4, Dots-and-Boxes, and Othello.

强化学习树搜索推理加速

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