Enhancing Early Tuberculosis Detection Using CGAN Augmentation and Deep Transfer Learning Models
Tuberculosis (TB) remains a leading infectious disease worldwide, and early, reliable screening using chest X-rays (CXRs) is essential in low-resource settings. The scarcity of labeled TB-positive CXR images limits the effectiveness of deep learning models. This study investigates whether Conditional Generative Adversarial Networks (CGANs) can generate realistic TB-positive CXR images to balance training data and improve the classification performance of fine-tuned deep transfer learning (DTL) models. We trained a CGAN (LSGAN formulation) to synthesize class-conditional grayscale CXR images at 128x128 resolution and used the generated images to augment the Shenzhen TB dataset. Three pre-trained DTL architectures (DenseNet121, VGG16, and MobileNetV3Small) were fine-tuned on both original and CGAN-augmented datasets. Experiments used stratified 70/10/20 train/validation/test splits and a fixed random seed (random_state=42) to ensure reproducibility. Model performance was evaluated using accuracy, precision, recall (sensitivity), F1-score, confusion matrices, and ROC/AUC curves. The experiments were executed on an NVIDIA Tesla P100 GPU (16GB) in a Kaggle runtime environment; total CGAN+classifier processing reported a wall-clock runtime of 39 minutes 30 seconds for the baseline experimental run. CGAN augmentation produced consistent improvements across models: DenseNet121 improved from 93.0% to 94.6% test accuracy, VGG16 improved from 96.3% to 96.8%, and MobileNetV3Small improved from 93.0% to 93.5%. Class-conditional GAN augmentation can modestly but usefully improve DTL classifier performance in TB detection when labeled data are scarce, though further cross-dataset validation is required before clinical deployment.
Authors
- Susan Mwelu (ORCID: https://orcid.org/0009-0005-9570-9112)
- Herbert Imboga (ORCID: https://orcid.org/0009-0003-9963-4977)
- Anthony Waititu (ORCID: https://orcid.org/0000-0003-0268-2968)
- Teresia Kamau (ORCID: https://orcid.org/0009-0004-6363-9976)
Institutions
- Jomo Kenyatta University of Agriculture and Technology (KE)
Publication Details
- Journal
- International Journal of Data Science and Analysis
- Published
- 2025-11-28
- DOI
- https://doi.org/10.11648/j.ijdsa.20251106.14
- Primary Topic
- COVID-19 diagnosis using AI
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- Jomo Kenyatta University of Agriculture and Technology