跳到正文
Dongxi 东锡 NLP· @dongxi_nlp · X·· 2026-08-05AI 评分32
AI 导读

DataOrchestra 是 GAIR NLP 推出的开源框架,将预训练数据清洗转为逐样本决策:1.7B 轻量编排器对每个数据块决定 Drop、Untouch 或 Clean,并只为需清洗块选择必要操作。基于执行反馈生成 300K 训练对,在 11 项基准上从 0.5B 到 7B 规模一致优于固定与单独处理策略,优势随模型增大而扩大。约 35% 数据块跳过清洗,节省约 37% 清洗算力。

正文 · 原文

DataOrchestra:

Learning to Orchestrate Per-Example Curation of Pretraining Data

引用Yikun Wang ✈️ ICML 2026@YikunWang001
Happy to introduce DataOrchestra by @Z_Huang_02 and GAIR NLP, and feel thrilled to participate!! Most pretraining pipelines ask: Which processing recipe should we apply to the corpus? DataOrchestra asks a more counterintuitive question: What should happen to this one example? Instead of forcing every sample through the same fixed pipeline, it orchestrates the curation process of every pretraining example. 🎼 DataOrchestra is an open-source framework that turns data curation into a per-example decision problem. Given a raw chunk, a lightweight 1.7B orchestrator decides whether to: raw chunk → Drop → Untouch → Clean For chunks that need cleaning, it selects only the necessary operations—Noise Pruning, Surface Rectification, and Pedagogical Augmentation—and generates a concrete instruction for every selected rewriting stage. A plausible processing plan may still fail when executed by the actual tool models. DataOrchestra therefore learns from execution feedback: teacher plan → tool execution → verification → plan evolution → orchestrator training 🔧 This process produces 300K tool-grounded training pairs. Models pretrained on DataOrchestra-curated data consistently outperform fixed and individual processing strategies across 11 benchmarks, from 0.5B to 7B scale. The advantage grows with model size. Per-example routing also avoids unnecessary work: around 35% of chunks stop at Drop or Untouch, reducing curation compute by approximately 37% compared with always running the full NP → SR → PA pipeline. The takeaway is simple: Do not process pretraining data uniformly. Orchestrate every example precisely. Code, models, prompts, and the complete pipeline ↓
在 X 查看被引用的帖子

来源:Dongxi 东锡 NLP · x.com