Preprint
Machine Learning

MAmmoTH

September 1, 2023

0

Citations

0

Influential Citations

Venue

2023

Year

Abstract

A series of LLMs specifically designed for general math problem-solving, trained on MathInstruct, a dataset compiled from 13 math datasets with intermediate rationales that combines chain-of-thought and program-of-thought approaches to accommodate different thought processes for various math problems.

Analysis

Why This Paper Matters

This paper addresses a critical bottleneck in AI: enabling large language models to perform robust mathematical reasoning across diverse problem types. While prior work often focused on either chain-of-thought (CoT) or program-of-thought (PoT) reasoning in isolation, MAmmoTH recognizes that different math problems benefit from different reasoning styles. By combining both approaches in a single training dataset, the authors aim to create more versatile math-solving models.

The creation of MathInstruct from 13 existing datasets is a significant practical contribution. It provides a standardized, high-quality resource for training and evaluating math reasoning, which could accelerate research in this area. The paper's focus on general math problem-solving, rather than narrow benchmarks, aligns with the growing need for AI systems that can handle real-world, varied mathematical tasks.

Technical Contributions

  • MathInstruct Dataset: Compiled from 13 diverse math datasets, each problem is annotated with both chain-of-thought (natural language step-by-step reasoning) and program-of-thought (code-like step-by-step reasoning) rationales. This dual-annotation strategy is the core innovation.
  • Hybrid Reasoning Training: The LLMs are fine-tuned on MathInstruct, learning to generate either CoT or PoT rationales as appropriate for a given problem, rather than being forced into a single reasoning paradigm.
  • Model Series: The paper produces a series of MAmmoTH models (presumably of varying sizes), allowing practitioners to choose a suitable trade-off between performance and computational cost.

Results

The abstract does not report specific numerical results, such as accuracy on benchmark datasets or comparisons to baselines. The contribution is primarily methodological and dataset-oriented. Future work would likely evaluate MAmmoTH on standard math reasoning benchmarks (e.g., GSM8K, MATH) to quantify improvements.

Significance

MAmmoTH's approach of blending reasoning styles has the potential to influence how future LLMs are trained for structured problem-solving. The MathInstruct dataset could become a standard resource for math reasoning research. By explicitly modeling the diversity of reasoning paths, this work moves beyond one-size-fits-all prompting strategies, which is a step toward more adaptive and capable AI systems.