Exploring the potential of prototype-based soft-labels data distillation for imbalanced data classification

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Rosu, Radu-Andrei, Breaban, Mihaela-Elena, Luchian, Henri
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913283209953280
author Rosu, Radu-Andrei
Breaban, Mihaela-Elena
Luchian, Henri
author_facet Rosu, Radu-Andrei
Breaban, Mihaela-Elena
Luchian, Henri
contents Dataset distillation aims at synthesizing a dataset by a small number of artificially generated data items, which, when used as training data, reproduce or approximate a machine learning (ML) model as if it were trained on the entire original dataset. Consequently, data distillation methods are usually tied to a specific ML algorithm. While recent literature deals mainly with distillation of large collections of images in the context of neural network models, tabular data distillation is much less represented and mainly focused on a theoretical perspective. The current paper explores the potential of a simple distillation technique previously proposed in the context of Less-than-one shot learning. The main goal is to push further the performance of prototype-based soft-labels distillation in terms of classification accuracy, by integrating optimization steps in the distillation process. The analysis is performed on real-world data sets with various degrees of imbalance. Experimental studies trace the capability of the method to distill the data, but also the opportunity to act as an augmentation method, i.e. to generate new data that is able to increase model accuracy when used in conjunction with - as opposed to instead of - the original data.
format Preprint
id arxiv_https___arxiv_org_abs_2403_17130
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Exploring the potential of prototype-based soft-labels data distillation for imbalanced data classification
Rosu, Radu-Andrei
Breaban, Mihaela-Elena
Luchian, Henri
Machine Learning
Artificial Intelligence
Dataset distillation aims at synthesizing a dataset by a small number of artificially generated data items, which, when used as training data, reproduce or approximate a machine learning (ML) model as if it were trained on the entire original dataset. Consequently, data distillation methods are usually tied to a specific ML algorithm. While recent literature deals mainly with distillation of large collections of images in the context of neural network models, tabular data distillation is much less represented and mainly focused on a theoretical perspective. The current paper explores the potential of a simple distillation technique previously proposed in the context of Less-than-one shot learning. The main goal is to push further the performance of prototype-based soft-labels distillation in terms of classification accuracy, by integrating optimization steps in the distillation process. The analysis is performed on real-world data sets with various degrees of imbalance. Experimental studies trace the capability of the method to distill the data, but also the opportunity to act as an augmentation method, i.e. to generate new data that is able to increase model accuracy when used in conjunction with - as opposed to instead of - the original data.
title Exploring the potential of prototype-based soft-labels data distillation for imbalanced data classification
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2403.17130