Skip to main navigation Skip to search Skip to main content

Exploring Equation as a Better Intermediate Meaning Representation for Numerical Reasoning of Large Language Models

  • Dingzirui Wang
  • , Longxu Dou
  • , Wenbin Zhang
  • , Junyu Zeng
  • , Wanxiang Che*
  • *Corresponding author for this work
  • Harbin Institute of Technology
  • Ltd.

Research output: Contribution to journalConference articlepeer-review

Abstract

Numerical reasoning is a vital capability for natural language processing models to understand and process numerical information in real-world scenarios. Most current methods first generate the Intermediate Meaning Representations (IMRs) of questions and then generate answers. Current SOTA methods generate programs as IMRs with large language models (LLMs). Intuitively, equations have fewer restrictions and closer semantics to the question than programs, leading to higher generation accuracy. However, current LLMs generate equations worse than programs, where we assume that the equation data is rare in pre-training data compared to programs. So in this paper, we try to use equations as IMRs to solve the numerical reasoning task by addressing two problems: (1) Theoretically, how to prove that the equation is an IMR with higher generation accuracy than programs; (2) Empirically, how to improve the generation accuracy of equations with LLMs. For the first problem, we propose and prove a proposition to theoretically compare the generation accuracy of different IMRs. For the second problem, we present a method called Boosting Numerical Reasoning by Decomposing the Generation of Equations (BRIDGE), which can improve the accuracy of LLMs in generating equations as IMRs by reducing the tendency of generating constant expressions and programs. Our method improves the performance by 2.2%, 0.9%, and 1.7% on GSM8K, SVAMP, and Algebra datasets compared to the previous state-of-the-art methods under the single reasoning path setting. Our code and prompts are available at https://github.com/ziruiHIT/Bridge for Numerical Reasoning.

Original languageEnglish
Pages (from-to)19116-19125
Number of pages10
JournalProceedings of the AAAI Conference on Artificial Intelligence
Volume38
Issue number17
DOIs
StatePublished - 25 Mar 2024
Event38th AAAI Conference on Artificial Intelligence, AAAI 2024 - Vancouver, Canada
Duration: 20 Feb 202427 Feb 2024

Fingerprint

Dive into the research topics of 'Exploring Equation as a Better Intermediate Meaning Representation for Numerical Reasoning of Large Language Models'. Together they form a unique fingerprint.

Cite this