Not All Samples Should Be Utilized Equally: Towards Understanding and Improving Dataset Distillation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Wang, Shaobo, Yang, Yantai, Wang, Qilong, Li, Kaixin, Zhang, Linfeng, Yan, Junchi
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913883941240832
author Wang, Shaobo
Yang, Yantai
Wang, Qilong
Li, Kaixin
Zhang, Linfeng
Yan, Junchi
author_facet Wang, Shaobo
Yang, Yantai
Wang, Qilong
Li, Kaixin
Zhang, Linfeng
Yan, Junchi
contents Dataset Distillation (DD) aims to synthesize a small dataset capable of performing comparably to the original dataset. Despite the success of numerous DD methods, theoretical exploration of this area remains unaddressed. In this paper, we take an initial step towards understanding various matching-based DD methods from the perspective of sample difficulty. We begin by empirically examining sample difficulty, measured by gradient norm, and observe that different matching-based methods roughly correspond to specific difficulty tendencies. We then extend the neural scaling laws of data pruning to DD to theoretically explain these matching-based methods. Our findings suggest that prioritizing the synthesis of easier samples from the original dataset can enhance the quality of distilled datasets, especially in low IPC (image-per-class) settings. Based on our empirical observations and theoretical analysis, we introduce the Sample Difficulty Correction (SDC) approach, designed to predominantly generate easier samples to achieve higher dataset quality. Our SDC can be seamlessly integrated into existing methods as a plugin with minimal code adjustments. Experimental results demonstrate that adding SDC generates higher-quality distilled datasets across 7 distillation methods and 6 datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2408_12483
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Not All Samples Should Be Utilized Equally: Towards Understanding and Improving Dataset Distillation
Wang, Shaobo
Yang, Yantai
Wang, Qilong
Li, Kaixin
Zhang, Linfeng
Yan, Junchi
Computer Vision and Pattern Recognition
Artificial Intelligence
Dataset Distillation (DD) aims to synthesize a small dataset capable of performing comparably to the original dataset. Despite the success of numerous DD methods, theoretical exploration of this area remains unaddressed. In this paper, we take an initial step towards understanding various matching-based DD methods from the perspective of sample difficulty. We begin by empirically examining sample difficulty, measured by gradient norm, and observe that different matching-based methods roughly correspond to specific difficulty tendencies. We then extend the neural scaling laws of data pruning to DD to theoretically explain these matching-based methods. Our findings suggest that prioritizing the synthesis of easier samples from the original dataset can enhance the quality of distilled datasets, especially in low IPC (image-per-class) settings. Based on our empirical observations and theoretical analysis, we introduce the Sample Difficulty Correction (SDC) approach, designed to predominantly generate easier samples to achieve higher dataset quality. Our SDC can be seamlessly integrated into existing methods as a plugin with minimal code adjustments. Experimental results demonstrate that adding SDC generates higher-quality distilled datasets across 7 distillation methods and 6 datasets.
title Not All Samples Should Be Utilized Equally: Towards Understanding and Improving Dataset Distillation
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2408.12483