Class-wise Flooding Regularization for Imbalanced Image Classification

Hiroaki Aizawa, Yuta Naito, Kohei Fukuda

The purpose of training neural networks is to achieve high generalization performance on unseen inputs. However, when trained on imbalanced datasets, a model's prediction tends to…