Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Forest-of-Thought: Scaling Test-Time Compute for Enhancing LLM Reasoning

About

Large Language Models (LLMs) have demonstrated remarkable abilities across various language tasks, but solving complex reasoning problems remains a significant challenge. While existing methods, such as Chain-of-Thought (CoT) and Tree-of-Thought (ToT), enhance reasoning by decomposing problems or structuring prompts, they typically perform a single pass of reasoning and may fail to revisit flawed paths, compromising accuracy. To address this limitation, we propose a novel reasoning framework called Forest-of-Thought (FoT), which integrates multiple reasoning trees to leverage collective decision-making for solving complex logical problems. FoT employs sparse activation strategies to select the most relevant reasoning paths, improving both efficiency and accuracy. Additionally, we introduce a dynamic self-correction strategy that enables real-time error correction, along with consensus-guided decision-making strategies to optimize both correctness and computational resources. Experimental results demonstrate that the FoT framework, combined with these strategies, significantly enhances the reasoning capabilities of LLMs, enabling them to solve complex tasks with greater precision and efficiency. Code will be available at https://github.com/iamhankai/Forest-of-Thought.

Zhenni Bi, Kai Han, Chuanjian Liu, Yehui Tang, Yunhe Wang• 2024

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningGSM8K (test)
Accuracy94
797
ReasoningBBH--
507
Arithmetic ReasoningGSM8K
Accuracy94.2
155
Long-context ReasoningLongBench
Score59.3
62
Multi-hop ReasoningMuSiQue
EM36.2
41
Mathematical ReasoningMATH
EM82.7
38
Counterfactual reasoningMMLU CF
EM70.6
30
Mathematical ReasoningOlympiadBench
EM12.6
30
Multi-hop Question AnsweringHotpotQA N=1,000 (test)
F1 Score66.7
23
Arithmetic ReasoningGame of 24 95 (test)
Success Rate93.7
9
Showing 10 of 14 rows

Other info

Follow for update