Table 1.
summarizes the parameters used in neural network architectures employed
| Model | Epochs | Learning rate | Optimizer | Loss function | Parameters (in millions) |
|---|---|---|---|---|---|
| Custom CNN | 40 | 1e-3 to 1e-4 | Adam | Sparse categorical cross entropy |
2.2 (X-ray) 3.6 (CT) |
| EfficientNetB0 | 40 | 1e-4 | Adam | Binary cross entropy | 5.3 |
| VGG-16 | 40 | 1e-4 | Adam | Binary cross entropy | 138 |
| ResNet-50 | 40 | 1e-4 | Adam | Binary cross entropy | 25.6 |