Distill Gold from Massive Ores: Bi-level Data Pruning towards Efficient Dataset Distillation
This work addresses the efficiency of dataset distillation for data-efficient learning in large multi-modal models, representing an incremental improvement with specific gains.
The paper tackles the problem of identifying essential samples for dataset distillation by proposing a bi-level data pruning strategy that uses empirical loss and causal effects to select the most contributing samples, achieving state-of-the-art performance on large-scale datasets like ImageNet-1K and Kinetics-400.
Data-efficient learning has garnered significant attention, especially given the current trend of large multi-modal models. Recently, dataset distillation has become an effective approach by synthesizing data samples that are essential for network training. However, it remains to be explored which samples are essential for the dataset distillation process itself. In this work, we study the data efficiency and selection for the dataset distillation task. By re-formulating the dynamics of distillation, we provide insight into the inherent redundancy in the real dataset, both theoretically and empirically. We propose to use the empirical loss value as a static data pruning criterion. To further compensate for the variation of the data value in training, we find the most contributing samples based on their causal effects on the distillation. The proposed selection strategy can efficiently exploit the training dataset, outperform the previous SOTA distillation algorithms, and consistently enhance the distillation algorithms, even on much larger-scale and more heterogeneous datasets, e.g., full ImageNet-1K and Kinetics-400. We believe this paradigm will open up new avenues in the dynamics of distillation and pave the way for efficient dataset distillation. Our code is available on https://github.com/silicx/GoldFromOres-BiLP.