mlsanityData to 14 Oct 2026

LLMSurgeon: Diagnosing Data Mixture of Large Language Models

arXiv 2605.30348 ↗First posted 28 May 2026cs.CL cs.AI cs.LGcited by 1

Not on the map. Why?

Abstract (arXiv)

The pretraining data mixture of Large Language Models (LLMs) constitutes their "digital DNA", shaping model behaviors, capabilities, and failure modes. Yet this composition is rarely disclosed, making post-hoc auditing of data combination or provenance difficult. In this work, we formalize Data Mixture Surgery (DMS): given only generated text from a target LLM, estimate the domain-level distribution of its pretraining corpus under a predefined taxonomy. We propose textbfLLMSurgeon, a strong framework that casts DMS as an inverse problem under the label-shift assumption. Rather than directly aggregating classifier outputs, LLMSurgeon estimates a calibrated textitsoft confusion matrix and solves a constrained inverse problem to correct systematic domain confusion and recover the latent mixture prior. To evaluate, we introduce textbfLLMScan, a recipe-verifiable evaluation suite built from open-source LLMs with transparent pretraining mixtures. Across LLMScan, LLMSurgeon recovers domain mixtures with high fidelity under fixed protocols. Our work presents a practical, post-hoc approach for auditing the digital DNA of foundation models without access to their training data.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 19 of its 31 references that are in the map. First posted on arXiv 28 May 2026.

Builds on

19 of 31 references.

References · 19

  1. 2016Deep Learning with Differential Privacycited by 2,728
  2. 2016Membership Inference Attacks against Machine Learning Modelscited by 2,281
  3. 2020Extracting Training Data from Large Language Modelscited by 2,189
  4. 2020RealToxicityPrompts: Evaluating Neural Toxic Degeneration in Language Modelscited by 1,255
  5. 2021Membership Inference Attacks From First Principlescited by 878
  6. 2024Dolma: an Open Corpus of Three Trillion Tokens for Language Model Pretraining Researchcited by 483
  7. 2023DoReMi: Optimizing Data Mixtures Speeds Up Language Model Pretrainingcited by 399
  8. 2023Understanding the Effects of RLHF on LLM Generalisation and Diversitycited by 396
  9. 2023Data Selection for Language Models via Importance Resamplingcited by 329
  10. 2020TLDR: Extreme Summarization of Scientific Documentscited by 136
  11. 2023Skill-it! A Data-Driven Skills Framework for Understanding and Training Language Modelscited by 121
  12. 2021Dataset Inference: Ownership Resolution in Machine Learningcited by 116
7 more
  1. 2023LLM360: Towards Fully Transparent Open-Source LLMscited by 104
  2. 2023SlimPajama-DC: Understanding Data Combinations for LLM Trainingcited by 72
  3. 2024ReCaLL: Membership Inference via Relative Conditional Log-Likelihoodscited by 64
  4. 2024SoK: Membership Inference Attacks on LLMs are Rushing Nowhere (and How to Fix It)cited by 59
  5. 2024Pretraining Data Detection for Large Language Models: A Divergence-based Calibration Methodcited by 55
  6. 2024Any-Shift Prompting for Generalization over Distributionscited by 24
  7. 2024Pandora's White-Box: Precise Training Data Detection and Extraction in Large Language Modelscited by 11

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.