Skip to content
GCC AI Research

DGM-DR: Domain Generalization with Mutual Information Regularized Diabetic Retinopathy Classification

arXiv · · Significant research

Summary

This paper introduces a domain generalization (DG) method for Diabetic Retinopathy (DR) classification that maximizes mutual information using a large pretrained model. The method aims to address the challenge of domain shift in medical imaging caused by variations in data acquisition. Experiments on public datasets demonstrate that the proposed method outperforms state-of-the-art techniques, achieving a 5.25% improvement in average accuracy.

Get the weekly digest

Top AI stories from the GCC region, every week.

Related

ConDiSR: Contrastive Disentanglement and Style Regularization for Single Domain Generalization

arXiv ·

This paper introduces a new Single Domain Generalization (SDG) method called ConDiSR for medical image classification, using channel-wise contrastive disentanglement and reconstruction-based style regularization. The method is evaluated on multicenter histopathology image classification, achieving a 1% improvement in average accuracy compared to state-of-the-art SDG baselines. Code is available at https://github.com/BioMedIA-MBZUAI/ConDiSR.

Designing the Architecture of a Convolutional Neural Network Automatically for Diabetic Retinopathy Diagnosis

arXiv ·

This paper introduces a method for automatically designing convolutional neural network (CNN) architectures tailored for diabetic retinopathy (DR) diagnosis using fundus images. The approach uses k-medoid clustering, PCA, and inter/intra-class variations to optimize CNN depth and width. Validated on datasets including a local Saudi dataset and Kaggle benchmarks, the custom-designed models outperform pre-trained CNNs with fewer parameters.

Interpretable and synergistic deep learning for visual explanation and statistical estimations of segmentation of disease features from medical images

arXiv ·

The study compares deep learning models trained via transfer learning from ImageNet (TII-models) against those trained solely on medical images (LMI-models) for disease segmentation. Results show that combining outputs from both model types can improve segmentation performance by up to 10% in certain scenarios. A repository of models, code, and over 10,000 medical images is available on GitHub to facilitate further research.

MedMerge: Merging Models for Effective Transfer Learning to Medical Imaging Tasks

arXiv ·

Researchers at MBZUAI have introduced MedMerge, a transfer learning technique that merges weights from independently initialized models to improve performance on medical imaging tasks. MedMerge learns kernel-level weights to combine features from different models into a single model. Experiments across various medical imaging tasks demonstrated performance gains of up to 7% in F1 score.