Artificial Intelligence for Automatic Detection and Classification Disease on the X-Ray Images

Liora Mayats-Alpay
2023-08-28
Abstract:Detecting and classifying diseases using X-ray images is one of the more challenging core tasks in the medical and research world. Due to the recent high interest in radiological images and AI, early detection of diseases in X-ray images has become notably more essential to prevent further spreading and flatten the curve. Innovations and revolutions of Computer Vision with Deep learning methods offer great promise for fast and accurate diagnosis of screening and detection from chest X-ray images (CXR). This work presents rapid detection of diseases in the lung using the efficient Deep learning pre-trained RepVGG algorithm for deep feature extraction and classification. We used X-ray images as an example to show the model's efficiency. To perform this task, we classify X-Ray images into Covid-19, Pneumonia, and Normal X-Ray images. Employ ROI object to improve the detection accuracy for lung extraction, followed by data pre-processing and augmentation. We are applying Artificial Intelligence technology for automatic highlighted detection of affected areas of people's lungs. Based on the X-Ray images, an algorithm was developed that classifies X-Ray images with height accuracy and power faster thanks to the architecture transformation of the model. We compared deep learning frameworks' accuracy and detection of disease. The study shows the high power of deep learning methods for X-ray images based on COVID-19 detection utilizing chest X-rays. The proposed framework offers better diagnostic accuracy by comparing popular deep learning models, i.e., VGG, ResNet50, inceptionV3, DenseNet, and InceptionResnetV2.
Image and Video Processing,Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the automatic detection and classification of diseases in X - ray images. Specifically, the paper focuses on using deep - learning techniques, especially the pre - trained RepVGG algorithm, to achieve rapid detection and classification of lung diseases. The main objectives of the paper include: 1. **Improve detection speed and accuracy**: By using an efficient deep - learning model (such as RepVGG), the paper aims to improve the speed and accuracy of detecting and classifying diseases from chest X - ray images. 2. **Automatically segment and locate the lung area**: The paper proposes a method that uses Histogram of Oriented Gradients (HOG) for Region of Interest (ROI) segmentation to automatically extract the lung area and uses Grad - CAM technology to highlight the affected area. 3. **Multi - class classification**: The paper classifies X - ray images into three categories: COVID - 19, pneumonia, and normal (healthy) cases. Through this method, different types of lung diseases can be diagnosed more accurately. 4. **Model optimization and comparison**: The paper not only proposes a method using the RepVGG model but also compares it with other popular deep - learning models (such as VGG, ResNet50, InceptionV3, DenseNet, and InceptionResNetV2), demonstrating the performance advantages of RepVGG. ### Main contributions of the paper - **Efficient feature extraction and classification**: By using the RepVGG model, the paper achieves efficient and accurate feature extraction and classification, especially when dealing with large - scale data sets. - **Application of visualization technology**: Using Grad - CAM technology, the paper can highlight the affected areas in X - ray images, thus providing more intuitive diagnosis results. - **Performance evaluation**: Through multiple evaluation metrics such as confusion matrix, classification report, ROC curve, and AUC value, the paper comprehensively evaluates the performance of the proposed model and compares it with existing methods. ### Conclusion The paper shows the great potential of deep - learning methods in the automatic detection and classification of diseases in X - ray images, especially in COVID - 19 detection. By using the RepVGG model and multiple computer vision techniques, the method proposed in the paper shows significant advantages in both accuracy and efficiency. These results are of great significance for early diagnosis and rapid screening in the medical field.