Reprompting: Automated Chain-of-Thought Prompt Inference Through Gibbs Sampling

  • 2023-05-17 07:35:43
  • Weijia Xu, Andrzej Banburski-Fahey, Nebojsa Jojic
  • 9

Abstract

We introduce Reprompting, an iterative sampling algorithm that searches forthe Chain-of-Thought (CoT) recipes for a given task without human intervention.Through Gibbs sampling, we infer CoT recipes that work consistently well for aset of training samples. Our method iteratively samples new recipes usingpreviously sampled solutions as parent prompts to solve other trainingproblems. On five Big-Bench Hard tasks that require multi-step reasoning,Reprompting achieves consistently better performance than the zero-shot,few-shot, and human-written CoT baselines. Reprompting can also facilitatetransfer of knowledge from a stronger model to a weaker model leading tosubstantially improved performance of the weaker model. Overall, Repromptingbrings up to +17 point improvements over the previous state-of-the-art methodthat uses human-written CoT prompts.

 

Quick Read (beta)

loading the full paper ...