Advancing Diagnostic Precision: Leveraging Machine Learning Techniques for Accurate Detection of Covid-19, Pneumonia, and Tuberculosis in Chest X-Ray Images

Aditya Kulkarni,Guruprasad Parasnis,Harish Balasubramanian,Vansh Jain,Anmol Chokshi,Reena Sonkusare
2023-10-10
Abstract:Lung diseases such as COVID-19, tuberculosis (TB), and pneumonia continue to be serious global health concerns that affect millions of people worldwide. In medical practice, chest X-ray examinations have emerged as the norm for diagnosing diseases, particularly chest infections such as COVID-19. Paramedics and scientists are working intensively to create a reliable and precise approach for early-stage COVID-19 diagnosis in order to save lives. But with a variety of symptoms, medical diagnosis of these disorders poses special difficulties. It is essential to address their identification and timely diagnosis in order to successfully treat and prevent these illnesses. In this research, a multiclass classification approach using state-of-the-art methods for deep learning and image processing is proposed. This method takes into account the robustness and efficiency of the system in order to increase diagnostic precision of chest diseases. A comparison between a brand-new convolution neural network (CNN) and several transfer learning pre-trained models including VGG19, ResNet, DenseNet, EfficientNet, and InceptionNet is recommended. Publicly available and widely used research datasets like Shenzen, Montogomery, the multiclass Kaggle dataset and the NIH dataset were used to rigorously test the model. Recall, precision, F1-score, and Area Under Curve (AUC) score are used to evaluate and compare the performance of the proposed model. An AUC value of 0.95 for COVID-19, 0.99 for TB, and 0.98 for pneumonia is obtained using the proposed network. Recall and precision ratings of 0.95, 0.98, and 0.97, respectively, likewise met high standards.
Image and Video Processing,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The main problem that this paper attempts to solve is to improve the diagnostic accuracy of lung diseases (such as COVID - 19, pneumonia and tuberculosis) in chest X - ray images. Specifically, the research aims to develop a multi - class classification model capable of accurately detecting these diseases by using advanced machine - learning techniques, especially deep - learning and image - processing methods. This is not only to improve the accuracy of early diagnosis, but also to provide a more efficient and reliable diagnostic tool to address these global health challenges. ### Main Objectives 1. **Improve Diagnostic Accuracy**: By proposing a new convolutional neural network (CNN) model and comparing it with existing pre - trained models (such as VGG19, ResNet, DenseNet, EfficientNet and InceptionNet), verify its performance in multi - class classification tasks. 2. **Dataset Evaluation**: Use multiple publicly available datasets (such as Shenzen, Montgomery, Kaggle multi - class dataset and NIH dataset) to rigorously test the model. 3. **Performance Evaluation**: Comprehensively evaluate and compare the performance of the proposed model with other models through metrics such as Recall, Precision, F1 - Score and Area Under the Curve (AUC). ### Specific Problems - **Multi - class Classification**: How to accurately distinguish COVID - 19, pneumonia and tuberculosis in chest X - ray images. - **Data Pre - processing**: How to improve image quality through techniques such as data augmentation, histogram equalization, Local Ternary Pattern (LTP) and adaptive thresholding, thereby enhancing the model's recognition ability. - **Model Optimization**: How to design and optimize the convolutional neural network structure to improve the robustness and efficiency of the model. - **Performance Comparison**: How to prove the superiority of the proposed model on different datasets through multiple performance metrics. ### Solutions - **Data Pre - processing**: Use techniques such as data augmentation, histogram equalization, LTP and adaptive thresholding to improve image quality. - **Model Design**: Propose a new CNN architecture, including an initial convolutional layer, a custom Inception - like module, further convolutional layers and a final classification layer. - **Performance Evaluation**: Verify the performance of the model on multiple datasets through multiple performance metrics (such as AUC, F1 - Score, etc.). ### Results - **Multi - class Dataset**: On the multi - class dataset, the proposed model has AUC values of 0.95 (COVID - 19), 0.99 (tuberculosis) and 0.98 (pneumonia) for all classes, showing excellent performance. - **Disease - specific Dataset**: On the tuberculosis dataset, the model has an AUC value of 0.99, showing excellent performance. - **Benchmark Dataset**: On the NIH dataset, the model has an AUC value of 0.889, demonstrating its competitiveness on large - scale datasets. In conclusion, this paper successfully improves the diagnostic accuracy of lung diseases in chest X - ray images by proposing a new CNN model and combining multiple data pre - processing techniques, providing strong support for medical diagnosis.