arXiv · 2403.12040
Distilling Datasets Into Less Than One Image
Abstract
Dataset distillation aims to compress a dataset into a much smaller one so that a model trained on the distilled dataset achieves high accuracy. Current methods frame this as maximizing the distilled classification accuracy for a budget of K distilled images-per-class, where K is a positive integer. In this paper, we push the boundaries of dataset distillation, compressing the dataset into less than an image-per-class. It is important to realize that the meaningful quantity is not the number of distilled images-per-class but the number of distilled pixels-per-dataset. We therefore, propose Poster Dataset Distillation (PoDD), a new approach that distills the entire original dataset into a single poster. The poster approach motivates new technical solutions for creating training images and learnable labels. Our method can achieve comparable or better performance with less than an image-per-class compared to existing methods that use one image-per-class. Specifically, our method establishes a new state-of-the-art performance on CIFAR-10, CIFAR-100, and CUB200 using as little as 0.3 images-per-class.
Explore related subjects
Keep this discovery
Asaf Shul, Eliahu Horwitz, Yedid Hoshen. 2024-03-18. Distilling Datasets Into Less Than One Image. https://arxiv.org/abs/2403.12040
Cite the original work for its findings. Save a collection to share your selection of sources.