LMD3: Language Model Data Density Dependence
Not on the map. Why?
Abstract (arXiv)
We develop a methodology for analyzing language model task performance at the individual example level based on training data density estimation. Experiments with paraphrasing as a controlled intervention on finetuning data demonstrate that increasing the support in the training distribution for specific test queries results in a measurable increase in density, which is also a significant predictor of the performance increase caused by the intervention. Experiments with pretraining data demonstrate that we can explain a significant fraction of the variance in model perplexity via density measurements. We conclude that our framework can provide statistical evidence of the dependence of a target model's predictions on subsets of its training data, and can more generally be used to characterize the support (or lack thereof) in the training data for a given test task.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Builds on 23 of its 29 references that are in the map. First posted on arXiv 10 May 2024.
Builds on
23 of 29 references.
References · 23
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,170
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2017Billion-scale similarity search with GPUscited by 2,497
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,411
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,741
- 2022Quantifying Memorization Across Neural Language Modelscited by 763
- 2020Aligning AI With Shared Human Valuescited by 723
- 2022Large Language Models Struggle to Learn Long-Tail Knowledgecited by 458
- 2023Paraphrasing evades detectors of AI-generated text, but retrieval is an effective defensecited by 429
11 more
- 2023Improving Text Embeddings with Large Language Modelscited by 303
- 2023TRAK: Attributing Model Behavior at Scalecited by 283
- 2023Rethinking Benchmark and Contamination for Language Models with Rephrased Samplescited by 195
- 2022Datamodels: Predicting Predictions from Training Datacited by 189
- 2023On the Reliability of Watermarks for Large Language Modelscited by 189
- 2022Scaling Laws and Interpretability of Learning from Repeated Datacited by 152
- 2023D4: Improving LLM Pretraining via Document De-Duplication and Diversificationcited by 150
- 2024Investigating Data Contamination for Pre-training Language Modelscited by 91
- 2022Datasheet for the Pilecited by 43
- 2023Data Similarity is Not Enough to Explain Language Model Performancecited by 9
- 2021DEANN: Speeding up Kernel-Density Estimation using Approximate Nearest Neighbor Searchcited by 8
Built on by
Citing papers per year: 2024: 4; 2025: 3.
7 papers.
- 2024Generalization v.s. Memorization: Tracing Language Models' Capabilities Back to Pretraining Datacited by 85
- 2024Unleashing the Power of Data Tsunami: A Comprehensive Survey on Data Assessment and Selection for Instruction Tuning of Language Modelscited by 29
- 2024How Much Can We Forget about Data Contamination?cited by 14
- 2025Toward Preference-aligned Large Language Models via Residual-based Model Steeringcited by 3
- 2024Leveraging Open Knowledge for Advancing Task Expertise in Large Language Modelscited by 2
- 2025Interrogating LLM design under a fair learning doctrinecited by 1
- 2025FictionalQA: A Dataset for Studying Memorization and Knowledge Acquisitioncited by 0
- 2025Toward Preference-aligned Large Language Models via Residual-based Model Steering28 Sep 2025
- 2025FictionalQA: A Dataset for Studying Memorization and Knowledge Acquisition5 Jun 2025
- 2025Interrogating LLM design under a fair learning doctrine22 Feb 2025
- 2024How Much Can We Forget about Data Contamination?4 Oct 2024
- 2024Leveraging Open Knowledge for Advancing Task Expertise in Large Language Models28 Aug 2024
- 2024Unleashing the Power of Data Tsunami: A Comprehensive Survey on Data Assessment and Selection for Instruction Tuning of Language Models4 Aug 2024
- 2024Generalization v.s. Memorization: Tracing Language Models' Capabilities Back to Pretraining Data20 Jul 2024