paper-with-me

홈 › Papers

OpenMathInstruct-2: Accelerating AI for Math with Massive Open-Source Instruction Data

2024-10-02 · Shubham Toshniwal, Wei Du, Ivan Moshkov, Branislav Kisacanin, Alexan Ayrapetyan, Igor Gitman

Mathematical reasoning continues to be a critical challenge in large language model (LLM) development with significant interest. However, most of the cutting-edge progress in mathematical reasoning with LLMs has become \emph{closed-source} due to lack of access to training data. This lack of data access limits researchers from understanding the impact of different choices for synthesizing and utilizing the data. With the goal of creating a high-quality finetuning (SFT) dataset for math reasoning, we conduct careful ablation experiments on data synthesis using the recently released \texttt{Llama3.1} family of models. Our experiments show that: (a) solution format matters, with excessively verbose solutions proving detrimental to SFT performance, (b) data generated by a strong teacher outperforms equally-sized data generated by a weak student model, (c) SFT is robust to low-quality solutions, allowing for imprecise data filtering, and (d) question diversity is crucial for achieving data scaling gains. Based on these insights, we create the OpenMathInstruct-2 dataset, which consists of 14M question-solution pairs ($\approx$ 600K unique questions), making it nearly eight times larger than the previous largest open-source math reasoning dataset. Finetuning the \texttt{Llama-3.1-8B-Base} using OpenMathInstruct-2 outperforms \texttt{Llama3.1-8B-Instruct} on MATH by an absolute 15.9\% (51.9\% $\rightarrow$ 67.8\%). Finally, to accelerate the open-source efforts, we release the code, the finetuned models, and the OpenMathInstruct-2 dataset under a commercially permissive license.

📄 PDF Abstract BibTeX arXiv:2410.01560

Code (1)

NVIDIA/NeMo-Skills

Tasks

Arithmetic ReasoningLarge Language ModelMathMathematical ReasoningMath Word Problem Solving

Methods 이 논문이 사용한 방법론

SFT Shrink and Fine-Tune, or SFT, is a type of distillation that avoids explicit distillation by copying parameters to a student student model and then fine-tuning.…

Similar Papers 제목 키워드 기반

OpenMathInstruct-1: A 1.8 Million Math Instruction Tuning Dataset

2024-02-15 · Shubham Toshniwal, Ivan Moshkov, Sean Narenthiran, Daria Gitman 외

Recent work has shown the immense potential of synthetically generated datasets for training large language models (LLMs), especially for acquiring targeted skills. Current large-scale math instruction tuning datasets su…

Arithmetic ReasoningGSM8KMathMath Word Problem Solving

HDPO: Hybrid Distillation Policy Optimization via Privileged Self-Distillation

2026-03-25 · Ken Ding arxiv

Large language models trained with reinforcement learning (RL) for mathematical reasoning face a fundamental challenge: on problems the model cannot solve at all - "cliff" prompts - the RL gradient vanishes entirely, pre…

Reinforcement LearningMathematical Reasoning

Calibrating Overconfidence Without Sacrificing Confidence: Probe-Conditioned Head Intervention for LLMs

2026-06-02 · Ke Li, Chongzhe Zhang, Zifan Zeng, Feng Liu 외 arxiv

Large language models often express high confidence in answers that are wrong. Standard calibration remedies typically act globally or at the score level, reducing unwarranted confidence but also risking erosion of warra…

Data Engineering for Everyone

2021-02-23 · Vijay Janapa Reddi, Greg Diamos, Pete Warden, Peter Mattson 외

Data engineering is one of the fastest-growing fields within machine learning (ML). As ML becomes more common, the appetite for data grows more ravenous. But ML requires more data than individual teams of data engineers …

BIG-bench Machine Learning

On Accelerating Edge AI: Optimizing Resource-Constrained Environments

2025-01-25 · Jacob Sander, Achraf Cohen, Venkat R. Dasari, Brent Venable 외

Resource-constrained edge deployments demand AI solutions that balance high performance with stringent compute, memory, and energy limitations. In this survey, we present a comprehensive overview of the primary strategie…

Knowledge DistillationModel CompressionNeural Architecture SearchQuantization+2