Skip to main content
. 2022 Apr 19;34(16):14003–14012. doi: 10.1007/s00521-022-07250-0

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