AUBADE-syn: a novel deep learning ensemble method for glaucoma detection using synthetic fundus images on imbalanced datasets

· Source: Machine learning : nature.com subject feeds · Field: Health & Wellbeing — Medical Devices & Health Technology, Clinical Care & Medical Practice, Artificial Intelligence & Machine Learning · Depth: Expert, quick

Summary

AUBADE-syn is a new deep learning ensemble framework designed for glaucoma detection, specifically addressing class imbalance in medical imaging datasets. This method integrates synthetic image generation with structured class-balancing strategies. It utilizes optic nerve head-centered regions and a classifier-free guided diffusion model to create realistic glaucomatous images, thereby enriching the minority class and enhancing model generalization. Benchmarked against methods like weighted loss functions, focal loss, Balanced-MixUp, ProCo, and FlexDA, AUBADE-syn achieved an area under the receiver operating characteristic curve of 0.992 on the EyePACS dataset, which has a 1:30 class imbalance. The framework also demonstrated top-tier or competitive performance across ten independent public datasets and after fine-tuning on three additional public datasets, consistently improving discrimination and calibration for glaucoma detection in highly imbalanced settings.

Key takeaway

For AI Scientists developing diagnostic tools for imbalanced medical imaging, AUBADE-syn offers a robust approach. You should consider integrating synthetic image generation, particularly using diffusion models focused on relevant anatomical features, with ensemble learning to significantly improve model performance and calibration. This strategy can overcome common challenges with scarce minority class data, leading to more reliable diagnostic systems.

Key insights

AUBADE-syn improves glaucoma detection on imbalanced datasets via synthetic image generation and ensemble learning.

Principles

Method

AUBADE-syn uses optic nerve head-centered regions and a classifier-free guided diffusion model to generate synthetic glaucomatous images, then integrates these with structured class-balancing strategies within a deep learning ensemble framework.

In practice

Topics

Best for: AI Scientist, Research Scientist, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Machine learning : nature.com subject feeds.