用离散扩散模型实现隐式搜索,提升棋类决策能力
Implicit Search via Discrete Diffusion: A Study on Chess
- 通过离散扩散建模预测未来局面,实现不依赖显式搜索的规划
- 在走子准确率上比单步策略高19.2%,比MCTS增强策略高14%
- 解谜能力提升30%,对弈强度提高540 Elo,适合强化学习与推理研究者
在后AlphaGo时代,蒙特卡洛树搜索(MCTS)等搜索技术重新受到关注,尤其在大型语言模型(LLMs)中的应用。当前的逐词预测模型常缺乏长期规划能力。能否在模型中内嵌类似搜索的能力,以增强规划而不依赖显式搜索?我们提出DiffuSearch,一种通过离散扩散建模展望未来世界来实现隐式搜索的方法。我们在经典棋类游戏国际象棋上验证该方法,其表现优于无搜索和基于显式搜索的策略。实验表明,DiffuSearch在动作准确率上比单步策略高出19.2%,比MCTS增强策略高14%;在解谜能力上提升30%,对弈强度提升540 Elo。结果表明,基于离散扩散的隐式搜索是替代单步策略的有效方案。代码已公开于https://github.com/HKUNLP/DiffuSearch。
原文摘要 · Abstract (English)
In the post-AlphaGo era, there has been a renewed interest in search techniques such as Monte Carlo Tree Search (MCTS), particularly in their application to Large Language Models (LLMs). This renewed attention is driven by the recognition that current next-token prediction models often lack the ability for long-term planning. Is it possible to instill search-like abilities within the models to enhance their planning abilities without relying on explicit search? We propose DiffuSearch , a model that does \textit{implicit search} by looking into the future world via discrete diffusion modeling. We instantiate DiffuSearch on a classical board game, Chess, where explicit search is known to be essential. Through extensive controlled experiments, we show DiffuSearch outperforms both the searchless and explicit search-enhanced policies. Specifically, DiffuSearch outperforms the one-step policy by 19.2% and the MCTS-enhanced policy by 14% on action accuracy. Furthermore, DiffuSearch demonstrates a notable 30% enhancement in puzzle-solving abilities compared to explicit search-based policies, along with a significant 540 Elo increase in game-playing strength assessment. These results indicate that implicit search via discrete diffusion is a viable alternative to explicit search over a one-step policy. All codes are publicly available at \href{https://github.com/HKUNLP/DiffuSearch}{https://github.com/HKUNLP/DiffuSearch}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。