Rethinking Representativeness and Diversity in Dynamic Data Selection

TL;DR

This paper introduces a novel dynamic data selection method that improves training efficiency by focusing on dataset coverage and process-level diversity, achieving over 2x acceleration without sacrificing accuracy.

Contribution

It redefines representativeness and diversity in data selection, proposing a framework that balances frequent and rare factors during training without extra computational overhead.

Findings

Achieves over 2x training acceleration.

Matches or exceeds full-data accuracy.

Demonstrates effectiveness across vision and text benchmarks.

Abstract

Dynamic data selection accelerates training by sampling a changing subset of the dataset while preserving accuracy. We rethink two core notions underlying sample evaluation: representativeness and diversity. Instead of local geometric centrality, we define representativeness as coverage of dataset-level common or high-frequency feature factors. Instead of within-subset dispersion, we define diversity at the process level, requiring the selection trajectory to gradually include complementary rare factors over training. Based on this view, we propose a dynamic selection framework with three components. First, we score representativeness in a plug-in feature space to prioritize samples covering frequent factors. We instantiate this with a sparse autoencoder trained on the target dataset, using sparse unit activations to summarize both individual samples and dataset-wide factor statistics. Second, we realize process-level diversity by combining rare-factor sampling with a Usage-Frequency Penalty that promotes sample rotation, provably discourages monopoly, and reduces gradient bias. Third, we couple the two-dimensional scoring with a smooth scheduler that transitions selection from core-pattern consolidation to rare-factor exploration, without extra gradients, influence estimates, or second-order computations on the training model. Extensive experiments on five benchmarks across vision and text tasks demonstrate improved accuracy-efficiency trade-offs across models. Our method matches or exceeds full-data accuracy with over 2x training acceleration. Code will be released.