X-Ray Machine Learning Classification with VGG-16 for Feature Extraction

The Covid-19 pandemic evidenced the need Computer Aided Diagnostic Systems to analyze medical images, such as CT and MRI scans and X-rays, to assist specialists in disease diagnosis. CAD systems have been shown to be effective at detecting COVID-19 in chest X-ray and CT images, with some studies reporting high levels of accuracy and sensitivity. Moreover, it can also detect some diseases in patients who may not have symptoms, preventing the spread of the virus. There are some types of CAD systems, such as Machine and Deep Learning-based and Transfer learning-based. This chapter proposes a pipeline for feature extraction and classification of Covid-19 in X-ray images using transfer learning for feature extraction with VGG-16 CNN and machine learning classifiers. Five classifiers were evaluated: Accuracy, Specificity, Sensitivity, Geometric mean, and Area under the curve. The SVM Classifier presented the best performance metrics for Covid-19 classification, achieving 90% accuracy, 97.5% of Specificity, 82.5% of Sensitivity, 89.6% of Geometric mean, and 90% for the AUC metric. On the other hand, the Nearest Centroid (NC) classifier presented poor sensitivity and geometric mean results, achieving 33.9% and 54.07%, respectively.

DOI: 10.1007/978-3-031-30788-1_5