Implicit Search via Discrete Diffusion: A Study on Chess

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Ye, Jiacheng, Wu, Zhenyu, Gao, Jiahui, Wu, Zhiyong, Jiang, Xin, Li, Zhenguo, Kong, Lingpeng
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915175650557952
author Ye, Jiacheng
Wu, Zhenyu
Gao, Jiahui
Wu, Zhiyong
Jiang, Xin
Li, Zhenguo
Kong, Lingpeng
author_facet Ye, Jiacheng
Wu, Zhenyu
Gao, Jiahui
Wu, Zhiyong
Jiang, Xin
Li, Zhenguo
Kong, Lingpeng
contents 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}.
format Preprint
id arxiv_https___arxiv_org_abs_2502_19805
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Implicit Search via Discrete Diffusion: A Study on Chess
Ye, Jiacheng
Wu, Zhenyu
Gao, Jiahui
Wu, Zhiyong
Jiang, Xin
Li, Zhenguo
Kong, Lingpeng
Machine Learning
Artificial Intelligence
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}.
title Implicit Search via Discrete Diffusion: A Study on Chess
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2502.19805