English

Imbalanced Classification via Explicit Gradient Learning From Augmented Data

Machine Learning 2022-11-13 v2

Abstract

Learning from imbalanced data is one of the most significant challenges in real-world classification tasks. In such cases, neural networks performance is substantially impaired due to preference towards the majority class. Existing approaches attempt to eliminate the bias through data re-sampling or re-weighting the loss in the learning process. Still, these methods tend to overfit the minority samples and perform poorly when the structure of the minority class is highly irregular. Here, we propose a novel deep meta-learning technique to augment a given imbalanced dataset with new minority instances. These additional data are incorporated in the classifier's deep-learning process, and their contributions are learned explicitly. The advantage of the proposed method is demonstrated on synthetic and real-world datasets with various imbalance ratios.

Keywords

Cite

@article{arxiv.2202.10550,
  title  = {Imbalanced Classification via Explicit Gradient Learning From Augmented Data},
  author = {Bronislav Yasinnik and Moshe Salhov and Ofir Lindenbaum and Amir Averbuch},
  journal= {arXiv preprint arXiv:2202.10550},
  year   = {2022}
}

Comments

arXiv admin note: text overlap with arXiv:1906.05591, arXiv:2202.09318