Comparison Of CNN, Resnet 50, And Vgg 16 For Pneumonia Classification Using Transfer Learning
Article Sidebar
Main Article Content
Abstract
Pneumonia is one of the leading causes of death from infectious diseases worldwide, making rapid and accurate radiological diagnosis crucial for successful medical treatment. This study implemented and compared three deep learning architectures—a custom Convolutional Neural Network (CNN), ResNet50, and VGG16—for binary classification of chest X-ray images into Normal and Pneumonia categories. The Chest X-ray Pneumonia dataset from Kaggle (5,863 images) was used with an 80/10/10 (train/validation/test) data split and data augmentation to address class imbalance. ResNet50 with transfer learning from ImageNet weights achieved the best performance: 95.1% accuracy, 92.3% precision, 96.7% recall, 94.4% F1-score, and 97.5% AUC-ROC, outperforming the custom CNN (89.4% accuracy, 95.2% AUC) and VGG16 (93.7% accuracy, 96.1% AUC). Statistical analysis confirmed that the performance difference between ResNet50 and the custom CNN was statistically significant (p < 0.05). The results showed that residual learning on ResNet50 effectively addressed the vanishing gradient problem in deep networks and achieved clinically relevant classification accuracy, supporting its potential integration into computer-aided diagnosis (CAD) systems.
Downloads
Article Details
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
References
Hanum, F., Budiman, A., & Senubekti, M. (2022). Deep Learning untuk Deteksi Pneumonia dari Citra Rontgen Dada Menggunakan VGG16. Nuansa Informatika, 16(1), 42–50. https://doi.org/10.25134/ilkom.v16i1.xxx
He, K., Zhang, X., Ren, S., & Sun, J. (2016). Deep Residual Learning for Image Recognition. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 770–778. https://doi.org/10.1109/CVPR.2016.90
Kermany, D. S., Goldbaum, M., Cai, W., Valentim, C. C. S., Liang, H., & Zhang, K. (2018). Identifying Medical Diagnoses and Treatable Diseases by Image-Based Deep Learning. Cell, 172(5), 1122–1131. https://doi.org/10.1016/j.cell.2018.02.010
Krizhevsky, A., Sutskever, I., & Hinton, G. E. (2012). ImageNet Classification with Deep Convolutional Neural Networks. Advances in Neural Information Processing Systems (NeurIPS), 25, 1097–1105.
LeCun, Y., Bengio, Y., & Hinton, G. (2015). Deep Learning. Nature, 521(7553), 436–444. https://doi.org/10.1038/nature14539
Litjens, G., Kooi, T., Bejnordi, B. E., Setio, A. A. A., Ciompi, F., Ghafoorian, M., van der Laak, J. A. W. M., van Ginneken, B., & Sánchez, C. I. (2017). A Survey on Deep Learning in Medical Image Analysis. Medical Image Analysis, 42, 60–88. https://doi.org/10.1016/j.media.2017.07.005
Rajaraman, S., Candemir, S., Kim, I., Thoma, G., & Antani, S. (2018). Visualization and Interpretation of Convolutional Neural Network Predictions in Detecting Pneumonia in Pediatric Chest Radiographs. Applied Sciences, 8(10), 1715. https://doi.org/10.3390/app8101715
Rajpurkar, P., Irvin, J., Ball, R. L., Zhu, K., Yang, B., Mehta, H., Duan, T., Ding, D., Bagul, A., Langlotz, C., Shpanskaya, K., Lungren, M. P., & Ng, A. Y. (2017). CheXNet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learning. arXiv preprint arXiv:1711.05225.
Simonyan, K., & Zisserman, A. (2014). Very Deep Convolutional Networks for Large-Scale Image Recognition. arXiv preprint arXiv:1409.1556.
Tajbakhsh, N., Shin, J. Y., Gurudu, S. R., Hurst, R. T., Kendall, C. B., Gotway, M. B., & Liang, J. (2016). Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning? IEEE Transactions on Medical Imaging, 35(5), 1299–1312. https://doi.org/10.1109/TMI.2016.2302
World Health Organization. (2019). Pneumonia. https://www.who.int/news-room/fact-sheets/detail/pneumonia