DataFlow: An LLM-Driven Framework for Unified Data Preparation and Workflow Automation in the Era of Data-Centric AI
Authors: Hao Liang, Xiaochen Ma, Zhou Liu, Zhen Hao Wong, Zhengyang Zhao, Zimo Meng, Runming He, Chengyu Shen, Qifeng Cai, Zhaoyang Han, Meiyi Qiang, Yalin Feng, Tianyi Bai, Zewei Pan, Ziyi Guo, Yizhen Jiang, Jingwen Deng, Qijie You, Peichao Lai, Tianyu Guo, Chi Hsu Tsai, Hengyi Feng, Rui Hu, Wenkai Yu, Junbo Niu, Bohan Zeng, Ruichuan An, Lu Ma, Jihao Huang, Yaowei Zheng, Conghui He, Linpeng Tang, Bin Cui, Weinan E, Wentao Zhang
AI-generated Key Points
- The DataFlow Technical Report presents a comprehensive and innovative framework for data preparation in Large Language Models (LLMs)
- DataFlow addresses the growing demand for high-quality data in LLMs and challenges posed by current ad-hoc practices
- DataFlow provides a unified and extensible solution with system-level abstractions enabling modular, reusable, and composable data transformations
- Designed with a PyTorch-like programming interface for usability, extensibility, and maintainability
- Offers nearly 200 reusable operators and six domain-general pipelines spanning various domains such as text, mathematical reasoning, code, Text-to-SQL, agentic RAG, and large-scale knowledge extraction
- DataFlow-Agent automates translation of natural-language specifications into executable pipelines through operator synthesis and iterative verification
- Extensive experiments demonstrate effectiveness of the framework across diverse data preparation scenarios
- Resulting datasets consistently match or surpass state-of-the-art baselines in quality and performance
- Mathematical reasoning data outperforms high-quality synthetic baselines on various benchmarks
- Pipelines achieve significant improvements in execution accuracy compared to existing corpora while using fewer training examples
- Training on only 10K samples can yield substantial gains in data efficiency compared to models trained on larger datasets when combining text and code data into a unified corpus with DataFlow.
Abstract: The rapidly growing demand for high-quality data in Large Language Models (LLMs) has intensified the need for scalable, reliable, and semantically rich data preparation pipelines. However, current practices remain dominated by ad-hoc scripts and loosely specified workflows, which lack principled abstractions, hinder reproducibility, and offer limited support for model-in-the-loop data generation. To address these challenges, we present DataFlow, a unified and extensible LLM-driven data preparation framework. DataFlow is designed with system-level abstractions that enable modular, reusable, and composable data transformations, and provides a PyTorch-style pipeline construction API for building debuggable and optimizable dataflows. The framework consists of nearly 200 reusable operators and six domain-general pipelines spanning text, mathematical reasoning, code, Text-to-SQL, agentic RAG, and large-scale knowledge extraction. To further improve usability, we introduce DataFlow-Agent, which automatically translates natural-language specifications into executable pipelines via operator synthesis, pipeline planning, and iterative verification. Across six representative use cases, DataFlow consistently improves downstream LLM performance. Our math, code, and text pipelines outperform curated human datasets and specialized synthetic baselines, achieving up to +3\% execution accuracy in Text-to-SQL over SynSQL, +7\% average improvements on code benchmarks, and 1--3 point gains on MATH, GSM8K, and AIME. Moreover, a unified 10K-sample dataset produced by DataFlow enables base models to surpass counterparts trained on 1M Infinity-Instruct data. These results demonstrate that DataFlow provides a practical and high-performance substrate for reliable, reproducible, and scalable LLM data preparation, and establishes a system-level foundation for future data-centric AI development.
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
By clicking on the button above, our algorithm compares the title and abstract of this paper with all the papers of the same arXiv category that we have summarized, and finds the closest ones. You can rerun it from time to time: the results improve as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.