When resampling/reweighting improves feature learning in imbalanced classification?: A toy-model study

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Obuchi, Tomoyuki, Tanaka, Toshiyuki
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866908330511826944
author Obuchi, Tomoyuki
Tanaka, Toshiyuki
author_facet Obuchi, Tomoyuki
Tanaka, Toshiyuki
contents A toy model of binary classification is studied with the aim of clarifying the class-wise resampling/reweighting effect on the feature learning performance under the presence of class imbalance. In the analysis, a high-dimensional limit of the input space is taken while keeping the ratio of the dataset size against the input dimension finite and the non-rigorous replica method from statistical mechanics is employed. The result shows that there exists a case in which the no resampling/reweighting situation gives the best feature learning performance irrespectively of the choice of losses or classifiers, supporting recent findings in Cao et al. (2019); Kang et al. (2019). It is also revealed that the key of the result is the symmetry of the loss and the problem setting. Inspired by this, we propose a further simplified model exhibiting the same property in the multiclass setting. These clarify when the class-wise resampling/reweighting becomes effective in imbalanced classification.
format Preprint
id arxiv_https___arxiv_org_abs_2409_05598
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle When resampling/reweighting improves feature learning in imbalanced classification?: A toy-model study
Obuchi, Tomoyuki
Tanaka, Toshiyuki
Machine Learning
Disordered Systems and Neural Networks
Information Theory
A toy model of binary classification is studied with the aim of clarifying the class-wise resampling/reweighting effect on the feature learning performance under the presence of class imbalance. In the analysis, a high-dimensional limit of the input space is taken while keeping the ratio of the dataset size against the input dimension finite and the non-rigorous replica method from statistical mechanics is employed. The result shows that there exists a case in which the no resampling/reweighting situation gives the best feature learning performance irrespectively of the choice of losses or classifiers, supporting recent findings in Cao et al. (2019); Kang et al. (2019). It is also revealed that the key of the result is the symmetry of the loss and the problem setting. Inspired by this, we propose a further simplified model exhibiting the same property in the multiclass setting. These clarify when the class-wise resampling/reweighting becomes effective in imbalanced classification.
title When resampling/reweighting improves feature learning in imbalanced classification?: A toy-model study
topic Machine Learning
Disordered Systems and Neural Networks
Information Theory
url https://arxiv.org/abs/2409.05598