Get an A in Math: Progressive Rectification Prompting

Authors

  • Zhenyu Wu Xi'an Jiaotong University
  • Meng Jiang University of Notre Dame
  • Chao Shen Xi'an Jiaotong University

DOI:

https://doi.org/10.1609/aaai.v38i17.29898

Keywords:

NLP: (Large) Language Models, NLP: Question Answering

Abstract

Chain-of-Thought (CoT) prompting methods have enabled large language models (LLMs) to generate reasoning paths and solve math word problems (MWPs). However, they are sensitive to mistakes in the paths, as any mistake can result in an incorrect answer. We propose a novel method named Progressive Rectification Prompting (PRP) to improve average accuracy on eight MWP datasets from 77.3 to 90.5. Given an initial answer from CoT, PRP iterates a verify-then-rectify process to progressively identify incorrect answers and rectify the reasoning paths. With the most likely correct answer, the LLM predicts a masked numerical value in the question; if the prediction does not match the masked value, the answer is likely incorrect. Then the LLM is prompted to re-generate the reasoning path hinted with a set of incorrect answers to prevent itself from repeating previous mistakes. PRP achieves the best performance compared against the CoT methods. Our implementation is made publicly available at https://wzy6642.github.io/prp.github.io/.

Published

2024-03-24

How to Cite

Wu, Z., Jiang, M., & Shen, C. (2024). Get an A in Math: Progressive Rectification Prompting. Proceedings of the AAAI Conference on Artificial Intelligence, 38(17), 19288-19296. https://doi.org/10.1609/aaai.v38i17.29898

Issue

Section

AAAI Technical Track on Natural Language Processing II