English

Open-Sampling: Exploring Out-of-Distribution data for Re-balancing Long-tailed datasets

Machine Learning 2022-07-06 v3 Computer Vision and Pattern Recognition

Abstract

Deep neural networks usually perform poorly when the training dataset suffers from extreme class imbalance. Recent studies found that directly training with out-of-distribution data (i.e., open-set samples) in a semi-supervised manner would harm the generalization performance. In this work, we theoretically show that out-of-distribution data can still be leveraged to augment the minority classes from a Bayesian perspective. Based on this motivation, we propose a novel method called Open-sampling, which utilizes open-set noisy labels to re-balance the class priors of the training dataset. For each open-set instance, the label is sampled from our pre-defined distribution that is complementary to the distribution of original class priors. We empirically show that Open-sampling not only re-balances the class priors but also encourages the neural network to learn separable representations. Extensive experiments demonstrate that our proposed method significantly outperforms existing data re-balancing methods and can boost the performance of existing state-of-the-art methods.

Keywords

Cite

@article{arxiv.2206.08802,
  title  = {Open-Sampling: Exploring Out-of-Distribution data for Re-balancing Long-tailed datasets},
  author = {Hongxin Wei and Lue Tao and Renchunzi Xie and Lei Feng and Bo An},
  journal= {arXiv preprint arXiv:2206.08802},
  year   = {2022}
}

Comments

Accepted by ICML 2022

R2 v1 2026-06-24T11:55:09.936Z