Transfer Learning Approach to Classify the X-Ray Image that Corresponds to Corona Disease Using ResNet50 Pre-Trained by ChexNet
- 1 Electrical and Computer Engineering, University of Louisville, Louisville, KY, USA
Abstract
The COVID-19 pandemic has had a widespread negative impact globally. It shares symptoms with other respiratory illnesses such as pneumonia and influenza, making rapid and accurate diagnosis essential to treat individuals and halt further transmission. X-ray imaging of the lungs is one of the most reliable diagnostic tools. Utilizing deep learning, we can train models to recognize the signs of infection, thus aiding in the identification of COVID-19 cases. For our project, we developed a deep learning model utilizing the ResNet50 architecture, pre-trained with ImageNet and CheXNet datasets. We tackled the challenge of an imbalanced dataset, the CoronaHack Chest X-Ray dataset provided by Kaggle, through both binary and multi-class classification approaches. Additionally, we evaluated the performance impact of using Focal loss versus Cross-entropy loss in our model.
- Kaggle (2019) CoronaHack-Chest X-Ray-Dataset. https://www.kaggle.com/datasets/praveengovi/coronahack-chest-xraydataset
- Bolhassani, M. (2021) Fully Supervised and Semi-Supervised Semantic Segmentation of cardiac MR using deep learning. Master’s Thesis, Thstanbul Technical University, İstanbul. http://hdl.handle.net/11527/20099
- He, K.M., Zhang, X.Y., Ren, S.Q. and Sun, J. (2016) Deep Residual Learning for Image Recognition. Proceedings of the IEEE Conference on Computer Vision and P attern Recognition , Las Vegas, NV, USA, 27-30 June 2016. https://doi.org/10.1109/CVPR.2016.90
- Huang, G., Liu, Z., Van Der Maaten, L., et al . (2017) Densely Connected Convolutional Networks. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition , Honolulu, HI, USA, 21-26 July 2017.
- Lin, T.-Y., Goyal, P., Girshick, R., et al . (2017) Focal Loss for Dense Object Detection. Proceedings of the IEEE Confe rence on Computer Vision and Pattern Reco g nition , Venice, Italy, 22-29 October 2017.
- Mangal, A., Kalia, S., Rajgopal, H., et al . (2020) CovidAID: COVID-19 Detection Using Chest X-ray https://arxiv.org/abs/2004.09803
- Rajpurkar, P., et al . (2017) Chexnet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learning. https://arxiv.org/abs/1711.05225