Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Dynamic Parallel Tree Search for Efficient LLM Reasoning

About

Tree of Thoughts (ToT) enhances Large Language Model (LLM) reasoning by structuring problem-solving as a spanning tree. However, recent methods focus on search accuracy while overlooking computational efficiency. The challenges of accelerating the ToT lie in the frequent switching of reasoning focus, and the redundant exploration of suboptimal solutions. To alleviate this dilemma, we propose Dynamic Parallel Tree Search (DPTS), a novel parallelism framework that aims to dynamically optimize the reasoning path in inference. It includes the Parallelism Streamline in the generation phase to build up a flexible and adaptive parallelism with arbitrary paths by fine-grained cache management and alignment. Meanwhile, the Search and Transition Mechanism filters potential candidates to dynamically maintain the reasoning focus on more possible solutions and have less redundancy. Experiments on Qwen-2.5 and Llama-3 with Math500 and GSM8K datasets show that DPTS significantly improves efficiency by 2-4x on average while maintaining or even surpassing existing reasoning algorithms in accuracy, making ToT-based reasoning more scalable and computationally efficient.

Yifu Ding, Wentao Jiang, Shunyu Liu, Yongcheng Jing, Jinyang Guo, Yingjie Wang, Jing Zhang, Zengmao Wang, Ziwei Liu, Bo Du, Xianglong Liu, Dacheng Tao• 2025

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMATH500 (test)
Accuracy76.2
514
Mathematical ReasoningMinerva
Accuracy (Acc)2.17
62
ReasoningGSM8K (test)
EM Accuracy89.4
35
Mathematical ReasoningMATH 500
Accuracy2.5
28
Logical reasoningBoolean BBEH
Accuracy5
12
Geometric ReasoningGeo BBEH
Accuracy7.5
12
Logical reasoningBoardgameQA BBEH
Accuracy0.095
12
Logical reasoningCausal BBEH
Accuracy4
12
General ReasoningGPQA
Accuracy3.44
12
Mathematical ReasoningGSM8K
Accuracy1.72
12
Showing 10 of 11 rows

Other info

Follow for update