Restoring balance: principled under/oversampling of data for optimal classification

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Loffredo, Emanuele, Pastore, Mauro, Cocco, Simona, Monasson, Rémi
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866913672161394688
author Loffredo, Emanuele
Pastore, Mauro
Cocco, Simona
Monasson, Rémi
author_facet Loffredo, Emanuele
Pastore, Mauro
Cocco, Simona
Monasson, Rémi
contents Class imbalance in real-world data poses a common bottleneck for machine learning tasks, since achieving good generalization on under-represented examples is often challenging. Mitigation strategies, such as under or oversampling the data depending on their abundances, are routinely proposed and tested empirically, but how they should adapt to the data statistics remains poorly understood. In this work, we determine exact analytical expressions of the generalization curves in the high-dimensional regime for linear classifiers (Support Vector Machines). We also provide a sharp prediction of the effects of under/oversampling strategies depending on class imbalance, first and second moments of the data, and the metrics of performance considered. We show that mixed strategies involving under and oversampling of data lead to performance improvement. Through numerical experiments, we show the relevance of our theoretical predictions on real datasets, on deeper architectures and with sampling strategies based on unsupervised probabilistic models.
format Preprint
id arxiv_https___arxiv_org_abs_2405_09535
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Restoring balance: principled under/oversampling of data for optimal classification
Loffredo, Emanuele
Pastore, Mauro
Cocco, Simona
Monasson, Rémi
Disordered Systems and Neural Networks
Machine Learning
Class imbalance in real-world data poses a common bottleneck for machine learning tasks, since achieving good generalization on under-represented examples is often challenging. Mitigation strategies, such as under or oversampling the data depending on their abundances, are routinely proposed and tested empirically, but how they should adapt to the data statistics remains poorly understood. In this work, we determine exact analytical expressions of the generalization curves in the high-dimensional regime for linear classifiers (Support Vector Machines). We also provide a sharp prediction of the effects of under/oversampling strategies depending on class imbalance, first and second moments of the data, and the metrics of performance considered. We show that mixed strategies involving under and oversampling of data lead to performance improvement. Through numerical experiments, we show the relevance of our theoretical predictions on real datasets, on deeper architectures and with sampling strategies based on unsupervised probabilistic models.
title Restoring balance: principled under/oversampling of data for optimal classification
topic Disordered Systems and Neural Networks
Machine Learning
url https://arxiv.org/abs/2405.09535