DataFlex: A Unified Framework for Data-Centric Dynamic Training of Large Language Models
Fuente:
arXiv
Guardado en:
| Autores principales: | , , , , , , , , , , , , , , , , , , , , , , , , |
|---|---|
| Formato: | Preprint |
| Publicado: |
2026
|
| Materias: | |
| Acceso en línea: | |
| Etiquetas: |
Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
|
| _version_ | 1866912984938315776 |
|---|---|
| author | Liang, Hao Zhao, Zhengyang Qiang, Meiyi Chen, Mingrui Ma, Lu Yu, Rongyi Feng, Hengyi Sun, Shixuan Meng, Zimo Ma, Xiaochen Yang, Xuanlin Cai, Qifeng An, Ruichuan Zeng, Bohan Wong, Zhen Hao Shen, Chengyu He, Runming Han, Zhaoyang Zheng, Yaowei Fu, Fangcheng He, Conghui Cui, Bin Li, Zhiyu E, Weinan Zhang, Wentao |
| author_facet | Liang, Hao Zhao, Zhengyang Qiang, Meiyi Chen, Mingrui Ma, Lu Yu, Rongyi Feng, Hengyi Sun, Shixuan Meng, Zimo Ma, Xiaochen Yang, Xuanlin Cai, Qifeng An, Ruichuan Zeng, Bohan Wong, Zhen Hao Shen, Chengyu He, Runming Han, Zhaoyang Zheng, Yaowei Fu, Fangcheng He, Conghui Cui, Bin Li, Zhiyu E, Weinan Zhang, Wentao |
| contents | Data-centric training has emerged as a promising direction for improving large language models (LLMs) by optimizing not only model parameters but also the selection, composition, and weighting of training data during optimization. However, existing approaches to data selection, data mixture optimization, and data reweighting are often developed in isolated codebases with inconsistent interfaces, hindering reproducibility, fair comparison, and practical integration. In this paper, we present DataFlex, a unified data-centric dynamic training framework built upon LLaMA-Factory. DataFlex supports three major paradigms of dynamic data optimization: sample selection, domain mixture adjustment, and sample reweighting, while remaining fully compatible with the original training workflow. It provides extensible trainer abstractions and modular components, enabling a drop-in replacement for standard LLM training, and unifies key model-dependent operations such as embedding extraction, inference, and gradient computation, with support for large-scale settings including DeepSpeed ZeRO-3. We conduct comprehensive experiments across multiple data-centric methods. Dynamic data selection consistently outperforms static full-data training on MMLU across both Mistral-7B and Llama-3.2-3B. For data mixture, DoReMi and ODM improve both MMLU accuracy and corpus-level perplexity over default proportions when pretraining Qwen2.5-1.5B on SlimPajama at 6B and 30B token scales. DataFlex also achieves consistent runtime improvements over original implementations. These results demonstrate that DataFlex provides an effective, efficient, and reproducible infrastructure for data-centric dynamic training of LLMs. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2603_26164 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | DataFlex: A Unified Framework for Data-Centric Dynamic Training of Large Language Models Liang, Hao Zhao, Zhengyang Qiang, Meiyi Chen, Mingrui Ma, Lu Yu, Rongyi Feng, Hengyi Sun, Shixuan Meng, Zimo Ma, Xiaochen Yang, Xuanlin Cai, Qifeng An, Ruichuan Zeng, Bohan Wong, Zhen Hao Shen, Chengyu He, Runming Han, Zhaoyang Zheng, Yaowei Fu, Fangcheng He, Conghui Cui, Bin Li, Zhiyu E, Weinan Zhang, Wentao Machine Learning Computation and Language Data-centric training has emerged as a promising direction for improving large language models (LLMs) by optimizing not only model parameters but also the selection, composition, and weighting of training data during optimization. However, existing approaches to data selection, data mixture optimization, and data reweighting are often developed in isolated codebases with inconsistent interfaces, hindering reproducibility, fair comparison, and practical integration. In this paper, we present DataFlex, a unified data-centric dynamic training framework built upon LLaMA-Factory. DataFlex supports three major paradigms of dynamic data optimization: sample selection, domain mixture adjustment, and sample reweighting, while remaining fully compatible with the original training workflow. It provides extensible trainer abstractions and modular components, enabling a drop-in replacement for standard LLM training, and unifies key model-dependent operations such as embedding extraction, inference, and gradient computation, with support for large-scale settings including DeepSpeed ZeRO-3. We conduct comprehensive experiments across multiple data-centric methods. Dynamic data selection consistently outperforms static full-data training on MMLU across both Mistral-7B and Llama-3.2-3B. For data mixture, DoReMi and ODM improve both MMLU accuracy and corpus-level perplexity over default proportions when pretraining Qwen2.5-1.5B on SlimPajama at 6B and 30B token scales. DataFlex also achieves consistent runtime improvements over original implementations. These results demonstrate that DataFlex provides an effective, efficient, and reproducible infrastructure for data-centric dynamic training of LLMs. |
| title | DataFlex: A Unified Framework for Data-Centric Dynamic Training of Large Language Models |
| topic | Machine Learning Computation and Language |
| url | https://arxiv.org/abs/2603.26164 |