Dataset distillation (DD) aims to compress large-scale datasets into compact synthetic counterparts for efficient model training. However, existing DD methods exhibit substantial performance degradation on long-tailed datasets. We identify two fundamental challenges: heuristic design choices for distribution discrepancy measure and uniform treatment of imbalanced classes. To address these limitations, we propose Class-Aware Spectral Distribution Matching (CSDM), which reformulates distribution alignment via the spectrum of a well-behaved kernel function. This technique maps the original samples into frequency space, resulting in the Spectral Distribution Distance (SDD). To mitigate class imbalance, we exploit the unified form of SDD to perform amplitude-phase decomposition, which adaptively prioritizes the realism in tail classes. On CIFAR-10-LT, with 10 images per class, CSDM achieves a 14.0% improvement over state-of-the-art DD methods, with only a 5.7% performance drop when the number of images in tail classes decreases from 500 to 25, demonstrating strong stability on long-tailed data.
Figures & tables
Figure 1: Dataset Distillation with Class-Aware Spectral Distribution Matching (CSDM). (a) Kernel Embedding. A feature network extracts features from real and synthetic data. Unlike previous methods using a linear kernel that maps to a suboptimal RKHS, we design a universal kernel that enables accurate measurement of distribution discrepancies. (b) Class-Aware Spectral Distribution Matching. We quantify distribution discrepancies using the Spectral Distribution Distance (SDD). By decomposing SDD into amplitude and phase and applying class-specific weights, CSDM dynamically adapts to class imbalance.
Kernel
Universal
Shift-Invariant
K(x,y)
p(t)
RBF
✓
✓
exp(−γ∥x−y∥2)
∝exp(−∥t∥2/(4γ))
Laplace
✓
✓
exp(−γ∥x−y∥)
∝∏i=1dγ2+ti2γ
Linear
×
×
x⊤y
–
Polynomial
×
×
(x⊤y+c)d
–
Sigmoid
×
×
tanh(αx⊤y+c)
–
Cosine
×
✓
cos(ω∥x−y∥)
∝δ(∥t∥−ω)
Table 1: Comparison of commonly used kernels.
CIFAR-10-LT
CIFAR-100-LT
Imbalance Factor
10
50
10
IPC
10
20
50
10
20
50
10
20
50
Random
32.5 ± 2.2
39.6 ± 0.9
51.9 ± 1.5
33.2 ± 0.4
42.0 ± 1.3
51.6 ± 1.3
14.2 ± 0.6
21.7 ± 0.6
32.1 ± 0.6
K-Center
21.9 ± 0.8
24.2 ± 0.8
31.7 ± 0.9
17.8 ± 0.2
20.8 ± 0.5
26.1 ± 0.2
10.7 ± 0.9
15.9 ± 1.0
24.8 ± 0.2
Graph-Cut
28.7 ± 0.9
34.2 ± 1.0
40.6 ± 1.0
24.2 ± 0.7
28.6 ± 0.8
33.9 ± 0.4
16.9 ± 0.3
22.2 ± 0.4
29.9 ± 0.4
DC
37.9 ± 0.9
38.5 ± 0.9
37.4 ± 1.4
37.3 ± 0.9
37.3 ± 0.9
35.8 ± 1.2
24.0 ± 0.3
27.4 ± 0.3
27.4 ± 0.3
Table 2: Results on CIFAR-10-LT and CIFAR-100-LT.
Dataset
ImageMeow
ImageSquawk
Imbalance Factor
5
10
5
10
IPC
1
10
1
10
1
10
1
10
Random
12.8 ± 0.6
14.5 ± 1.0
10.2 ± 0.7
10.3 ± 0.6
13.1 ± 0.9
13.2 ± 0.8
9.6 ± 0.6
11.7 ± 0.8
DM
9.5 ± 0.8
12.5 ± 1.3
8.4 ± 1.1
12.8 ± 1.3
10.0 ± 2.3
10.8 ± 0.8
8.3 ± 0.8
10.8 ± 0.3
MTT
12.3 ± 0.9
13.2 ± 1.0
11.1 ± 1.0
14.1 ± 0.5
11.1 ± 1.4
12.5 ± 0.8
9.2 ± 0.7
12.2 ± 1.1
RDED
9.9 ± 0.6
13.7 ± 0.7
9.9 ± 1.0
12.7 ± 0.7
10.5 ± 1.1
11.6 ± 0.5
7.8 ± 0.8
10.8 ± 0.4
Table 3: Results on long-tailed ImageNet Subsets.
Figure 5
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
Dataset
CIFAR-10
CIFAR-100
IPC
1
10
50
10
50
Ratio (%)
0.02
0.2
2
2
10
Random
14.4±0.2
26.0±1.0
43.4±1.0
15.1±0.5
30.7±0.4
Herding
21.5±1.2
31.0±1.0
46.3±1.1
18.5±1.0
34.5±0.6
Forgetting
13.5±1.2
23.3±1.0
23.3±1.1
15.1±1.0
30.5±0.7
DC
28.3±0.5
44.9±0.5
49.5±0.5
25.5±0.5
34.8±0.5
Appendix
Table 5: Experimental Results on CIFAR-10 and CIFAR-100 Datasets
Figure 4: Performance of Different Kernel Methods on CIFAR-10LT with Varying Imbalance Factors.
Figure 5: Effect of the α parameter: Images synthesized by CSDM on CIFAR-10 (IPC=10) for values of 0 and 1. As shown in the images, amplitude-only distillation ( α=1 ) yields diverse but less realistic images, whereas phase-only distillation ( α=0 ) produces realistic images but suffers from limited diversity.
Dataset distillation compresses large-scale datasets into compact synthetic sets while preserving their training utility, enabling efficient 3D point cloud training. Current point cloud dataset distillation methods only tackle geometric and representation challenges while ignoring the distributional imbalance prevalent in point cloud datasets where both training and test splits follow long-tailed class distributions. To our knowledge, we present the first study on long-tailed point cloud dataset distillation. Rather than focusing primarily on geometric and representation properties or simply constructing a class-balanced synthetic set, our framework explicitly accounts for long-tailed class distributions via two core modules. First, we design Adaptive Synthetic Budgeting to allocate class-wise synthetic budgets according to class quantity and the expected benefit of additional synthetic samples. Given the allocated budgets, we further design 3D Long-Tailed Distribution Matching to optimize synthetic point clouds through Global-Local Feature Alignment and Prior-Aware Supervision. The former preserves both global class distributions and diverse intra-class structures, while the latter provides class-dependent expert supervision to keep tail-class samples recognizable while maintaining diverse head-class patterns. Extensive experiments demonstrate the effectiveness of our method, lifting classification accuracy by 7.0 points on ShapeNet55 against state-of-the-art methods.
Jiahao You, Xu Han, Jinfeng Xu +1
Huazhong University of Science and Technology · Hunan University
Dataset Distillation (DD) synthesizes a compact synthetic dataset that preserves the training utility of a full dataset. However, its standard formulation assumes that test data follow the same distribution as training data, an assumption that rarely holds in practice. A straightforward extension-applying post-hoc Domain Generalization (DG) techniques to distilled data-is ill-suited because existing DG methods rely on the natural diversity of real datasets, which compact synthetic sets inherently lack, while also incurring substantial augmentation overhead that conflicts with the efficiency objective of dataset distillation. To address this limitation, we introduce Domain Generalizable Dataset Distillation (DGDD), a new problem setting that explicitly targets out-of-distribution (OOD) generalization of distilled datasets. We study this problem through a widely adopted DD baseline of Distribution Matching (DM). We attribute the OOD vulnerability of DM to the entanglement of class-discriminative and domain-specific information within the compressed synthetic set, and propose Spectral Gradient Surgery (SGS) to disentangle the two. The key insight of SGS is that cross-domain agreement among domain-wise gradients in the spectral domain reveals which gradient components are shared across source domains-and are therefore class-discriminative-and which are domain-specific. Based on this observation, SGS augments the standard DM update with two complementary gradients: one that reinforces cross-domain shared components and another that explicitly promotes diversity within the distilled dataset. Extensive experiments on diverse-scale benchmarks demonstrate that SGS substantially improves OOD generalization while remaining plug-and-play compatible with existing DM methods.
Minyoung Oh, Najeong Chae, Jae-Young Sim
Graduate School of Artificial Intelligence Ulsan National Institute of Science and Technology (UNIST) Ulsan, Republic of Korea
Dataset Distillation aims to synthesize compact datasets that can approximate the training efficacy of large-scale real datasets, offering an efficient solution to the increasing computational demands of modern deep learning. Recently, diffusion-based dataset distillation methods have shown great promise by leveraging the strong generative capacity of diffusion models to produce diverse and structurally consistent samples. However, a fundamental goal misalignment persists: diffusion models are optimized for generative likelihood rather than discriminative utility, resulting in over-concentration in high-density regions and inadequate coverage of boundary samples crucial for classification. To address this issue, we propose two complementary strategies. Inversion-Matching (IM) introduces an inversion-guided fine-tuning process that aligns denoising trajectories with their inversion counterparts, broadening distributional coverage and enhancing diversity. Selective Subgroup Sampling(S^3) is a training-free sampling mechanism that improves inter-class separability by selecting synthetic subsets that are both representative and distinctive. Extensive experiments demonstrate that our approach significantly enhances the discriminative quality and generalization of distilled datasets, achieving state-of-the-art performance among diffusion-based methods.
Chenru Wang, Yunyi Chen, Zijun Yang +2
AGI Lab, Westlake University · Eindhoven University of Technology · Jinan University +2