Efficient and Accurate Pneumonia Detection Using a Novel Multi-Scale Transformer Approach

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Saber, Alireza, Fateh, Amirreza, Parhami, Pouria, Siahkarzadeh, Alimohammad, Fateh, Mansoor, Ferdowsi, Saideh
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866914086280757248
author Saber, Alireza
Fateh, Amirreza
Parhami, Pouria
Siahkarzadeh, Alimohammad
Fateh, Mansoor
Ferdowsi, Saideh
author_facet Saber, Alireza
Fateh, Amirreza
Parhami, Pouria
Siahkarzadeh, Alimohammad
Fateh, Mansoor
Ferdowsi, Saideh
contents Pneumonia, a prevalent respiratory infection, remains a leading cause of morbidity and mortality worldwide, particularly among vulnerable populations. Chest X-rays serve as a primary tool for pneumonia detection; however, variations in imaging conditions and subtle visual indicators complicate consistent interpretation. Automated tools can enhance traditional methods by improving diagnostic reliability and supporting clinical decision-making. In this study, we propose a novel multi-scale transformer approach for pneumonia detection that integrates lung segmentation and classification into a unified framework. Our method introduces a lightweight transformer-enhanced TransUNet for precise lung segmentation, achieving a Dice score of 95.68% on the "Chest X-ray Masks and Labels" dataset with fewer parameters than traditional transformers. For classification, we employ pre-trained ResNet models (ResNet-50 and ResNet-101) to extract multi-scale feature maps, which are then processed through a modified transformer module to enhance pneumonia detection. This integration of multi-scale feature extraction and lightweight transformer modules ensures robust performance, making our method suitable for resource-constrained clinical environments. Our approach achieves 93.75% accuracy on the "Kermany" dataset and 96.04% accuracy on the "Cohen" dataset, outperforming existing methods while maintaining computational efficiency. This work demonstrates the potential of multi-scale transformer architectures to improve pneumonia diagnosis, offering a scalable and accurate solution to global healthcare challenges. https://github.com/amirrezafateh/Multi-Scale-Transformer-Pneumonia
format Preprint
id arxiv_https___arxiv_org_abs_2408_04290
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Efficient and Accurate Pneumonia Detection Using a Novel Multi-Scale Transformer Approach
Saber, Alireza
Fateh, Amirreza
Parhami, Pouria
Siahkarzadeh, Alimohammad
Fateh, Mansoor
Ferdowsi, Saideh
Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
Pneumonia, a prevalent respiratory infection, remains a leading cause of morbidity and mortality worldwide, particularly among vulnerable populations. Chest X-rays serve as a primary tool for pneumonia detection; however, variations in imaging conditions and subtle visual indicators complicate consistent interpretation. Automated tools can enhance traditional methods by improving diagnostic reliability and supporting clinical decision-making. In this study, we propose a novel multi-scale transformer approach for pneumonia detection that integrates lung segmentation and classification into a unified framework. Our method introduces a lightweight transformer-enhanced TransUNet for precise lung segmentation, achieving a Dice score of 95.68% on the "Chest X-ray Masks and Labels" dataset with fewer parameters than traditional transformers. For classification, we employ pre-trained ResNet models (ResNet-50 and ResNet-101) to extract multi-scale feature maps, which are then processed through a modified transformer module to enhance pneumonia detection. This integration of multi-scale feature extraction and lightweight transformer modules ensures robust performance, making our method suitable for resource-constrained clinical environments. Our approach achieves 93.75% accuracy on the "Kermany" dataset and 96.04% accuracy on the "Cohen" dataset, outperforming existing methods while maintaining computational efficiency. This work demonstrates the potential of multi-scale transformer architectures to improve pneumonia diagnosis, offering a scalable and accurate solution to global healthcare challenges. https://github.com/amirrezafateh/Multi-Scale-Transformer-Pneumonia
title Efficient and Accurate Pneumonia Detection Using a Novel Multi-Scale Transformer Approach
topic Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2408.04290