Page 43 - AIH-2-1
P. 43

Artificial Intelligence in Health                            Deep learning on chest X-ray and CT for COVID-19






































            Figure 5. Misclassified X-ray images with prediction details using ResNet, DenseNet, SeResNext, and EfficientNet models. On the top of each subplot,
            “Prediction/Actual/Loss/Probability” details for each individual image are shown.

                         A                                   B















            Figure 6. Sensitivity (A) and specificity (B) of ResNet (34), DenseNet, SeResNext, and EfficientNet for COVID-19, normal, and pneumonia class detection
                                        Note: The number 34 indicates the number of convolution layers).
            leading to more robust and accurate predictions. This   This research article introduces several key innovations in
            enhanced diversity helps mitigate biases and ensures that   the  training  methodology  that  distinguish  it  from  prior
            the model performs well in real-world, heterogeneous   works:
            environments, ultimately improving its reliability and   (1)  Dynamic learning rate optimization. We developed a
            effectiveness in detecting COVID-19, pneumonia, and   novel dynamic learning rate scheduler that adaptively
            normal cases at the global scale and for future pandemics,   identifies the optimal learning rate within carefully
            as well.                                              set boundaries (10  – 10 ). This approach mitigates
                                                                                 −4
                                                                                       −5
                                                                  the risk of getting trapped in local minima, a common
            5. Conclusion                                         challenge in neural network training.
            In this work, we implemented various CNN models with   (2)  Layer-specific discriminative learning. This study
            a transfer learning-based approach to classify COVID-19   implemented a discriminative learning rate strategy,
            and pneumonia from normal patients through chest X-rays.   applying different learning rates (ranging from 0.0001


            Volume 2 Issue 1 (2025)                         37                               doi: 10.36922/aih.2888
   38   39   40   41   42   43   44   45   46   47   48