arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.07055eess.IVcs.CVcs.LG

Vision Transformer for COVID-19 CXR Diagnosis using Chest X-ray Feature Corpus

Korea Advanced Institute of Science and Technology(韩国科学技术院);Asan Medical Center(峨山医院);University of Ulsan College of Medicine(蔚山大学医学院);College of Medicine, Yeungnam University(岭南大学医学院);School of Medicine, Kyungpook National University(庆北国立大学医学院)

Sangjoon Park, Gwanghyun Kim, Yujin Oh, Joon Beom Seo, Sang Min Lee, Jin Hwan Kim, Sungjun Moon, Jae-Kwang Lim, Jong Chul Ye

英文摘要:

Under the global COVID-19 crisis, developing robust diagnosis algorithm for COVID-19 using CXR is hampered by the lack of the well-curated COVID-19 data set, although CXR data with other disease are abundant. This situation is suitable for vision transformer architecture that can exploit the abundant unlabeled data using pre-training. However, the direct use of existing vision transformer that uses the corpus generated by the ResNet is not optimal for correct feature embedding. To mitigate this problem, we propose a novel vision Transformer by using the low-level CXR feature corpus that are obtained to extract the abnormal CXR features. Specifically, the backbone network is trained using large public datasets to obtain the abnormal features in routine diagnosis such as consolidation, glass-grass opacity (GGO), etc. Then, the embedded features from the backbone network are used as corpus for vision transformer training. We examine our model on various external test datasets acquired from totally different institutions to assess the generalization ability. Our experiments demonstrate that our method achieved the state-of-art performance and has better generalization capability, which are crucial for a widespread deployment.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑