Table 6 Ablation Study
From: Vi-GeN 1.0: GAN-Augmented Vision Transformer Pipeline for Diversified Pneumonia Classification
Model Configuration | Augmentation | Classifier | AUC-ROC | Sensitivity(%) | Specificity(%) |
|---|---|---|---|---|---|
Baseline CNN | None | CNN | 0.922 | 90.5 | 91.2 |
ViT(without Augmentation) | None | VIT | 0.961 | 95.2 | 93.8 |
CNN+WGAN-GP | WGAN-GP | CNN | 0.945 | 93.4 | 92.5 |
Vi-GeN 1.0(ViT+WGAN-GP) | WGAN-GP | VIT | 0.985 | 98.1 | 96.5 |