Detection of pulmonary pathologies using convolutional neural networks, Data Augmentation, ResNet50 and Vision Transformers

Pablo Ramirez Amador,Dinarle Milagro Ortega,Arnold Cesarano
2024-09-22
Abstract:Pulmonary diseases are a public health problem that requires accurate and fast diagnostic techniques. In this paper, a method based on convolutional neural networks (CNN), Data Augmentation, ResNet50 and Vision Transformers (ViT) is proposed to detect lung pathologies from medical images. A dataset of X-ray images and CT scans of patients with different lung diseases, such as cancer, pneumonia, tuberculosis and fibrosis, is used. The results obtained by the proposed method are compared with those of other existing methods, using performance metrics such as accuracy, sensitivity, specificity and area under the ROC curve. The results show that the proposed method outperforms the other methods in all metrics, achieving an accuracy of 98% and an area under the ROC curve of 99%. It is concluded that the proposed method is an effective and promising tool for the diagnosis of pulmonary pathologies by medical imaging.
Image and Video Processing,Artificial Intelligence,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper aims to address the issues of accuracy and speed in the diagnosis of lung diseases. Specifically, the authors propose a method based on Convolutional Neural Networks (CNN), Data Augmentation, ResNet50, and Vision Transformers (ViT) for detecting lung pathology from medical images. The paper uses a dataset containing X-ray images and CT scans of various lung diseases, including cancer, pneumonia, tuberculosis, and fibrosis. By comparing with other existing methods (such as basic CNN, CNN+Data Augmentation, ResNet50, and ViT), and using performance metrics like accuracy, sensitivity, specificity, and the area under the ROC curve, the results are evaluated. The experimental results show that the proposed method outperforms other methods in all metrics, achieving 98% accuracy and 99% area under the ROC curve. This indicates that the method is an effective and promising tool for diagnosing lung diseases through medical imaging. Additionally, the method has significant implications for improving the health and quality of life of patients with lung diseases and demonstrates potential technological advancements and innovations in the field of artificial intelligence.