DeepAI AI Chat
Log In Sign Up

Dataset Distillation by Matching Training Trajectories

by   George Cazenavette, et al.

Dataset distillation is the task of synthesizing a small dataset such that a model trained on the synthetic set will match the test accuracy of the model trained on the full dataset. In this paper, we propose a new formulation that optimizes our distilled data to guide networks to a similar state as those trained on real data across many training steps. Given a network, we train it for several iterations on our distilled data and optimize the distilled data with respect to the distance between the synthetically trained parameters and the parameters trained on real data. To efficiently obtain the initial and target network parameters for large-scale datasets, we pre-compute and store training trajectories of expert networks trained on the real dataset. Our method handily outperforms existing methods and also allows us to distill higher-resolution visual data.


page 15

page 16

page 17

page 18

page 19

page 20

page 21

page 22


Rethinking Data Distillation: Do Not Overlook Calibration

Neural networks trained on distilled data often produce over-confident o...

Dataset Distillation using Parameter Pruning

The acquisition of advanced models relies on large datasets in many fiel...

Photometric Data-driven Classification of Type Ia Supernovae in the Open Supernova Catalog

We propose a novel approach for a machine-learning-based detection of th...

DiM: Distilling Dataset into Generative Model

Dataset distillation reduces the network training cost by synthesizing s...

Minimizing the Accumulated Trajectory Error to Improve Dataset Distillation

Model-based deep learning has achieved astounding successes due in part ...

Data Incubation – Synthesizing Missing Data for Handwriting Recognition

In this paper, we demonstrate how a generative model can be used to buil...

Code Repositories


Official code for our CVPR '22 paper "Dataset Distillation by Matching Training Trajectories"

view repo