Deep Learning on a Data Diet: Finding Important Examples Early in Training
arXiv:2107.07075
Abstract
Recent success in deep learning has partially been driven by training increasingly overparametrized networks on ever larger datasets. It is therefore natural to ask: how much of the data is superfluous, which examples are important for generalization, and how do we find them? In this work, we make the striking observation that, in standard vision datasets, simple scores averaged over several weight initializations can be used to identify important examples very early in training. We propose two such scores -- the Gradient Normed (GraNd) and the Error L2-Norm (EL2N) scores -- and demonstrate their efficacy on a range of architectures and datasets by pruning significant fractions of training data without sacrificing test accuracy. In fact, using EL2N scores calculated a few epochs into training, we can prune half of the CIFAR10 training set while slightly improving test accuracy. Furthermore, for a given dataset, EL2N scores from one architecture or hyperparameter configuration generalize to other configurations. Compared to recent work that prunes data by discarding examples that are rarely forgotten over the course of training, our scores use only local information early in training. We also use our scores to detect noisy examples and study training dynamics through the lens of important examples -- we investigate how the data distribution shapes the loss surface and identify subspaces of the model's data representation that are relatively stable over training.
arXiv comment21 pages, 18 figures
References in corpus (10)
- Neural Tangent Kernel: Convergence and Generalization in Neural Networks
- Understanding Black-box Predictions via Influence Functions
- Active Learning for Convolutional Neural Networks: A Core-Set Approach
- Estimating Training Data Influence by Tracing Gradient Descent
- Coresets for Scalable Bayesian Logistic Regression
- What Neural Networks Memorize and Why: Discovering the Long Tail via Influence Estimation
- The large learning rate phase of deep learning: the catapult mechanism
- Deep Active Learning for Named Entity Recognition
- Coresets for Data-efficient Training of Machine Learning Models
- Coresets for Robust Training of Neural Networks against Noisy Labels