Data-Efficient Generation for Dataset Distillation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Li, Zhe, Zhang, Weitong, Cechnicka, Sarah, Kainz, Bernhard
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914940103688192
author Li, Zhe
Zhang, Weitong
Cechnicka, Sarah
Kainz, Bernhard
author_facet Li, Zhe
Zhang, Weitong
Cechnicka, Sarah
Kainz, Bernhard
contents While deep learning techniques have proven successful in image-related tasks, the exponentially increased data storage and computation costs become a significant challenge. Dataset distillation addresses these challenges by synthesizing only a few images for each class that encapsulate all essential information. Most current methods focus on matching. The problems lie in the synthetic images not being human-readable and the dataset performance being insufficient for downstream learning tasks. Moreover, the distillation time can quickly get out of bounds when the number of synthetic images per class increases even slightly. To address this, we train a class conditional latent diffusion model capable of generating realistic synthetic images with labels. The sampling time can be reduced to several tens of images per seconds. We demonstrate that models can be effectively trained using only a small set of synthetic images and evaluated on a large real test set. Our approach achieved rank \(1\) in The First Dataset Distillation Challenge at ECCV 2024 on the CIFAR100 and TinyImageNet datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2409_03929
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Data-Efficient Generation for Dataset Distillation
Li, Zhe
Zhang, Weitong
Cechnicka, Sarah
Kainz, Bernhard
Computer Vision and Pattern Recognition
While deep learning techniques have proven successful in image-related tasks, the exponentially increased data storage and computation costs become a significant challenge. Dataset distillation addresses these challenges by synthesizing only a few images for each class that encapsulate all essential information. Most current methods focus on matching. The problems lie in the synthetic images not being human-readable and the dataset performance being insufficient for downstream learning tasks. Moreover, the distillation time can quickly get out of bounds when the number of synthetic images per class increases even slightly. To address this, we train a class conditional latent diffusion model capable of generating realistic synthetic images with labels. The sampling time can be reduced to several tens of images per seconds. We demonstrate that models can be effectively trained using only a small set of synthetic images and evaluated on a large real test set. Our approach achieved rank \(1\) in The First Dataset Distillation Challenge at ECCV 2024 on the CIFAR100 and TinyImageNet datasets.
title Data-Efficient Generation for Dataset Distillation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.03929