1
Return

Dataset Distillation via a Noise-Unconstrained Generative Model

delete2026-05-06
delete0
PRE
AI
J
Jingxuan Zhang
L
Lei Dai
F
Fei Ye
Z
Zhihua Chen
P
Ping Li
X
Xiaokang Yang
盛斌 (Bin Sheng)
DOI:10.1109/tpami.2026.3690778delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Dataset distillation (DD) aims to synthesize a more compact dataset than the original one and models trained on it are expected to have the same generalization capabilities as on the original dataset. Previous work via a generative model (GM) faces several limitations. First, GM struggles to generate representative samples due to a lack of constraints. Second, it overlooks the relationships between generated samples, limiting its effectiveness. In this paper, a new noise-unconstrained GM-based DD framework is proposed. In the distillation stage, an adaptive matching coefficient is introduced to align generated images with representative class elements and the MiniMax loss function is extended to reduce the optimization difficulty. In the deployment stage, features among each generative image are ensembled by gradient-matching based DD. Theoretical analysis based on McDiarmid’s inequality demonstrates that the proposed components can reduce the generalization error of the original baseline method. We also provide insights into the potential of generated images as an effective proxy dataset for DD. For example, on the ImageWoof dataset with 50 distilled images per class using a 6-layer ConvNet for evaluation, generated images outperform 25%, 50%, and 75% original images by 8.4%, 6.3%, and 8.3% in distillation performance. Our method effectively handles both low- and high-resolution datasets, with experiments on 11 benchmarks demonstrating its efficacy.
Keywords:
Dataset distillation
generative adversarial network
diffusion model
image classification

Journal

IEEE Transactions on Pattern Analysis and Machine Intelligence cover
IEEE Transactions on Pattern Analysis and Machine Intelligence
IF:
18.6
Papers:
831
Citations:
9.8W

Organization

U
university of electronic science and technology of china
Scholars:
1.1W
Papers: 4.3K
Citations: 4
H
hong kong polytechnic university
Scholars:
3.0W
Papers: 4.0W
Citations: 921
S
shanghai jiao tong university
Scholars:
15.1W
Papers: 11.5W
Citations: 159
E
east china university of science and technology
Scholars:
7.3K
Papers: 2.4K
Citations: 3
Cited Papers

Cited Papers

Citing Papers

Citing Papers