We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.CV

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Computer Vision and Pattern Recognition

Title: Dataset Condensation with Gradient Matching

Abstract: As the state-of-the-art machine learning methods in many fields rely on larger datasets, storing datasets and training models on them become significantly more expensive. This paper proposes a training set synthesis technique for data-efficient learning, called Dataset Condensation, that learns to condense large dataset into a small set of informative synthetic samples for training deep neural networks from scratch. We formulate this goal as a gradient matching problem between the gradients of deep neural network weights that are trained on the original and our synthetic data. We rigorously evaluate its performance in several computer vision benchmarks and demonstrate that it significantly outperforms the state-of-the-art methods. Finally we explore the use of our method in continual learning and neural architecture search and report promising gains when limited memory and computations are available.
Subjects: Computer Vision and Pattern Recognition (cs.CV); Machine Learning (cs.LG)
Journal reference: International Conference on Learning Representations 2021
Cite as: arXiv:2006.05929 [cs.CV]
  (or arXiv:2006.05929v3 [cs.CV] for this version)

Submission history

From: Bo Zhao [view email]
[v1] Wed, 10 Jun 2020 16:30:52 GMT (2520kb,D)
[v2] Sat, 3 Oct 2020 12:49:58 GMT (2757kb,D)
[v3] Mon, 8 Mar 2021 13:31:22 GMT (2598kb,D)

Link back to: arXiv, form interface, contact.