Trustworthy DD · Optimization and training recipes
C^2R
Mind Your Margin and Boundary: Are Your Distilled Datasets Truly Robust?
Muquan Li, Yingyi Ma, Yihong Huang, Hang Gou, Ke Qin, Ming Li, Yuan-Fang Li, Tao He
ICML 2026 · first public 2026-05-20 · arXiv 2605.20606
In one paragraph
Argues prior robust dataset distillation trades off accuracy for robustness poorly because it treats all adversarial examples uniformly and does not explicitly widen inter-class decision-boundary separation; C^2R instead derives a robust-margin perturbation score to build an attack-aware curriculum that prioritizes the smallest-margin adversaries driving robust error, paired with a class-balanced contrastive robustness loss that widens boundary separation. Reports the best robust accuracy across six attacks on CIFAR-10/100, Tiny-ImageNet and ImageNet-1K subsets, beating prior robust DD by 2.8% on average.
Where it sits
- Optimization and training recipes (Orthogonal design choices)
- Setting: Image classification
Abstract (verbatim from arXiv)
Dataset distillation (DD) compresses a large training set into a small synthetic set for efficient training, but most DD methods optimize only clean accuracy and leave robustness uncontrolled. Recent robust DD methods improve robustness, yet they often suffer from a poor accuracy-robustness trade-off because they (i) treat all adversarially perturbed examples uniformly, despite robust risk being dominated by near-zero robust margins, and (ii) do not explicitly increase inter-class separation in the decision boundary where attacks concentrate. We present Contrastive Curriculum for Robust Dataset Distillation (C^2R), a framework that couples an attack-aware curriculum with a contrastive robustness objective. From a robust-margin perspective, we derive a perturbation score that approximates each sample's robust hinge, enabling a curriculum that prioritizes the smallest-margin adversaries that most directly drive robust error. In parallel, a class-balanced contrastive robustness loss enforces adversarial invariance while explicitly widening boundary separation across classes. Experiments on CIFAR-10/100, Tiny-ImageNet, and multiple ImageNet-1K subsets under six attacks show that C^2R achieves the best robust accuracy, outperforming prior robust DD by 2.8% on average.
BibTeX (generated; prefer the venue's official entry)
@article{li2026mind,
title = {Mind Your Margin and Boundary: Are Your Distilled Datasets Truly Robust?},
author = {Muquan Li and Yingyi Ma and Yihong Huang and Hang Gou and Ke Qin and Ming Li and Yuan-Fang Li and Tao He},
journal = {ICML 2026},
year = {2026}
}Nearby in Optimization and training recipes
PTM-ST — Multimodal Dataset Distillation via Phased Teacher Models
Shengbin Guo, Hang Zhao, Senqiao Yang et al. · ICLR 2026notableVision–languagepaper ↗code ↗