Saved in:
Bibliographic Details
Main Authors: Wei, Wei, De Schepper, Tom, Mets, Kevin
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2406.09860
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929386614161408
author Wei, Wei
De Schepper, Tom
Mets, Kevin
author_facet Wei, Wei
De Schepper, Tom
Mets, Kevin
contents Dataset condensation (DC) methods aim to learn a smaller synthesized dataset with informative data records to accelerate the training of machine learning models. Current distribution matching (DM) based DC methods learn a synthesized dataset by matching the mean of the latent embeddings between the synthetic and the real dataset. However two distributions with the same mean can still be vastly different. In this work we demonstrate the shortcomings of using Maximum Mean Discrepancy to match latent distributions i.e. the weak matching power and lack of outlier regularization. To alleviate these shortcomings we propose our new method: Latent Quantile Matching (LQM) which matches the quantiles of the latent embeddings to minimize the goodness of fit test statistic between two distributions. Empirical experiments on both image and graph-structured datasets show that LQM matches or outperforms previous state of the art in distribution matching based DC. Moreover we show that LQM improves the performance in continual graph learning (CGL) setting where memory efficiency and privacy can be important. Our work sheds light on the application of DM based DC for CGL.
format Preprint
id arxiv_https___arxiv_org_abs_2406_09860
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Dataset Condensation with Latent Quantile Matching
Wei, Wei
De Schepper, Tom
Mets, Kevin
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
Dataset condensation (DC) methods aim to learn a smaller synthesized dataset with informative data records to accelerate the training of machine learning models. Current distribution matching (DM) based DC methods learn a synthesized dataset by matching the mean of the latent embeddings between the synthetic and the real dataset. However two distributions with the same mean can still be vastly different. In this work we demonstrate the shortcomings of using Maximum Mean Discrepancy to match latent distributions i.e. the weak matching power and lack of outlier regularization. To alleviate these shortcomings we propose our new method: Latent Quantile Matching (LQM) which matches the quantiles of the latent embeddings to minimize the goodness of fit test statistic between two distributions. Empirical experiments on both image and graph-structured datasets show that LQM matches or outperforms previous state of the art in distribution matching based DC. Moreover we show that LQM improves the performance in continual graph learning (CGL) setting where memory efficiency and privacy can be important. Our work sheds light on the application of DM based DC for CGL.
title Dataset Condensation with Latent Quantile Matching
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2406.09860