Documents
Presentation Slides
Learning Imbalanced Datasets with Maximum Margin Loss
- Citation Author(s):
- Submitted by:
- Haeyong Kang
- Last updated:
- 27 September 2021 - 8:14am
- Document Type:
- Presentation Slides
- Event:
- Presenters:
- Haeyong Kang
- Categories:
- Log in to post comments
A learning algorithm referred to as Maximum Margin (MM) is proposed for considering the class-imbalance data learning issue: the deep model tends to predict the majority classes rather than the minority ones. For better generalization on the minority classes, the proposed Maximum Margin (MM) loss function is newly designed by minimizing a margin-based generalization bound through the shifting decision bound. As a prior study, the theoretically principled label-distributionaware margin (LDAM) loss had been successfully applied with classical strategies such as re-weighting or re-sampling. However, the maximum margin loss function has not been investigated so far. In this study, we evaluate the two types of hard maximum margin-based decision boundary shift with training schedule on artificially imbalanced CIFAR-10/100 and show the effectiveness.