Dataset Distillation from First Principles: Integrating Core Information Extraction and Purposeful Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Kungurtsev, Vyacheslav, Peng, Yuanfang, Gu, Jianyang, Vahidian, Saeed, Quinn, Anthony, Idlahcen, Fadwa, Chen, Yiran
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914934107930624
author Kungurtsev, Vyacheslav
Peng, Yuanfang
Gu, Jianyang
Vahidian, Saeed
Quinn, Anthony
Idlahcen, Fadwa
Chen, Yiran
author_facet Kungurtsev, Vyacheslav
Peng, Yuanfang
Gu, Jianyang
Vahidian, Saeed
Quinn, Anthony
Idlahcen, Fadwa
Chen, Yiran
contents Dataset distillation (DD) is an increasingly important technique that focuses on constructing a synthetic dataset capable of capturing the core information in training data to achieve comparable performance in models trained on the latter. While DD has a wide range of applications, the theory supporting it is less well evolved. New methods of DD are compared on a common set of benchmarks, rather than oriented towards any particular learning task. In this work, we present a formal model of DD, arguing that a precise characterization of the underlying optimization problem must specify the inference task associated with the application of interest. Without this task-specific focus, the DD problem is under-specified, and the selection of a DD algorithm for a particular task is merely heuristic. Our formalization reveals novel applications of DD across different modeling environments. We analyze existing DD methods through this broader lens, highlighting their strengths and limitations in terms of accuracy and faithfulness to optimal DD operation. Finally, we present numerical results for two case studies important in contemporary settings. Firstly, we address a critical challenge in medical data analysis: merging the knowledge from different datasets composed of intersecting, but not identical, sets of features, in order to construct a larger dataset in what is usually a small sample setting. Secondly, we consider out-of-distribution error across boundary conditions for physics-informed neural networks (PINNs), showing the potential for DD to provide more physically faithful data. By establishing this general formulation of DD, we aim to establish a new research paradigm by which DD can be understood and from which new DD techniques can arise.
format Preprint
id arxiv_https___arxiv_org_abs_2409_01410
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Dataset Distillation from First Principles: Integrating Core Information Extraction and Purposeful Learning
Kungurtsev, Vyacheslav
Peng, Yuanfang
Gu, Jianyang
Vahidian, Saeed
Quinn, Anthony
Idlahcen, Fadwa
Chen, Yiran
Machine Learning
Computation
Dataset distillation (DD) is an increasingly important technique that focuses on constructing a synthetic dataset capable of capturing the core information in training data to achieve comparable performance in models trained on the latter. While DD has a wide range of applications, the theory supporting it is less well evolved. New methods of DD are compared on a common set of benchmarks, rather than oriented towards any particular learning task. In this work, we present a formal model of DD, arguing that a precise characterization of the underlying optimization problem must specify the inference task associated with the application of interest. Without this task-specific focus, the DD problem is under-specified, and the selection of a DD algorithm for a particular task is merely heuristic. Our formalization reveals novel applications of DD across different modeling environments. We analyze existing DD methods through this broader lens, highlighting their strengths and limitations in terms of accuracy and faithfulness to optimal DD operation. Finally, we present numerical results for two case studies important in contemporary settings. Firstly, we address a critical challenge in medical data analysis: merging the knowledge from different datasets composed of intersecting, but not identical, sets of features, in order to construct a larger dataset in what is usually a small sample setting. Secondly, we consider out-of-distribution error across boundary conditions for physics-informed neural networks (PINNs), showing the potential for DD to provide more physically faithful data. By establishing this general formulation of DD, we aim to establish a new research paradigm by which DD can be understood and from which new DD techniques can arise.
title Dataset Distillation from First Principles: Integrating Core Information Extraction and Purposeful Learning
topic Machine Learning
Computation
url https://arxiv.org/abs/2409.01410