Smartphone Capable Lightweight Convolutional Neural Network Model for Detecting COVID-19 in Chest X-rays: Addressing the need of Resource-strapped Locations
Ayan Nair (Mentors: Professor Adam Alessio & Wenjie Qi, Michigan State University)
- Abstract
- COVID-19 has been a global health crisis for the past year, and scientists are still finding ways of preventing the spread of the virus as well as detecting it. One way of testing COVID-19 is with the use of chest radiographs. You take an x-ray, give it to a radiologist, and get your results within 24 hours. However, many hospitals do not have a radiologist on hand, making it harder for hospitals to give accurate and timely results to patients. An article from The Atlantic stated that “There are more radiologists working in the four teaching hospitals on Longwood Avenue in Boston, Massachusetts, than there are in West Africa,” with about 130 radiologists in Massachusetts General, Boston, and only 2 working in the country of Liberia. These statistics necessitates an alternative to a radiologist in these resource-strapped locations. Accounting for the rapid rise of deep learning throughout the last decade, this project aims to use a lightweight convolutional neural network to accurately detect COVID-19 from standard chest radiographs. Specifically, a lightweight neural network that can be used on a smartphone, which allows it to be mobile and to be used in locations where a radiologist is not available. My project used two convolutional neural networks; a ResNet model and a lightweight mobile neural network. A ResNet is a fully convolutional neural network that can identify and classify an input image into 1000 object categories. These models are heavy in computational power and are very “deep” (contain a lot of layers). The lightweight mobile neural network, on the other hand, is a highly stripped-down version containing less than 15 layers. The research question my project aims to answer is if a lightweight neural network tool can accurately detect COVID-19 pathologies from chest radiographs. To answer the research question, I first gathered 1345 viral pneumonia chest radiographs provided by Chowdhury et. al. (2020), Rahman et. al. (2020), 1070 images of COVID-19 radiographs from GitHub provided by Cohen et. Al. (2020), and 1358 images of normal chest radiographs obtained from CheXpert (Rajpurkar et. al. 2019), a large public radiograph dataset. Using 3000 chest radiographs from this data, I first trained the ResNet model and used the weights from the training model to test 800 radiographs. The performance of this model (accuracy, sensitivity, specificity) was compared with the performance attained from the lightweight model. The results show that the lightweight model provided very high accuracies, sensitivities, and specificities, which are 98.4%, 97.7%, and 98.7% respectively. This offers potential for using such a smart phone capable lightweight model in resource strapped locations.
- Presented by
- Ayan
- Institution
- Okemos High School
- Keywords
- Convolutional Neural Network, Lightweight, Smartphone capable, Tool, Low cost, X-ray, COVID-19


























