Embedding Self-Correction as an Inherent Ability in Large Language Models for Enhanced Mathematical Reasoning

TL;DR

This paper introduces the Chain of Self-Correction (CoSC), a novel mechanism enabling large language models to iteratively validate and correct their mathematical reasoning, significantly improving accuracy without requiring demonstrations.

Contribution

The paper presents CoSC, a self-correction framework for LLMs that enhances mathematical reasoning by iterative validation and correction, trained with a two-phase fine-tuning approach.

Findings

CoSC improves performance on mathematical datasets.

CoSC-Code-34B achieves 53.5% on MATH dataset.

Operates effectively in zero-shot setting.

Abstract

Accurate mathematical reasoning with Large Language Models (LLMs) is crucial in revolutionizing domains that heavily rely on such reasoning. However, LLMs often encounter difficulties in certain aspects of mathematical reasoning, leading to flawed reasoning and erroneous results. To mitigate these issues, we introduce a novel mechanism, the Chain of Self-Correction (CoSC), specifically designed to embed self-correction as an inherent ability in LLMs, enabling them to validate and rectify their own results. The CoSC mechanism operates through a sequence of self-correction stages. In each stage, the LLMs generate a program to address a given problem, execute this program using program-based tools to obtain an output, subsequently verify this output. Based on the verification, the LLMs either proceed to the next correction stage or finalize the answer. This iterative self-correction process allows the LLMs to refine its reasoning steps and improve the accuracy of its mathematical reasoning. We implement CoSC using a two-phase fine-tuning approach. First, LLMs are trained with a relatively small volume of seeding data generated from GPT-4. Then, we enhance CoSC by training with a larger volume of self-generated data, without relying on GPT-4. Experiments show that CoSC significantly boosts performance on standard mathematical datasets compared to existing open-source LLMs. Notably, our CoSC-Code-34B model achieved a 53.5% score on the challenging MATH dataset, outperforming models like ChatGPT, GPT-4, and multi-modal LLMs such as GPT-4V and Gemini-1.0. Importantly, CoSC operates in a zero-shot manner without requiring demonstrations.