Research map / Large language models / Agents, code and math
DataGen: Unified Synthetic Dataset Generation via Large Language Models
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Proposes DataGen, an LLM-powered framework that generates controllable, diverse, and truthful text datasets using attribute-guided generation, group checking, code-based label verification, and RAG-based factual validation, and applies it to LLM benchmarking and data augmentation.
Problem
Existing LLM-driven dataset generation struggles with generalization, controllability, diversity, and truthfulness (hallucinated or incorrect labels), and often only perturbs existing items or is limited to specific formats.
Approach
A unified pipeline that takes a base dataset, a natural-language description, and optional user constraints, then combines few-shot clustering-based example selection, attribute-guided generation, embedding-based group checking, self-reflection/enhancement, code-verified math labels, and Wikipedia RAG validation, plus difficulty enhancement as post-processing.
Key results
- Code-based math verification raised label correctness on generated GSM8K from 44% to 88% (reported as 92% in text), and RAG validation corrected 4.2% of examples.
- Attribute-guided generation and group checking raised remote-clique diversity from 0.695 to 0.743 on GPT-4-generated data.
- Data augmentation with 200 DataGen samples improved performance on all ten evaluated datasets, and the genset alpaca-tuned Llama3 model scored 6.403 on MT-Bench versus 5.259 for the original-alpaca model.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Fine-tune on 1,000 DataGen-generated rather than existing Alpaca samples | Llama3-7B · MT-Bench | Average score: 5.259 -> 6.403 | Helped |
| Add attribute-guided generation (ablation) | DataGen generated data · Generated data | Remote-clique: 0.695 -> 0.735 | Helped |
| Add group checking after attribute-guided generation (ablation) | DataGen generated data · Generated data | Remote-clique: 0.735 -> 0.743 | Helped |
| Increase generation temperature from 0 to 1 (ablation) | GPT-4 data generation · TruthfulQA examples | Remote-clique: 0.683 -> 0.721 | Helped |
| Apply code-based mathematical evaluation (ablation) | DataGen math data · Generated math data | Accuracy: 44% -> 88% | Helped |
| Apply RAG-based factual validation (ablation) | DataGen generated data · Generated data | Corrected percentage: 4.2% corrected | Helped |
| Apply difficulty enhancement (ablation) | DataGen-generated benchmark data · GSM8K | accuracy: 0.665 -> 0.585 | Hurt |
| Use DataGen-generated instead of original benchmark data | ChatGPT · GSM8K | accuracy: 0.762 -> 0.665 | Hurt |
| Use DataGen-generated instead of original benchmark data | ChatGPT · MMLU | accuracy: 0.609 -> 0.798 | Helped |
| Use DataGen-generated instead of original benchmark data | ChatGPT · TruthfulQA | accuracy: 0.825 -> 0.837 | Helped |
| Use DataGen-generated instead of original benchmark data | ChatGPT · HellaSwag | accuracy: 0.611 -> 0.960 | Helped |
Surprising findings
- Most evaluated LLMs score lower on GPT-4-generated GSM8K than on the original GSM8K, suggesting overstated reasoning ability.
- Llama2-7b's TruthfulQA performance declined after fine-tuning on generated data.
Limitations
- Error analysis shows factuality errors (e.g., 41% on GSM8K, 79% on TruthfulQA) and question-quality errors remain in generated data.
- Limited gains on knowledge-intensive datasets such as MMLU and TruthfulQA after augmentation.
- Generation cost up to about $0.20 per item with RAG and code validation, and self-alignment or weak-to-strong settings were not studied.
Experimental setup
Datasets: GSM8K MMLU TruthfulQA HellaSwag MetaTool MultiNLI ARC-Challenge BoolQ BBH alpaca
Benchmarks: GSM8K MMLU TruthfulQA HellaSwag MetaTool ARC-Challenge BoolQ BBH MT-Bench
Metrics: accuracy remote-clique diversity self-BLEU entity overlap rate error proportion
Baselines: DyVal 2 DyVal S3Eval MetaMath Qameleon
Builds on: DyVal 2 MetaMath LLaMA-Factory
Closest papers by meaning
- Towards a Theoretical Understanding of Synthetic Data in LLM Post-Training: A Reverse-Bottleneck Perspective (ICLR 2025 · similarity 0.88)
- Controlled Text Generation via Language Model Arithmetic (ICLR 2024 · similarity 0.87)
- It's All Training: A Fully Synthetic Single-Stage Recipe for LLMs (NeurIPS 2026 · similarity 0.87)
- Automating Dataset Updates Towards Reliable and Timely Evaluation of Large Language Models (NeurIPS 2024 · similarity 0.87)
- ORACLE: Optimizing Reasoning Abilities of Large Language Models via Constraint-Led Synthetic Data Elicitation (AAAI 2026 · similarity 0.87)
- Not All LLM-Generated Data Are Equal: Rethinking Data Weighting in Text Classification (ICLR 2025 · similarity 0.86)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
