HyperAIHyperAI

Command Palette

Search for a command to run...

MIND: 大規模言語モデルの事前学習のための数学情報を含む合成対話

Syeda Nahida Akter Shrimai Prabhumoye John Kamalu Sanjeev Satheesh Eric Nyberg Mostofa Patwary Mohammad Shoeybi Bryan Catanzaro

Nemotron-MIND: 数学に基づく合成的対話データセット

データセットへ移動

概要

近年の大規模言語モデル(LLM)において、事前学習データの品質を向上させ、下流タスクの精度を改善するための合成データの有用性が広く研究されている。しかしながら、これらの手法は、複雑で多段階の推論や数学的推論を必要とするタスクでは不十分であり、合成データが既存の生コーパスに相補的な知識を追加できないことが多い。本研究では、LLMの数学的推論能力を向上させる、大規模で多様な数学情報を含む合成対話(MIND)生成手法を新たに提案する。具体的には、MINDを用いてOpenWebMath(OWM)に基づく合成会話を生成し、新しい数学コーパスであるMIND-OWMを構築する。様々な会話設定を用いた実験により、対話参加者間の知識ギャップを組み込むことが、高品質な数学データの生成に不可欠であることを明らかにする。さらに、事前学習中に合成データと生データを効果的にフォーマットし統合する方法を特定し、数学的推論の向上を最大化するためには、生データをそのまま使用するのではなく再構成する必要があることを強調する。生データのみで事前学習したモデルと比較して、MIND-OWMで事前学習したモデルは、数学的推論(GSM8K:+13.42%、MATH:+2.30%)において顕著な向上を示し、専門知識(MMLU:+4.55%、MMLU-STEM:+4.28%)や汎用推論タスク(GENERAL REASONING:+2.51%)においても優れた性能を発揮する。MINDを用いて生成した138Bトークンの合成会話をhttps://huggingface.co/datasets/nvidia/Nemotron-MINDで公開する。

One-sentence Summary

Researchers from NVIDIA, Carnegie Mellon University, and Boston University propose MIND, a large-scale method that generates math-informed synthetic dialogues from OpenWeb-Math to enhance LLM mathematical reasoning, demonstrating that incorporating knowledge gaps between dialog participants and restructuring raw data during pretraining maximizes gains, yielding significant improvements on GSM8K (+13.42%+13.42\%+13.42%), MATH (+2.30%+2.30\%+2.30%), MMLU (+4.55%+4.55\%+4.55%), and general reasoning (+2.51%+2.51\%+2.51%), with a released 138B-token138\text{B-token}138B-token corpus.

Key Contributions

  • The paper introduces MIND, a large-scale synthetic dialogue generation method that converts mathematical web content into structured conversations with embedded multi-step reasoning, and shows that encoding knowledge gaps between dialogue participants is a key factor in producing high-quality math training data.
  • This work identifies an effective strategy for formatting and integrating synthetic and raw data during pretraining, demonstrating that restructuring raw data before mixing it with synthetic dialogues yields greater gains in mathematical reasoning than using raw data as-is.
  • Results show that models pretrained on the resulting MIND-OWM corpus achieve significant improvements over raw-data baselines, including +13.42% on GSM8K, +2.30% on MATH, +4.55% on MMLU, +4.28% on MMLU-STEM, and +2.51% on general reasoning benchmarks, while outperforming models trained on 3.6× larger datasets; the paper also releases a 138B-token synthetic conversation corpus.

Introduction

Large language models (LLMs) have shown strong reasoning capabilities, particularly in complex mathematical tasks, but their performance heavily depends on the quality and structure of pretraining corpora. Effective math data must be formatted to encourage step-by-step problem decomposition, yet curating high-quality, structured mathematical datasets is costly and scarce. Prior synthetic data generation methods often lack the multi-step reasoning structure essential for complex math tasks, and many advanced models do not disclose their data strategies, leaving open-source efforts at a disadvantage.

To address these gaps, the authors introduce MIND, a scalable approach that converts raw web text into multi-turn, structured conversations using an off-the-shelf open-source LLM. MIND decomposes content into step-by-step dialogue trajectories, explores each step in depth, and supports diverse conversational styles to enrich data diversity. This method preserves web corpus diversity while embedding dynamic reasoning, enabling continuous synthetic data expansion. The authors demonstrate that pretraining a 7B model on MIND-generated conversations improves mathematical reasoning by up to 6.29% over raw data, outperforms models trained on 3.6× larger raw corpora, and extends gains to general tasks, offering a resource-efficient alternative for structured reasoning data.

Dataset

The authors generate a synthetic mathematical dialogue dataset called MIND (Math Informed Synthetic Dialogue) from unstructured mathematical web text.

Dataset Composition and Sources

  • The seed corpus is OpenWebMath, containing 14.7B tokens of high quality mathematical web text.
  • The authors use LLAMA3-70B-INSTRUCT to convert raw text into multi-turn conversations using a set of prompts (e.g., two students, teacher_student, debate).
  • For each generation, they iteratively take contexts of 500 tokens to preserve details from the original text.

Subset Versions

  • OWM-4B: a smaller subset of OpenWebMath containing 4B tokens for quick evaluation of all seven prompts. Synthetic data from this subset is labeled MIND-OWM-4B.
  • OWM-14B: the full 14.7B token OpenWebMath corpus. Synthetic data from this is called MIND-OWM-14B.

Filtering and Processing

  • Heuristic filtering is used to discard bad generations. LLM-based scoring was tested but rejected because it rated all generations highly, making it unsuitable.
  • No specific cropping strategy is mentioned beyond the 500-token context windows.

Use in Pretraining

  • The base model is pretrained on a blend of 13 CommonCrawl snapshots (73.37%), books/patents (9%), papers (9%), code (5.12%), stack-exchange (2.66%), and Wikipedia (0.8%) for 700B tokens.
  • Continued pretraining uses an additional 50B tokens with a new data blend (D) that maintains a 2:1 ratio of OpenWebMath (either raw R or synthetic S') to 13 snapshots of CommonCrawl (17B tokens, filtered by the FineWebEdu classifier).
  • The CommonCrawl portion is fixed across experiments, while the OpenWebMath component varies between raw, synthetic, or a combination to assess their relative importance.
  • The token distribution is kept constant across experiments, so every model sees the same number of tokens from each source regardless of the source size.

Method

The authors propose MIND, a conversational synthetic data generation framework designed to enhance the mathematical and complex multi-hop reasoning capabilities of Large Language Models (LLMs). Unlike standard rephrasing techniques that only modify surface forms, MIND introduces semantic variations and structured complexity by transforming raw text into multi-turn dialogues. This approach mimics human problem-solving by breaking down complex contexts into step-by-step sub-contexts through conversational interactions.

Refer to the framework diagram. The overall pipeline begins with a raw dataset R\mathcal{R}R and a curated set of conversational prompts P\mathcal{P}P. A pretrained LLM, denoted as M\mathcal{M}M, serves as the conversation generator. The system employs seven distinct conversational settings to guide the generation process, including scenarios such as "Teacher ⇔\Leftrightarrow⇔ Student", "Student ⇔\Leftrightarrow⇔ Student", "Professor ⇔\Leftrightarrow⇔ Professor", "Debater ⇔\Leftrightarrow⇔ Debater", "Interviewer ⇔\Leftrightarrow⇔ Interviewee", "Layman ⇔\Leftrightarrow⇔ Knowall", and "Interactive Problem Solving". These prompts are specifically designed to instruct the LLM to preserve original information while encouraging detailed, step-by-step reasoning and explanations.

For each raw text example rjr_jrj​ from the dataset and a selected prompt pip_ipi​, the generator M\mathcal{M}M produces a synthetic conversation si,js_{i,j}si,j​. This transformation is formalized as:

si,j=M(pi∣∣rj)s_{i,j} = \mathcal{M}(p_i || r_j)si,j​=M(pi​∣∣rj​)

where ∣∣||∣∣ denotes the concatenation of the prompt and the raw text. The total synthetic data generated for a fixed prompt iii is represented as S={si,j∣j∈[1,N]}\mathcal{S} = \{s_{i,j} \mid j \in [1, N]\}S={si,j​∣j∈[1,N]}. The authors utilize LLAMA3-70B-INSTRUCT as the generator M\mathcal{M}M due to its robust performance in dialogue tasks. To ensure context fidelity and prevent detail loss, the generation is performed iteratively on chunks of 500 tokens.

Following generation, a heuristic filtering process H\mathcal{H}H is applied to the synthetic dataset to remove low-quality or incorrect generations. The refined dataset is denoted as S′\mathcal{S}'S′:

S′=H(S)\mathcal{S}' = \mathcal{H}(\mathcal{S})S′=H(S)

This filtering step is crucial for maintaining the high quality of the synthetic dialogue corpus, which is specifically structured to improve reasoning abilities.

The resulting synthetic data S′\mathcal{S}'S′ is utilized for continuous pretraining to evaluate its effectiveness. The authors start with a base LLM C\mathcal{C}C and train it on a data blend D\mathcal{D}D that combines the synthetic data S′\mathcal{S}'S′ with a subset of high-quality pretraining data Rpt\mathcal{R}_{pt}Rpt​, derived from CommonCrawl snapshots. The continuous pretraining process is defined as:

E←pretrain(C,D);where D={S′∪Rpt}\mathcal{E} \leftarrow \text{pretrain}(\mathcal{C}, \mathcal{D}); \text{where } \mathcal{D} = \{\mathcal{S}' \cup \mathcal{R}_{pt}\}E←pretrain(C,D);where D={S′∪Rpt​}

This yields an enhanced model E\mathcal{E}E with improved mathematical reasoning capabilities. The training configuration maintains consistency with the initial pretraining setup, utilizing the AdamW optimizer and specific learning rate schedules to ensure stable convergence.

Experiment

The experiments continuously pretrain a 7B model on synthetic conversational data generated by MIND, evaluating it against raw and rephrased data across math, general reasoning, and specialized knowledge benchmarks. MIND-generated conversations consistently outperform baselines, with the longest conversation format yielding the highest gains across all tasks. Ablations reveal that prompt style matters, as a clear knowledge gap between participants improves explanation depth, while a style like TWO PROFESSORS with no gap underperforms. The benefits extend to high-quality seed corpora like MATHPILE and persist even with a smaller generator model, indicating that conversational structure, rather than model scale, drives the improvements.

Continuous training on MIND-generated conversational data from diverse styles consistently improves performance across all reasoning tasks compared to raw or rephrased data. Selecting the longest conversation for each raw text yields the highest gains, especially in math and specialized knowledge benchmarks. All conversational styles outperform raw and rephrased baselines on GSM8K, MATH, MMLU-STEM, MMLU, and general reasoning. Longest-conversation selection delivers the largest improvements across tasks, highlighting the benefit of multi-perspective dialogues. Debate and teacher-student styles show the strongest math gains, while layman-knowall and two-students excel in specialized knowledge and general reasoning.

Training a 7B LLM on synthetic conversational data generated by MIND improves performance across math, specialized knowledge, and general reasoning benchmarks compared to using raw OpenWebMath or the original pretraining blend. The conversational approach, particularly with the TWO STUDENTS style, yields consistent gains across all evaluated tasks. MIND-generated synthetic conversations outperform raw data and the original pretraining mixture on every benchmark. The TWO STUDENTS conversation style produces the largest improvements, especially on math reasoning tasks. Continuous training with synthetic dialogues enhances both specialized knowledge and general reasoning abilities.

The TWO PROFESSORS conversational style performs comparably or worse than raw data during continuous pretraining, unlike other styles that improve reasoning. This underperformance is attributed to the zero-knowledge gap between expert participants, leading to surface-level discussions without deeper analysis or problem breakdowns. TWO PROFESSORS dialogues yield equivalent or worse performance relative to raw data across benchmarks. The style lacks a knowledge gap between participants, preventing detailed explanations and reasoning expansion. Other styles benefit either from explicit knowledge gaps (e.g., LAYMAN KNOWALL, TEACHER STUDENT) or collaborative non-expert problem solving (e.g., PROBLEM SOLVING, DEBATE).

Synthetic conversations generated from high-quality raw data improve performance on math tasks, with the TWO STUDENTS style showing notable gains over raw data. The effectiveness of conversational styles depends on the presence of knowledge gaps, where clear gaps benefit general reasoning and collaborative problem-solving benefits math tasks. Conversational data from MATHPILE outperforms raw data on math benchmarks, with TWO STUDENTS style yielding further improvements. Styles with explicit or implicit knowledge gaps (e.g., TWO STUDENTS) lead to more detailed reasoning, which is key for math performance.

Synthetic conversational data consistently outperforms raw data across all training configurations. Combining all generated conversation styles from a smaller 4B model surpasses the performance of a larger 14B model trained on a 3.6x larger corpus, highlighting the efficiency and quality of synthetic conversations. While pure synthetic data yields the largest gains on math benchmarks, mixing raw and synthetic data benefits general reasoning tasks. Synthetic conversation data beats raw data in every combination tested. Conversations generated from a 4B model outperform a 14B model with 3.6x more data across all tasks. Pure synthetic data gives the biggest boost on math benchmarks, while blending raw and synthetic data improves general reasoning.

Across multiple benchmarks, training a 7B model on MIND-generated synthetic conversational data consistently outperforms raw data and the original pretraining blend in math, specialized knowledge, and general reasoning. Selecting the longest conversation per raw text yields the largest gains, and styles with knowledge gaps or collaborative problem solving, such as two students and debate, prove most effective, while two professors underperforms due to a lack of depth. Synthetic conversations also demonstrate efficiency, with a 4B teacher beating a larger 14B model trained on more raw data, and blending raw and synthetic data benefits general reasoning even as pure synthetic data maximizes math gains.


AIでAIを構築

アイデアからローンチまで — 無料のAIコーディング支援、すぐに使える環境、最高のGPU価格でAI開発を加速。

AI コーディング補助
すぐに使える GPU
最適な料金体系

HyperAI Newsletters

最新情報を購読する
北京時間 毎週月曜日の午前9時 に、その週の最新情報をメールでお届けします
メール配信サービスは MailChimp によって提供されています