Measuring Data Diversity with Embeddings: A Taxonomy and Benchmark for Diversity Measures
On the map: LLM pre-training data and scaling laws → Data mixtures and data selection also fits: Pre-training corpora and quality filtering
Abstract (arXiv)
Diversity is a critical property of text corpora, with direct implications for training data curation, test set design, and model generalization. However, its measurement remains fragmented and poorly understood. Existing embedding-based diversity measures vary widely in how they are computed and can yield substantially different diversity scores on the same data, yet no systematic comparison of their empirical behaviour exists. We make two contributions toward addressing this gap. First, we present a taxonomy of 21 embedding-based diversity measures, organizing them into four families based on their underlying computations. Second, we introduce EmbDivBench, a benchmark that evaluates these measures along three conceptually distinct components (variety, balance and disparity) on both simulated and natural text data. Although Mean Pairwise Distance is the most commonly used measure, our experiments show that it is outperformed by several alternatives on the three diversity components, with Graph Entropy being the best choice overall. Lastly, we distil these findings into practical recommendations for diversity measure selection.
Key facts. No citing papers in the map yet (data through 19 Oct 2026). Placed in Data mixtures and data selection, one of 8 directions in LLM pre-training data and scaling laws. First posted on arXiv 17 Oct 2026.
Builds on
0 of 0 references.
No references here yet.
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2023Beyond Scale: The Diversity Coefficient as a Data Quality Metric for Variability in Natural Language Data
- 2025Measuring Diversity in Synthetic Datasets
- 2026PLDiv: Measuring Dataset Diversity and Semantic Collapse via Persistence Landscapes
- 2025Analyzing Similarity Metrics for Data Selection for Language Model Pretraining
- 2024Recent advances in text embedding: A Comprehensive Review of Top-Performing Methods on the MTEB Benchmark
- 2025Interpretable Embeddings with Sparse Autoencoders: A Data Analysis Toolkit
- 2025Learning from the Best, Differently: A Diversity-Driven Rethinking on Data Selection
- 2023When Less is More: Investigating Data Pruning for Pretraining LLMs at Scale