Lt-vit: A Vision Transformer For Multi-label Chest X-ray Classification | Awesome LLM Papers

Lt-vit: A Vision Transformer For Multi-label Chest X-ray Classification

Umar Marikkar, Sara Atito, Muhammad Awais, Adam Mahdi Β· 2023 IEEE International Conference on Image Processing (ICIP) Β· 2023

Vision Transformers (ViTs) are widely adopted in medical imaging tasks, and some existing efforts have been directed towards vision-language training for Chest X-rays (CXRs). However, we envision that there still exists a potential for improvement in vision-only training for CXRs using ViTs, by aggregating information from multiple scales, which has been proven beneficial for non-transformer networks. Hence, we have developed LT-ViT, a transformer that utilizes combined attention between image tokens and randomly initialized auxiliary tokens that represent labels. Our experiments demonstrate that LT-ViT (1) surpasses the state-of-the-art performance using pure ViTs on two publicly available CXR datasets, (2) is generalizable to other pre-training methods and therefore is agnostic to model initialization, and (3) enables model interpretability without grad-cam and its variants.

Similar Work
Loading…