Deep Learning on a Data Diet

Last modified: July 21, 2026

#data-efficient-ml

Proposed Ideas

  • 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
  • a scoring method to identify diff and important samples early on in training
  • identify how this score pruned sub-population affect loss surface and the training dynamics

Imp Results

  • we can prune 25% of CIFAR-100 (verified by IISc task results) compared to 50% of CIFAR-10 due to difficulty in learning more variety.
  • Performance drops at very high pruning levels. They have hypothesised that this might be due to a significant sub-population being excluded from the training dataset.
  • EL2N scores is a property of the dataset and not specific to a network. They show that a ResNet18 and a ResNet50 trained on CIFAR-10 have similar performance curves and the same amount of data can be pruned,.
    • E2LN scores calculated for one set of networks and hyp-param config can be used to pruned for different architectures and configs.
  • They also found that the very high scored data points tend to be either unrepresentative outliers of a class, have non standard backgrounds or odd angles, are subject to label noise, or are otherwise difficult.

Proposed Ideas

  • GraNd Score (Gradient Normed)

    • The GraNd score of a training example $(x, y)$ at a specific training time $t$ is defined as the expected L2-norm of the loss gradient. It measures how much an individual example influences the change in training loss for the rest of the dataset.$$\chi_{t}(x,y)=\mathbb{E}_{w_{t}}||g_{t}(x,y)||_{2}$$
    • Components: $g_{t}(x,y)$ is the gradient of the loss with respect to the weights $w_t$.
    • Expectation: The L2 norm is calculated over the vector of all model weight gradients. It is also averaged over multiple random weight initializations (different seeds)to remove dependence on specific initial weights.
    • Often calculated very early or at initialisation. But since the model hasn’t learnt the data structure yet, its not that effective for pruning. Warmup epochs + EL2n performs better
  • EL2N Score (Error L2-Norm)

    • The EL2N score is a simplified approximation of the GraNd score that becomes accurate after a few epochs of training. It measures the expected L2-norm of the error vector, which is the difference between the model’s predicted probabilities and the actual label.$$EL2N\space Score=\mathbb{E}||p(w_{t},x)-y||_{2}$$
    • Components: $p(w_t, x)$ is the network’s output probability vector and $y$ is the one-hot encoded label.
    • Utility: It is computationally cheaper than GraNd because it does not require calculating gradients through the entire network, yet it often provides an even stronger signal for data pruning.