DataFlow: An LLM-Driven Framework for Unified Data Preparation and Workflow Automation in the Era of Data-Centric AI

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Liang, Hao, Ma, Xiaochen, Liu, Zhou, Wong, Zhen Hao, Zhao, Zhengyang, Meng, Zimo, He, Runming, Shen, Chengyu, Cai, Qifeng, Han, Zhaoyang, Qiang, Meiyi, Feng, Yalin, Bai, Tianyi, Pan, Zewei, Guo, Ziyi, Jiang, Yizhen, Deng, Jingwen, You, Qijie, Lai, Peichao, Guo, Tianyu, Tsai, Chi Hsu, Feng, Hengyi, Hu, Rui, Yu, Wenkai, Niu, Junbo, Zeng, Bohan, An, Ruichuan, Ma, Lu, Huang, Jihao, Zheng, Yaowei, He, Conghui, Tang, Linpeng, Cui, Bin, E, Weinan, Zhang, Wentao
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909969004101632
author Liang, Hao
Ma, Xiaochen
Liu, Zhou
Wong, Zhen Hao
Zhao, Zhengyang
Meng, Zimo
He, Runming
Shen, Chengyu
Cai, Qifeng
Han, Zhaoyang
Qiang, Meiyi
Feng, Yalin
Bai, Tianyi
Pan, Zewei
Guo, Ziyi
Jiang, Yizhen
Deng, Jingwen
You, Qijie
Lai, Peichao
Guo, Tianyu
Tsai, Chi Hsu
Feng, Hengyi
Hu, Rui
Yu, Wenkai
Niu, Junbo
Zeng, Bohan
An, Ruichuan
Ma, Lu
Huang, Jihao
Zheng, Yaowei
He, Conghui
Tang, Linpeng
Cui, Bin
E, Weinan
Zhang, Wentao
author_facet Liang, Hao
Ma, Xiaochen
Liu, Zhou
Wong, Zhen Hao
Zhao, Zhengyang
Meng, Zimo
He, Runming
Shen, Chengyu
Cai, Qifeng
Han, Zhaoyang
Qiang, Meiyi
Feng, Yalin
Bai, Tianyi
Pan, Zewei
Guo, Ziyi
Jiang, Yizhen
Deng, Jingwen
You, Qijie
Lai, Peichao
Guo, Tianyu
Tsai, Chi Hsu
Feng, Hengyi
Hu, Rui
Yu, Wenkai
Niu, Junbo
Zeng, Bohan
An, Ruichuan
Ma, Lu
Huang, Jihao
Zheng, Yaowei
He, Conghui
Tang, Linpeng
Cui, Bin
E, Weinan
Zhang, Wentao
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2512_16676
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle DataFlow: An LLM-Driven Framework for Unified Data Preparation and Workflow Automation in the Era of Data-Centric AI
Liang, Hao
Ma, Xiaochen
Liu, Zhou
Wong, Zhen Hao
Zhao, Zhengyang
Meng, Zimo
He, Runming
Shen, Chengyu
Cai, Qifeng
Han, Zhaoyang
Qiang, Meiyi
Feng, Yalin
Bai, Tianyi
Pan, Zewei
Guo, Ziyi
Jiang, Yizhen
Deng, Jingwen
You, Qijie
Lai, Peichao
Guo, Tianyu
Tsai, Chi Hsu
Feng, Hengyi
Hu, Rui
Yu, Wenkai
Niu, Junbo
Zeng, Bohan
An, Ruichuan
Ma, Lu
Huang, Jihao
Zheng, Yaowei
He, Conghui
Tang, Linpeng
Cui, Bin
E, Weinan
Zhang, Wentao
Machine Learning
Computation and Language
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.
title DataFlow: An LLM-Driven Framework for Unified Data Preparation and Workflow Automation in the Era of Data-Centric AI
topic Machine Learning
Computation and Language
url https://arxiv.org/abs/2512.16676