DataOrchestra: Learning to Orchestrate Per-Example Curation of Pretraining Data

· Source: Computation and Language · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Data Science & Analytics · Depth: Expert, quick

Summary

DataOrchestra is a novel framework designed to optimize Large Language Model (LLM) pretraining data processing by orchestrating example-specific pipelines. Unlike existing methods that apply fixed strategies uniformly, DataOrchestra unifies various processing operations and decides for each data chunk whether to drop, untouch, or clean it. For cleaning, it selects one or more downstream operations, including programmatic editing or LLM-based rewriting, and generates concrete instructions for execution by tool models. Models pretrained from 0.5B to 7B on web data processed by DataOrchestra achieved stable average gains across 11 benchmarks compared to individual data-processing methods. The framework also proved effective for math continued pretraining, outperforming stronger baselines while reducing processing compute by skipping unnecessary operations.

Key takeaway

For ML engineers optimizing LLM pretraining data pipelines, you should consider adopting a dynamic, example-specific curation framework like DataOrchestra. This approach can yield stable performance gains across 11 benchmarks for models from 0.5B to 7B and reduce processing compute compared to uniform strategies, especially for specialized tasks like math pretraining. Evaluate its integration to enhance both model quality and operational efficiency.

Key insights

DataOrchestra dynamically curates pretraining data per-example, unifying operations for improved LLM performance and efficiency.

Principles

Method

DataOrchestra's orchestrator decides to drop, untouch, or clean data chunks. For cleaning, it selects programmatic or LLM-based rewriting operations, generating specific instructions for tool models.

In practice

Topics

Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.