WGAN-GP for Synthetic Retinal Image Generation: Enhancing Sensor-Based Medical Imaging for Classification Models
Accurate synthetic image generation is crucial for addressing data scarcity challenges in medical image classification tasks, particularly in sensor-derived medical imaging. In this work, we propose a novel method using a Wasserstein Generative Adversarial Network with Gradient Penalty (WGAN-GP) and...
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2024-12-01
|
| Series: | Sensors |
| Subjects: | |
| Online Access: | https://www.mdpi.com/1424-8220/25/1/167 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1841548927339331584 |
|---|---|
| author | Héctor Anaya-Sánchez Leopoldo Altamirano-Robles Raquel Díaz-Hernández Saúl Zapotecas-Martínez |
| author_facet | Héctor Anaya-Sánchez Leopoldo Altamirano-Robles Raquel Díaz-Hernández Saúl Zapotecas-Martínez |
| author_sort | Héctor Anaya-Sánchez |
| collection | DOAJ |
| description | Accurate synthetic image generation is crucial for addressing data scarcity challenges in medical image classification tasks, particularly in sensor-derived medical imaging. In this work, we propose a novel method using a Wasserstein Generative Adversarial Network with Gradient Penalty (WGAN-GP) and nearest-neighbor interpolation to generate high-quality synthetic images for diabetic retinopathy classification. Our approach enhances training datasets by generating realistic retinal images that retain critical pathological features. We evaluated the method across multiple retinal image datasets, including Retinal-Lesions, Fine-Grained Annotated Diabetic Retinopathy (FGADR), Indian Diabetic Retinopathy Image Dataset (IDRiD), and the Kaggle Diabetic Retinopathy dataset. The proposed method outperformed traditional generative models, such as conditional GANs and PathoGAN, achieving the best performance on key metrics: a Fréchet Inception Distance (FID) of 15.21, a Mean Squared Error (MSE) of 0.002025, and a Structural Similarity Index (SSIM) of 0.89 in the Kaggle dataset. Additionally, expert evaluations revealed that only 56.66% of synthetic images could be distinguished from real ones, demonstrating the high fidelity and clinical relevance of the generated data. These results highlight the effectiveness of our approach in improving medical image classification by generating realistic and diverse synthetic datasets. |
| format | Article |
| id | doaj-art-77243ffa1c554931a25bcfad3d390004 |
| institution | Kabale University |
| issn | 1424-8220 |
| language | English |
| publishDate | 2024-12-01 |
| publisher | MDPI AG |
| record_format | Article |
| series | Sensors |
| spelling | doaj-art-77243ffa1c554931a25bcfad3d3900042025-01-10T13:21:06ZengMDPI AGSensors1424-82202024-12-0125116710.3390/s25010167WGAN-GP for Synthetic Retinal Image Generation: Enhancing Sensor-Based Medical Imaging for Classification ModelsHéctor Anaya-Sánchez0Leopoldo Altamirano-Robles1Raquel Díaz-Hernández2Saúl Zapotecas-Martínez3Computer Science Department, Instituto Nacional de Astrofísica Óptica y Electrónica, Luis Enrrique Erro No. 1, Sta. María Tonantzintla, Puebla 72840, MexicoComputer Science Department, Instituto Nacional de Astrofísica Óptica y Electrónica, Luis Enrrique Erro No. 1, Sta. María Tonantzintla, Puebla 72840, Mexico Optics Department, Instituto Nacional de Astrofísica Óptica y Electrónica, Luis Enrrique Erro No. 1, Sta. María Tonantzintla, Puebla 72840, MexicoComputer Science Department, Instituto Nacional de Astrofísica Óptica y Electrónica, Luis Enrrique Erro No. 1, Sta. María Tonantzintla, Puebla 72840, MexicoAccurate synthetic image generation is crucial for addressing data scarcity challenges in medical image classification tasks, particularly in sensor-derived medical imaging. In this work, we propose a novel method using a Wasserstein Generative Adversarial Network with Gradient Penalty (WGAN-GP) and nearest-neighbor interpolation to generate high-quality synthetic images for diabetic retinopathy classification. Our approach enhances training datasets by generating realistic retinal images that retain critical pathological features. We evaluated the method across multiple retinal image datasets, including Retinal-Lesions, Fine-Grained Annotated Diabetic Retinopathy (FGADR), Indian Diabetic Retinopathy Image Dataset (IDRiD), and the Kaggle Diabetic Retinopathy dataset. The proposed method outperformed traditional generative models, such as conditional GANs and PathoGAN, achieving the best performance on key metrics: a Fréchet Inception Distance (FID) of 15.21, a Mean Squared Error (MSE) of 0.002025, and a Structural Similarity Index (SSIM) of 0.89 in the Kaggle dataset. Additionally, expert evaluations revealed that only 56.66% of synthetic images could be distinguished from real ones, demonstrating the high fidelity and clinical relevance of the generated data. These results highlight the effectiveness of our approach in improving medical image classification by generating realistic and diverse synthetic datasets.https://www.mdpi.com/1424-8220/25/1/167generative AIGANsAImachine learningretinal images |
| spellingShingle | Héctor Anaya-Sánchez Leopoldo Altamirano-Robles Raquel Díaz-Hernández Saúl Zapotecas-Martínez WGAN-GP for Synthetic Retinal Image Generation: Enhancing Sensor-Based Medical Imaging for Classification Models Sensors generative AI GANs AI machine learning retinal images |
| title | WGAN-GP for Synthetic Retinal Image Generation: Enhancing Sensor-Based Medical Imaging for Classification Models |
| title_full | WGAN-GP for Synthetic Retinal Image Generation: Enhancing Sensor-Based Medical Imaging for Classification Models |
| title_fullStr | WGAN-GP for Synthetic Retinal Image Generation: Enhancing Sensor-Based Medical Imaging for Classification Models |
| title_full_unstemmed | WGAN-GP for Synthetic Retinal Image Generation: Enhancing Sensor-Based Medical Imaging for Classification Models |
| title_short | WGAN-GP for Synthetic Retinal Image Generation: Enhancing Sensor-Based Medical Imaging for Classification Models |
| title_sort | wgan gp for synthetic retinal image generation enhancing sensor based medical imaging for classification models |
| topic | generative AI GANs AI machine learning retinal images |
| url | https://www.mdpi.com/1424-8220/25/1/167 |
| work_keys_str_mv | AT hectoranayasanchez wgangpforsyntheticretinalimagegenerationenhancingsensorbasedmedicalimagingforclassificationmodels AT leopoldoaltamiranorobles wgangpforsyntheticretinalimagegenerationenhancingsensorbasedmedicalimagingforclassificationmodels AT raqueldiazhernandez wgangpforsyntheticretinalimagegenerationenhancingsensorbasedmedicalimagingforclassificationmodels AT saulzapotecasmartinez wgangpforsyntheticretinalimagegenerationenhancingsensorbasedmedicalimagingforclassificationmodels |