Margin-Aware Adaptive-Weighted-Loss for Deep Learning Based Imbalanced Data Classification

In supervised learning algorithms, the class imbalance problem often leads to generating results biased towards the majority classes. Present methods used to deal with the class imbalance problem ignore a principal aspect of separating the overlapping classes. This is the reason why most of these methods are prone to overfit on the training data. To this end, we propose a novel loss function, namely margin-aware adaptive-weighted loss. Here, we first use the large margin softmax to leverage intraclass compactness and interclass separability. Further to learn an unbiased representation of the classes, we put forward a dynamically weighted loss for imbalanced data classification. This weight dynamically adapts on every minibatch based on the inverse class frequencies. In addition, it takes care of the hard-to-train samples by using the confidence scores to learn discriminative hidden representations of the data. The overall framework is found to be effective when evaluated on the following two widely used datasets: 1) Canadian Institute for Advanced Research (CIFAR)-10 and 2) Fashion-MNIST. Additional experiments on human against machine and Asia Pacific tele-ophthalmology society 2019 blindness detection datasets prove the robustness of our methodology. © 2023 IEEE.