Abstract:With advances in digital technology, the classification of medical images has become a crucial step for image-based clinical decision support systems. Automatic medical image classification represents a pivotal domain where the use of AI holds the potential to create a significant social impact. However, several challenges act as obstacles to the development of practical and effective solutions. One of these challenges is the prevalent class imbalance problem in most medical imaging datasets. As a result, existing AI techniques, particularly deep-learning-based methodologies, often underperform in such scenarios. In this study, we propose a novel framework called Large Margin aware Focal (LMF) loss to mitigate the class imbalance problem in medical imaging. The LMF loss represents a linear combination of two loss functions optimized by two hyperparameters. This framework harnesses the distinct characteristics of both loss functions by enforcing wider margins for minority classes while simultaneously emphasizing challenging samples found in the datasets. We perform rigorous experiments on three neural network architectures and with four medical imaging datasets. We provide empirical evidence that our proposed framework consistently outperforms other baseline methods, showing an improvement of 2%-9% in macro-f1 scores. Through class-wise analysis of f1 scores, we also demonstrate how the proposed framework can significantly improve performance for minority classes. The results of our experiments show that our proposed framework can perform consistently well across different architectures and datasets. Overall, our study demonstrates a simple and effective approach to addressing the class imbalance problem in medical imaging datasets. We hope our work will inspire new research toward a more generalized approach to medical image classification.

Borderline-margin loss based deep metric learning framework for imbalanced data

InterFace:Adjustable Angular Margin Inter-class Loss for Deep Face Recognition

Margin-Aware Adaptive-Weighted-Loss for deep learning based imbalanced data classification

Towards Deeper Insights into Deep Learning from Imbalanced Data.

Geometric imbalanced deep learning with feature scaling and boundary sample mining

LMFLOSS: A Hybrid Loss For Imbalanced Medical Image Classification

An Adaptive Deep Metric Learning Loss Function for Class-Imbalance Learning via Intraclass Diversity and Interclass Distillation

Learning Imbalanced Datasets with Maximum Margin Loss

Deep Learning-Based Imbalanced Classification With Fuzzy Support Vector Machine

Threshold-Consistent Margin Loss for Open-World Deep Metric Learning

Learning Towards The Largest Margins.

Boundary-restricted metric learning

An Adaptive Multi-Class Imbalanced Classification Framework Based on Ensemble Methods and Deep Network

Label correlation guided borderline oversampling for imbalanced multi-label data learning

Iterative Metric Learning for Imbalance Data Classification

Margin Loss Based on Adaptive Metric for Image Recognition

Inter-class Angular Margin Loss for Face Recognition

Learning Deep Landmarks for Imbalanced Classification

Learning in presence of class imbalance and class overlapping by using one-class SVM and undersampling technique

An oversampling FCM-KSMOTE algorithm for imbalanced data classification

A New Safe-Level Enabled Borderline-SMOTE for Condition Recognition of Imbalanced Dataset