cs.AIApr 12, 2026

Tracing the Roots: A Multi-Agent Framework for Uncovering Data Lineage in Post-Training LLMs

Authors: Yu Li, Xiaoran Shang, Qizhi Pei, Yun Zhu, Xin Gao, Honglin Lin, Zhanping Zhong, Zhuoshi Pan, 

Organizations: Shanghai Artificial Intelligence Laboratory, OpenDataLab · University of Science and Technology of China · Shanghai Jiao Tong University

Abstract

Post-training data plays a pivotal role in shaping the capabilities of Large Language Models (LLMs), yet datasets are often treated as isolated artifacts, overlooking the systemic connections that underlie their evolution. To disentangle these complex relationships, we introduce the concept of data lineage to the LLM ecosystem and propose an automated multi-agent framework to reconstruct the evolutionary graph of dataset development. Through large-scale lineage analysis, we characterize domain-specific structural patterns, such as vertical refinement in math-oriented datasets and horizontal aggregation in general-domain corpora. Moreover, we uncover pervasive systemic issues, including structural redundancy induced by implicit dataset intersections and the propagation of benchmark contamination along lineage paths. To demonstrate the practical value of lineage analysis for data construction, we leverage the reconstructed lineage graph to create a lineage-aware diversity-oriented dataset. By anchoring instruction sampling at upstream root sources, this approach mitigates downstream homogenization and hidden redundancy, yielding a more diverse post-training corpus. We further highlight lineage-centric analysis as an efficient and robust topological alternative to sample-level dataset comparison for large-scale data ecosystems. By grounding data construction in explicit lineage structures, our work advances post-training data curation toward a more systematic and controllable paradigm.

Figures & tables

Explore similar work

CardsList
  1. Towards Next-Generation LLM Training: From the Data-Centric Perspective

    Mar 16, 2026Hao Liang, Zhengyang Zhao, Zhaoyang Han +8LLM TrainingEfficient Language Model Training

  2. Rethinking Data Curation in LLM Training: Online Reweighting Offers Better Generalization than Offline Methods

    Apr 19, 2026Wanru Zhao, Yihong Chen, Yuzhi Tang +6Language Model PretrainingLLM Fine-Tuning

  3. Demystifying Data Organization for Enhanced LLM Training

    May 28, 2026Yalun Dai, Yangyu Huang, Tongshen Yang +8Language Model PretrainingTraining Data Curation