English

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

Image and Video Processing 2024-09-24 v1 Artificial Intelligence Computer Vision and Pattern Recognition

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.

Keywords

Cite

@article{arxiv.2409.14446,
  title  = {Detection of pulmonary pathologies using convolutional neural networks, Data Augmentation, ResNet50 and Vision Transformers},
  author = {Pablo Ramirez Amador and Dinarle Milagro Ortega and Arnold Cesarano},
  journal= {arXiv preprint arXiv:2409.14446},
  year   = {2024}
}

Comments

10 pages

R2 v1 2026-06-28T18:52:52.910Z