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

Tree of Thoughts: Deliberate Problem Solving with Large Language Models

About

Language models are increasingly being deployed for general problem solving across a wide range of tasks, but are still confined to token-level, left-to-right decision-making processes during inference. This means they can fall short in tasks that require exploration, strategic lookahead, or where initial decisions play a pivotal role. To surmount these challenges, we introduce a new framework for language model inference, Tree of Thoughts (ToT), which generalizes over the popular Chain of Thought approach to prompting language models, and enables exploration over coherent units of text (thoughts) that serve as intermediate steps toward problem solving. ToT allows LMs to perform deliberate decision making by considering multiple different reasoning paths and self-evaluating choices to decide the next course of action, as well as looking ahead or backtracking when necessary to make global choices. Our experiments show that ToT significantly enhances language models' problem-solving abilities on three novel tasks requiring non-trivial planning or search: Game of 24, Creative Writing, and Mini Crosswords. For instance, in Game of 24, while GPT-4 with chain-of-thought prompting only solved 4% of tasks, our method achieved a success rate of 74%. Code repo with all prompts: https://github.com/princeton-nlp/tree-of-thought-llm.

Shunyu Yao, Dian Yu, Jeffrey Zhao, Izhak Shafran, Thomas L. Griffiths, Yuan Cao, Karthik Narasimhan• 2023

Related benchmarks

TaskDatasetResultRank
Commonsense ReasoningHellaSwag
Accuracy63.33
1891
Mathematical ReasoningGSM8K
Accuracy96.2
1362
Code GenerationHumanEval
Pass@180.53
1036
Mathematical ReasoningMATH
Accuracy86.9
882
Language UnderstandingMMLU
Accuracy81.04
825
Mathematical ReasoningGSM8K (test)
Accuracy90
770
Commonsense ReasoningPIQA
Accuracy31.61
751
ReasoningBBH
Accuracy90.8
672
Mathematical ReasoningMATH500 (test)
Accuracy72.4
514
Code GenerationHumanEval (test)
Pass@154.4
506
Showing 10 of 329 rows
...

Other info

Code

Follow for update