Difficulty-guided Sampling: Bridging the Target Gap between Dataset Distillation and Downstream Tasks

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Li, Mingzhuo, Li, Guang, Ye, Linfeng, Mao, Jiafeng, Ogawa, Takahiro, Plataniotis, Konstantinos N., Haseyama, Miki
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866908767380045824
author Li, Mingzhuo
Li, Guang
Ye, Linfeng
Mao, Jiafeng
Ogawa, Takahiro
Plataniotis, Konstantinos N.
Haseyama, Miki
author_facet Li, Mingzhuo
Li, Guang
Ye, Linfeng
Mao, Jiafeng
Ogawa, Takahiro
Plataniotis, Konstantinos N.
Haseyama, Miki
contents In this paper, we propose difficulty-guided sampling (DGS) to bridge the target gap between the distillation objective and the downstream task, therefore improving the performance of dataset distillation. Deep neural networks achieve remarkable performance but have time and storage-consuming training processes. Dataset distillation is proposed to generate compact, high-quality distilled datasets, enabling effective model training while maintaining downstream performance. Existing approaches typically focus on features extracted from the original dataset, overlooking task-specific information, which leads to a target gap between the distillation objective and the downstream task. We propose leveraging characteristics that benefit the downstream training into data distillation to bridge this gap. Focusing on the downstream task of image classification, we introduce the concept of difficulty and propose DGS as a plug-in post-stage sampling module. Following the specific target difficulty distribution, the final distilled dataset is sampled from image pools generated by existing methods. We also propose difficulty-aware guidance (DAG) to explore the effect of difficulty in the generation process. Extensive experiments across multiple settings demonstrate the effectiveness of the proposed methods. It also highlights the broader potential of difficulty for diverse downstream tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2601_10090
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Difficulty-guided Sampling: Bridging the Target Gap between Dataset Distillation and Downstream Tasks
Li, Mingzhuo
Li, Guang
Ye, Linfeng
Mao, Jiafeng
Ogawa, Takahiro
Plataniotis, Konstantinos N.
Haseyama, Miki
Computer Vision and Pattern Recognition
Artificial Intelligence
In this paper, we propose difficulty-guided sampling (DGS) to bridge the target gap between the distillation objective and the downstream task, therefore improving the performance of dataset distillation. Deep neural networks achieve remarkable performance but have time and storage-consuming training processes. Dataset distillation is proposed to generate compact, high-quality distilled datasets, enabling effective model training while maintaining downstream performance. Existing approaches typically focus on features extracted from the original dataset, overlooking task-specific information, which leads to a target gap between the distillation objective and the downstream task. We propose leveraging characteristics that benefit the downstream training into data distillation to bridge this gap. Focusing on the downstream task of image classification, we introduce the concept of difficulty and propose DGS as a plug-in post-stage sampling module. Following the specific target difficulty distribution, the final distilled dataset is sampled from image pools generated by existing methods. We also propose difficulty-aware guidance (DAG) to explore the effect of difficulty in the generation process. Extensive experiments across multiple settings demonstrate the effectiveness of the proposed methods. It also highlights the broader potential of difficulty for diverse downstream tasks.
title Difficulty-guided Sampling: Bridging the Target Gap between Dataset Distillation and Downstream Tasks
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2601.10090