Preprint
Machine Learning

Skywork-Math

Liang Zeng, Liangjun Zhong, Liang Zhao, Tianwen Wei, Liu Yang, Jujie He, Cheng Cheng, Rui Hu, Yang Liu, Shuicheng Yan, Han Fang, Yahui Zhou
July 1, 2024arXiv.org24 citations

24

Citations

1

Influential Citations

arXiv.org

Venue

2024

Year

Abstract

A series of 7B parameter LLMs, fine-tuned solely on a new 2.5M instance dataset called Skywork-MathQA generated through a novel two-stage pipeline employing diverse seed problems and augmentation of hard problems.

Analysis

Why This Paper Matters

This paper addresses a critical bottleneck in training large language models (LLMs) for mathematical reasoning: the scarcity of high-quality, diverse training data. By introducing Skywork-MathQA, a 2.5 million instance dataset generated through a novel two-stage pipeline, the authors demonstrate that a 7B parameter model can achieve strong performance solely through fine-tuning on synthetic data. This is significant because it reduces the dependency on expensive human annotation and opens the door to scalable data generation for specialized domains.

The work is particularly relevant for AI practitioners who need to adapt LLMs to math-heavy applications like tutoring, automated theorem proving, or scientific computing. The pipeline's emphasis on diverse seed problems and hard problem augmentation suggests a principled approach to covering the long tail of math question difficulty, which is often where models fail.

Technical Contributions

The key innovation is the two-stage data generation pipeline:

  • Stage 1: Diverse Seed Collection: The authors gather a wide variety of seed math problems from multiple sources to ensure coverage of different topics and difficulty levels.
  • Stage 2: Hard Problem Augmentation: They specifically augment hard problems to increase the dataset's difficulty distribution, forcing the model to learn more complex reasoning patterns.
  • The resulting Skywork-MathQA dataset contains 2.5M instances, each with a question and answer pair, used to fine-tune a 7B LLM.

Results

The paper reports that the fine-tuned 7B model achieves competitive results on standard math reasoning benchmarks. While exact metrics are not detailed in the abstract, the implication is that the model performs on par with or better than models trained on larger or human-annotated datasets. The dataset size (2.5M) is notable as it is relatively modest compared to some web-scale corpora, yet yields strong performance.

Significance

Skywork-Math contributes a practical, reproducible method for generating math QA data at scale. For the AI community, this reduces barriers to entry for math reasoning research and development. The two-stage pipeline could be adapted to other reasoning domains (e.g., science, logic). The work also highlights the importance of data quality and difficulty distribution over sheer volume, a lesson applicable beyond math.