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.

MCICT: Graph convolutional network-based end-to-end model for multi-label classification of imbalanced clinical text

Multi-label Classification for Clinical Text with Feature-level Attention

Multi-label text classification based on semantic-sensitive graph convolutional network

TLMCM Network for Medical Image Hierarchical Multi-Label Classification

An Improved Double Channel Long Short-Term Memory Model for Medical Text Classification

Multi-instance Multi-task Learning for Joint Clinical Outcome and Genomic Profile Predictions from the Histopathological Images

Medical Text Classification Using Hybrid Deep Learning Models with Multihead Attention

CTransCNN: Combining transformer and CNN in multilabel medical image classification

Clinical text classification with rule-based features and knowledge-guided convolutional neural networks

Triplet attention and dual-pool contrastive learning for clinic-driven multi-label medical image classification

Clinical multi-label free text classification by exploiting disease label relation

A model of integrating convolution and BiGRU dual-channel mechanism for Chinese medical text classifications

Medical Text Classification using Convolutional Neural Networks

Seeing The Whole Patient: Using Multi-Label Medical Text Classification Techniques to Enhance Predictions of Medical Codes

A clinical text classification paradigm using weak supervision and deep representation

Multi-label classification of fundus images with graph convolutional network and LightGBM

MFLSCI: Multi-granularity fusion and label semantic correlation information for multi-label legal text classification

Coupled Graph Convolutional Neural Networks for Text-Oriented Clinical Diagnosis Inference

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

An efficient medical image classification network based on multi-branch CNN, token grouping Transformer and mixer MLP