LLMSurgeon: Diagnosing Data Mixture of Large Language Models
cs.CL, cs.AI, cs.LG
Submitted: 2026-05-28
Updated: 2026-10-15
Comments: ACL 2026 Main. Code at https://github.com/Yaxin9Luo/LLMSurgeon
Code: https://github.com/Yaxin9Luo/LLMSurgeon
License: http://creativecommons.org/licenses/by/4.0/
The gist: The pretraining data mixture of Large Language Models (LLMs) constitutes their "digital DNA", shaping model behaviors, capabilities, and failure modes.
Terminology
Abstract
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 LLMSurgeon, 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 soft confusion matrix and solves a constrained inverse problem to correct systematic domain confusion and recover the latent mixture prior. To evaluate, we introduce LLMScan, 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.
Sources
- TLDR: Extreme Summarization of Scientific Documents
- Do Membership Inference Attacks Work on Large Language Models?
- The Pile: An 800GB Dataset of Diverse Text for Language Modeling
- OLMo: Accelerating the Science of Language Models
- Understanding the Effects of RLHF on LLM Generalisation and Diversity
- The Stack: 3 TB of permissively licensed source code
- StarCoder: may the source be with you!
- DeepSeek-V3.2: Pushing the Frontier of Open Large Language Models
- LLM360: Towards Fully Transparent Open-Source LLMs
- Dataset Inference: Ownership Resolution in Machine Learning
- SlimPajama-DC: Understanding Data Combinations for LLM Training
- Detecting Pretraining Data from Large Language Models
- LLaMA: Open and Efficient Foundation Language Models
- Pandora's White-Box: Precise Training Data Detection and Extraction in Large Language Models
- ReCaLL: Membership Inference via Relative Conditional Log-Likelihoods
- Qwen3 Technical Report
- Membership Inference Attacks Cannot Prove that a Model Was Trained On Your Data
- Min-K%++: Improved Baseline for Detecting Pre-Training Data from Large Language Models
- Pretraining Data Detection for Large Language Models: A Divergence-based Calibration Method
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering