Influence of augmentation on the performance of the double ResNet-based model for chest X-ray classification.

Purpose: A pandemic disease elicited by the SARS-CoV-2 virus has become a serious health issue due to infecting millions of people all over the world. Recent publications prove that artificial intelligence (AI) can be used for medical diagnosis purposes, including interpretation of X-ray images. X-ray scanning is relatively cheap, and scan processing is not computationally demanding. Material and methods: In our experiment a baseline transfer learning schema of processing of lung X-ray images, including augmentation, in order to detect COVID-19 symptoms was implemented. Seven different scenarios of augmentation were proposed. The model was trained on a dataset consisting of more than 30,000 X-ray images. Results: The obtained model was evaluated using real images from a Polish hospital, with the use of standard metrics, and it achieved accuracy = 0.9839, precision = 0.9697, recall = 1.0000, and F1-score = 0.9846. Conclusions: Our experiment proved that augmentations and masking could be important steps of data pre-processing and could contribute to improvement of the evaluation metrics. Because medical professionals often tend to lack confidence in AI-based tools, we have designed the proposed model so that its results would be explainable and could play a supporting role for radiology specialists in their work.

A Novel Lightweight Approach to COVID-19 Diagnostics Based on Chest X-ray Images.

BACKGROUND: This paper presents a novel lightweight approach based on machine learning methods supporting COVID-19 diagnostics based on X-ray images. The presented schema offers effective and quick diagnosis of COVID-19. METHODS: Real data (X-ray images) from hospital patients were used in this study. All labels, namely those that were COVID-19 positive and negative, were confirmed by a PCR test. Feature extraction was performed using a convolutional neural network, and the subsequent classification of samples used Random Forest, XGBoost, LightGBM and CatBoost. RESULTS: The LightGBM model was the most effective in classifying patients on the basis of features extracted from X-ray images, with an accuracy of 1.00, a precision of 1.00, a recall of 1.00 and an F1-score of 1.00. CONCLUSION: The proposed schema can potentially be used as a support for radiologists to improve the diagnostic process. The presented approach is efficient and fast. Moreover, it is not excessively complex computationally.