Skip to main content

Modern Data Centric AI system for Large Language Models

Project description

DataFlow

Documents

🎉 If you like our project, please give us a star ⭐ on GitHub for the latest update.

简体中文 | English

🚀 Features⚡ Quick Start📖 Documentation🧪 Experiments

https://github.com/user-attachments/assets/05e047a5-99bb-4043-bc71-2b5ccdab2126

📰 1. News

🎉 [2025-06-28] We’re excited to announce that DataFlow, our Data-centric AI system, is now released! Stay tuned for future updates.

🔍 2. Overview

DataFlow is a data preparation and training system designed to parse, generate, process and evaluate high-quality data from noisy sources (PDF, plain-text, low-quality QA), thereby improving the performance of large language models (LLMs) in specific domains through targeted training (Pre-training, Supervised Fine-tuing, RL training) or RAG using knowledge base cleaning. DataFlow has been empirically validated to improve domain-oriented LLM's performance in fields such as healthcare, finance, and law.

Specifically, we constructing diverse operators leveraging rule-based methods, deep learning models, LLMs, and LLM APIs. These operators are systematically integrated into distinct pipelines, collectively forming the comprehensive DataFlow system. Additionally, we develop an intelligent DataFlow-agent capable of dynamically assembling new pipelines by recombining existing operators on demand.

🛠️ 3. Pipelines Functionality

🔧 3.1 Ready-to-Use PipeLines

Current Pipelines in Dataflow are as follows:

  • 📝 Text Pipeline: Mine question-answer pairs from large-scale plain-text data (mostly crawed from InterNet) for use in SFT and RL training.
  • 🧠 Reasoning Pipeline: Enhances existing question–answer pairs with (1) extended chain-of-thought, (2) category classification, and (3) difficulty estimation.
  • 🗃️ Text2SQL Pipeline: Translates natural language questions into SQL queries, supplemented with explanations, chain-of-thought reasoning, and contextual schema information.
  • 📚 Knowlege Base Cleaning Pipeline: Extract and structure knowledge from unorganized sources like tables, PDFs, and Word documents into usable entries for downstream RAG or QA pair generation.
  • 🤖 Agentic RAG Pipeline: Identify and extract QA pairs from existing QA datasets or knowledge bases that require external knowledge to answer, for use in downstream training of Agnetic RAG tasks.

⚙️ 3.2 Flexible Operator PipeLines

In this framework, operators are categorized into Fundamental Operators, Generic Operators, Domain-Specific Operators, and Evaluation Operators, etc., supporting data processing and evaluation functionalities. Please refer to the documentation for details.

🤖 3.3 Agent Guided Pipelines

⚡ 4. Quick Start

For environment setup and installation, please using the following commands👇

conda create -n dataflow python=3.10 
conda activate dataflow

pip install open-dataflow

If you want to use your own GPU to inference locally, please use:

pip install open-dataflow[vllm]

Dataflow supports Python>=3.10

You can use follwing command to check if installed correctly:

dataflow -v

You are expected to see following outputs:

open-dataflow codebase version: 1.0.0
        Checking for updates...
        Local version:  1.0.0
        PyPI newest version:  1.0.0
You are using the latest version: 1.0.0.

For Quick-Start and Guide, please visit our Documentation.

Documents

🧪 5. Experimental Results

For Detailed Experiments setting, please visit our documentation.

📝 5.1 Text PipeLine

5.1.1 Pre-training data filter pipeline

The pre-training data processing pipeline was applied to randomly sampled data from the RedPajama dataset, resulting in a final data retention rate of 13.65%. The analysis results using QuratingScorer are shown in the figure. As can be seen, the filtered pretraining data significantly outperforms the original data across four scoring dimensions: writing style, requirement for expert knowledge, factual content, and educational value. This demonstrates the effectiveness of the DataFlow pretraining data processing.

5.1.2 SFT data filter pipeline

We filted 3k record from alpaca dataset and compare it with radom selected 3k data from alpaca dataset by training it on Qwen2.5-7B. Results are:

🧠 5.2 Reasoning Pipeline

We verify our reasoning pipeline by SFT on a Qwen2.5-32B-Instruct with Reasoning Pipeline synsthized data. We generated 1k and 5k SFT data pairs. Results are:

🗃️ 5.3 Text2SQL PipeLine

We fine-tuned the Qwen2.5-Coder-14B model on the Bird dataset using both Supervised Fine-tuning (SFT) and Reinforcement Learning (RL), with data constructed via the DataFlow-Text2SQL Pipeline. Results are:

💐 6. Acknowledgements

We sincerely appreciate MinerU's outstanding contribution, particularly its robust text extraction capabilities from PDFs and documents, which greatly facilitates data loading.

🤝 7. Community & Support

Join the DataFlow open-source community to ask questions, share ideas, and collaborate with other developers!

• 📮 GitHub Issues: Report bugs or suggest features

• 🔧 GitHub Pull Requests: Contribute code improvements

• 💬 Join our community groups to connect with us and other contributors!

📜 8. Citation

If you use DataFlow in your research, feel free to give us a cite.

@misc{dataflow2025,
  author       = {DataFlow Develop Team},
  title        = {DataFlow: A Unified Framework for Data-Centric AI},
  year         = {2025},
  howpublished = {\url{https://github.com/OpenDCAI/DataFlow}},
  note         = {Accessed: 2025-07-08}
}

📊 9. Statistics


Connect with the PKU-DCAI Research Team on Xiaohongshu: 26133106768

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

open_dataflow_adp-1.1.1.tar.gz (1.5 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

open_dataflow_adp-1.1.1-py3-none-any.whl (1.7 MB view details)

Uploaded Python 3

File details

Details for the file open_dataflow_adp-1.1.1.tar.gz.

File metadata

  • Download URL: open_dataflow_adp-1.1.1.tar.gz
  • Upload date:
  • Size: 1.5 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.13.4

File hashes

Hashes for open_dataflow_adp-1.1.1.tar.gz
Algorithm Hash digest
SHA256 3222222a0791b6fad05a550242a10fe5f0d3f6a3561e32c6c21b5a5cc1b52a5b
MD5 19db1532567b2753b10175468863af35
BLAKE2b-256 5fe3b2e650a040c0e1da48db39682cc4e39983f3696f7eb87220d5a0bcc15237

See more details on using hashes here.

File details

Details for the file open_dataflow_adp-1.1.1-py3-none-any.whl.

File metadata

File hashes

Hashes for open_dataflow_adp-1.1.1-py3-none-any.whl
Algorithm Hash digest
SHA256 786ad93e481dbb460eeb3686747e5463c5bfeb4f9750927528141d9494bbe1a8
MD5 fadafee647dbbb0366cc51e47f2ec34d
BLAKE2b-256 8f747b59c95f434b4081b645d7e2071fca8cd5a25cd479f96321341b511c9fee

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page