Abstract:Medical image datasets, particularly those comprising Magnetic Resonance (MR) images, are essential for accurate diagnosis and treatment planning. However, these datasets often suffer from class imbalance, where certain classes of abnormalities have unequal representation. Models trained on imbalanced datasets can be biased towards the prominent class, leading to misclassification. Addressing class imbalance problems is crucial to developing robust deep-learning MR image analysis models. This research focuses on the class imbalance problem in MR image datasets and proposes a novel approach to enhance deep learning models. We have introduced a unified approach equipped with a selective attention mechanism, unified loss function, and progressive resizing. The selective attention strategy identifies prominent regions within the underlying image to find the feature maps, retaining only the relevant activations of the minority class. Fine-tuning of the multiple hyperparameters was achieved using a novel unified loss function that plays a vital role in enhancing the overwhelming error performance for minority classes and accuracy for common classes. To address the class imbalances phenomenon, we incorporate progressive resizing that can dynamically adjust the input image size as the model trains. This dynamic nature helps handle class imbalances and improve overall performance. The research evaluates the effectiveness of the proposed approach by embedding it into five state-of-the-art CNN models: UNet, FCN, RCNN, SegNet, and Deeplab-V3. For experimental purposes, we have selected five diverse MR image datasets, BUS2017, MICCAI 2015 head and neck, ATLAS, BRATS 2015, and Digital Database Thyroid Image (DDTI), to evaluate the performance of the proposed approach against state-of-the-art techniques. The assessment of the proposed approach reveals improved performance across all metrics for five different MR imaging datasets. DeepLab-V3 demonstrated the best performance, achieving IoU, DSC, Precision, and Recall scores of 0.893, 0.953, 0.943, and 0.944, respectively, on the BUS dataset. These scores indicate an improvement of 5% in DSC, 6% in IoU, 4% in precision, and approximately 4% in recall compared to the baseline. The most significant increases were observed in the ATLAS and LiTS MICCAI 2017 datasets, with a 5% and 7% increase in IoU and DSC over the baseline (DSC = 0.628, DSC = 0.695) for the ATLAS dataset and a 5% and 9% increase in IoU and DSC for the LiTS MICCAI 2017 dataset.

Assessing and mitigating the effects of class imbalance in machine learning with application to X-ray imaging

Class imbalance on medical image classification: towards better evaluation practices for discrimination and calibration performance

Machine Learning and Bias in Medical Imaging: Opportunities and Challenges

Gender imbalance in medical imaging datasets produces biased classifiers for computer-aided diagnosis

Handling imbalanced medical datasets: review of a decade of research

Bias Assessment and Data Drift Detection in Medical Image Analysis: A Survey

Comparison of sampling methods for handling imbalance data in deep learning-based predictions of chest X-ray abnormality tags

The Limits of Fair Medical Imaging AI In The Wild

Training neural network classifiers for medical decision making: The effects of imbalanced datasets on classification performance

Evaluating the Fairness of Neural Collapse in Medical Image Classification

MedISure: Towards Assuring Machine Learning-based Medical Image Classifiers using Mixup Boundary Analysis

Imbalanced Data Classification:A Survey and Experiments in Medical Domain

Addressing Imbalance for Class Incremental Learning in Medical Image Classification

Demonstrating The Risk of Imbalanced Datasets in Chest X-ray Image-based Diagnostics by Prototypical Relevance Propagation

Overlap-Based Undersampling Method for Classification of Imbalanced Medical Datasets

Handling imbalanced medical image data: A deep-learning-based one-class classification approach

Effects of Class Imbalance Countermeasures on Interpretability

Mitigating Calibration Bias Without Fixed Attribute Grouping for Improved Fairness in Medical Imaging Analysis

Bias and Class Imbalance in Oncologic Data—Towards Inclusive and Transferrable AI in Large Scale Oncology Data Sets

Interpretable ML for Imbalanced Data

A Unified Approach Addressing Class Imbalance in Magnetic Resonance Image for Deep Learning Models