Complexity-Based Prompting for Multi-Step Reasoning

By Yao Fu, Hao-Chun Peng, Ashish Sabharwal, Peter Clark, Tushar Khot. 2022

629 citations recorded by the source corpus.

We study the task of prompting large-scale language models to perform multi-step reasoning. Existing work shows that when prompted with a chain of thoughts (CoT), sequences of short sentences describing intermediate reasoning steps towards a final answer, large language models can generate new reasoning chains and predict answers for new inputs. A central question is which reasoning examples make the most effective prompts. In this work, we propose complexity-based prompting, a simple and effective example selection scheme for multi-step reasoning. We show that prompts with higher reasoning complexity, i.e., chains with more reasoning steps, achieve substantially better performance on multi-step reasoning tasks over strong baselines. We further extend our complexity-based criteria from prompting (selecting inputs) to decoding (selecting outputs), where we sample multiple reasoning chains from the model, then choose the majority of generated answers from complex reasoning chains (over simple chains). When used to prompt GPT-3 and Codex, our approach substantially improves multi-step reasoning accuracy and achieves new state-of-the-art (SOTA) performance on three math benchmarks (GSM8K, MultiArith, and MathQA) and two BigBenchHard tasks (Date Understanding and Penguins), with an average +5.3 and up to +18 accuracy improvements. Compared with existing example selection schemes like manual tuning or retrieval-based selection, selection based on reasoning complexity is intuitive, easy to implement, and annotation-efficient. Further results demonstrate the robustness of performance gains from complex prompts under format perturbation and distribution shift.

Subfields: Artificial Intelligence, Information Systems, Prompting & Reasoning

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Jianfeng Gao
  2. Mohit Bansal
  3. Furu Wei
  4. Percy Liang
  5. Wayne Xin Zhao
  6. Ji-rong Wen
  7. Yejin Choi
  8. Luke S. Zettlemoyer

In Information Systems

  1. Wayne Xin Zhao
  2. Ji-rong Wen
  3. Yupeng Hou
  4. Kun Zhou
  5. Heng Ji
  6. Jian-Yun Nie
  7. Aston Zhang
  8. Junjie Zhang

In Prompting & Reasoning

  1. Jie Huang
  2. Kevin Chen-Chuan Chang
  3. Pan Lu
  4. Furu Wei
  5. Tao Ge
  6. Damai Dai
  7. Xu Sun
  8. Fei Huang

Citations in the corpus