ControlMath: Controllable Data Generation Promotes Math Generalist Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chen, Nuo, Wu, Ning, Chang, Jianhui, Li, Jia
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914955729567744
author Chen, Nuo
Wu, Ning
Chang, Jianhui
Li, Jia
author_facet Chen, Nuo
Wu, Ning
Chang, Jianhui
Li, Jia
contents Utilizing large language models (LLMs) for data augmentation has yielded encouraging results in mathematical reasoning. However, these approaches face constraints in problem diversity, potentially restricting them to in-domain/distribution data generation. To this end, we propose ControlMath, an iterative method involving an equation-generator module and two LLM-based agents. The module creates diverse equations, which the Problem-Crafter agent then transforms into math word problems. The Reverse-Agent filters and selects high-quality data, adhering to the "less is more" principle, achieving better results with fewer data points. This approach enables the generation of diverse math problems, not limited to specific domains or distributions. As a result, we collect ControlMathQA, which involves 190k math word problems. Extensive results prove that combining our dataset with in-domain datasets like GSM8K can help improve the model's mathematical ability to generalize, leading to improved performances both within and beyond specific domains.
format Preprint
id arxiv_https___arxiv_org_abs_2409_15376
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle ControlMath: Controllable Data Generation Promotes Math Generalist Models
Chen, Nuo
Wu, Ning
Chang, Jianhui
Li, Jia
Machine Learning
Artificial Intelligence
Computation and Language
Utilizing large language models (LLMs) for data augmentation has yielded encouraging results in mathematical reasoning. However, these approaches face constraints in problem diversity, potentially restricting them to in-domain/distribution data generation. To this end, we propose ControlMath, an iterative method involving an equation-generator module and two LLM-based agents. The module creates diverse equations, which the Problem-Crafter agent then transforms into math word problems. The Reverse-Agent filters and selects high-quality data, adhering to the "less is more" principle, achieving better results with fewer data points. This approach enables the generation of diverse math problems, not limited to specific domains or distributions. As a result, we collect ControlMathQA, which involves 190k math word problems. Extensive results prove that combining our dataset with in-domain datasets like GSM8K can help improve the model's mathematical ability to generalize, leading to improved performances both within and beyond specific domains.
title ControlMath: Controllable Data Generation Promotes Math Generalist Models
topic Machine Learning
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2409.15376