Sorry, you need to enable JavaScript to visit this website.

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:
 

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.

up
0 users have voted: