Key-Point-Driven Data Synthesis with its Enhancement on Mathematical Reasoning

4 Mar 2024  ·  Yiming Huang, Xiao Liu, Yeyun Gong, Zhibin Gou, Yelong Shen, Nan Duan, Weizhu Chen ·

Large language models (LLMs) have shown great potential in complex reasoning tasks, yet their performance is often hampered by the scarcity of high-quality and reasoning-focused training datasets. Addressing this challenge, we propose Key-Point-Driven Data Synthesis (KPDDS), a novel data synthesis framework that synthesizes question-answer pairs by leveraging key points and exemplar practices from authentic data sources. KPDDS ensures the generation of novel questions with rigorous quality control and substantial scalability. As a result, we present KPMath, an extensive synthetic dataset tailored for mathematical reasoning, comprising over 800K question-answer pairs. Utilizing KPMath and augmenting it with additional reasoning-intensive corpora, we create the comprehensive KPMath-Plus dataset. The Qwen1.5-72B model, fine-tuned on KPMath-Plus, achieves 87.0% PASS@1 accuracy on GSM8K and 58.3% on MATH, surpassing competitors in the 7B to 70B range and best commercial models like GPT-4 across multiple math reasoning datasets.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Uses Extra
Training Data
Result Benchmark
Math Word Problem Solving MATH Llemma-34B-KPMath-Plus Accuracy 48.6 # 32
Parameters (Billions) 34 # 27
Math Word Problem Solving MATH Llama2-13B-KPMath-Plus Accuracy 41 # 53
Parameters (Billions) 13 # 39
Math Word Problem Solving MATH DeepSeekMath-7B-KPMath-Plus Accuracy 48.8 # 30
Parameters (Billions) 7 # 58
Math Word Problem Solving MATH Mistral-7B-KPMath-Plus Accuracy 46.8 # 37
Parameters (Billions) 7 # 58

Methods