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

Anirudh Mazumder, Jianguo Liu
9/26/2024
eess.IVcs.CV

No confident code implementation found for this paper. 1 possible match can be shown below.

Abstract

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.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. 1 weaker match is hidden by default — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Automated diagnosis of lung diseases using vision transformer: a comparative study on chest x-ray classification

91% match2025

Fast-staged CNN Model for Accurate pulmonary diseases and Lung cancer detection

87% match2024

Lung Disease Detection with Vision Transformers: A Comparative Study of Machine Learning Methods

86% match2024

Deep Learning for Lung Disease Classification Using Transfer Learning and a Customized CNN Architecture with Attention

86% match2024

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

86% match2024

A CT Image Classification Network Framework for Lung Tumors Based on Pre-trained MobileNetV2 Model and Transfer learning, And Its Application and Market Analysis in the Medical field

85% match2025

Cite this paper

@article{mazumder2024developing,
  title  = {Developing a Dual-Stage Vision Transformer Model for Lung Disease Classification},
  author = {Anirudh Mazumder and Jianguo Liu},
  year   = {2024},
  eprint = {2409.18257},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2409.18257v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2409.18257.svg)](https://codeofpaper.com/paper/2409.18257)

Discussion