Abstract:Deep learning models, integral components of contemporary technological landscapes, exhibit enhanced learning capabilities with larger datasets. Traditional data augmentation techniques, while effective in generating new data, have limitations, especially in fields like ocular disease diagnosis. In response, alternative augmentation approaches, including the utilization of generative AI, have emerged. In our study, we employed a diffusion-based model (Stable Diffusion) to synthesize data by faithfully recreating crucial vascular structures in the retina, vital for detecting eye diseases by using the Ocular Disease Intelligent Recognition dataset. Our goal was to augment retinography images for ocular disease diagnosis using diffusion-based models, optimizing the outputs of the fine-tuned Stable Diffusion model, and ensuring the generated data closely resembles real-world scenarios. This strategic approach resulted in improved performance in classification models and augmentation outperformed traditional methods, exhibiting high precision rates ranging from 85% to 76.2% and recall values of 86%, and 75% for 5 classes. Beyond performance enhancement, we demonstrated that the inclusion of synthetic data, coupled with data reduction using the t-SNE method, effectively addressed dataset imbalance. As a result of synthetic data addition, notable increases of 3.4% in the precision metric and 12.8% in the recall metric were observed in the 7-class case. Strategically synthesizing data addressed underrepresented classes, creating a balanced dataset for comprehensive model learning. Surpassing performance improvements, this approach underscores synthetic data's ability to overcome the limitations of traditional methods, particularly in sensitive medical domains like ocular disease diagnosis, ensuring accurate classification. The codes of the study will be shared on GitHub in a way that benefits everyone interested: https://github.com/miralab-ai/generative-data-augmentation.

Generating Progressive Images from Pathological Transitions via Diffusion Model

AnomalyDiffusion: Few-Shot Anomaly Image Generation with Diffusion Model

Comparative Analysis of Diffusion Generative Models in Computational Pathology

PathUp: Patch-wise Timestep Tracking for Multi-class Large Pathology Image Synthesising Diffusion Model

Controllable and Efficient Multi-Class Pathology Nuclei Data Augmentation using Text-Conditioned Diffusion Models

Mask-guided cross-image attention for zero-shot in-silico histopathologic image generation with a diffusion model

Style-Extracting Diffusion Models for Semi-Supervised Histopathology Segmentation

Synthetic Augmentation with Large-scale Unconditional Pre-training

Pathological Evidence Exploration in Deep Retinal Image Diagnosis

Adapted generative latent diffusion models for accurate pathological analysis in chest X-ray images

A Progressive Generative Adversarial Method for Structurally Inadequate Medical Image Data Augmentation

Unpaired Stain Transfer Using Pathology-Consistent Constrained Generative Adversarial Networks

Improving text-conditioned latent diffusion for cancer pathology

An Organism Starts with a Single Pix-Cell: A Neural Cellular Diffusion for High-Resolution Image Synthesis

Boosting Dermatoscopic Lesion Segmentation via Diffusion Models with Visual and Textual Prompts

DiffInfinite: Large Mask-Image Synthesis via Parallel Random Patch Diffusion in Histopathology

Diverse Data Generation for Retinal Layer Segmentation with Potential Structure Modelling

Diffusion-based data augmentation methodology for improved performance in ocular disease diagnosis using retinography images

Spatial Diffusion for Cell Layout Generation

Diffusion Curriculum: Synthetic-to-Real Generative Curriculum Learning via Image-Guided Diffusion

Automatic Generation of Pathological Benchmark Dataset from Hyperspectral Images of Double Stained Tissues