arXiv Open Access 2024

Developing a Dual-Stage Vision Transformer Model for Lung Disease Classification

Anirudh Mazumder Jianguo Liu

Lihat Sumber

Abstrak

Lung diseases have become a prevalent problem throughout the United States, affecting over 34 million people. Accurate and timely diagnosis of the different types of lung diseases is critical, and Artificial Intelligence (AI) methods could speed up these processes. A dual-stage vision transformer is built throughout this research by integrating a Vision Transformer (ViT) and a Swin Transformer to classify 14 different lung diseases from X-ray scans of patients with these diseases. The proposed model achieved an accuracy of 92.06% on a label-level when making predictions on an unseen testing subset of the dataset after data preprocessing and training the neural network. The model showed promise for accurately classifying lung diseases and diagnosing patients who suffer from these harmful diseases.

Topik & Kata Kunci

eess.IV cs.CV

Penulis (2)

Anirudh Mazumder

Jianguo Liu

Format Sitasi

APA MLA BibTeX

Mazumder, A., Liu, J. (2024). Developing a Dual-Stage Vision Transformer Model for Lung Disease Classification. https://arxiv.org/abs/2409.18257

Akses Cepat

Lihat di Sumber

Informasi Jurnal

Tahun Terbit: 2024
Bahasa: en
Sumber Database: arXiv
Akses: Open Access ✓