Vision Transformer Model for Lung Disease Classification Using Chest X-Ray Images
摘要
We develope a pneumoconiosis classification model for chest X-ray images using Vision Transformer(ViT) and compare its accuracy with that of Convolutional Neural Networks (CNNs). Since there is variability in accuracy and differences in the accuracy across models, three ViT pretrained models are used for verification. Common CNN models, VGG16 and ResNet50, are used as comparison models. Then trials are performed for each training condition, and the average correct response rate is used as the evaluation index. The training results have shown that ViT is as accurate as or more accurate than CNN in all training conditions. This suggests that ViT is useful for pneumoconiosis classification. In terms of training time, ViT requires less time than CNN for a small number of epochs.