Explainable AI for Lung Disease Classification: Looking Inside the Black Box
摘要
In this paper, we propose a novel multimodal classification algorithm for pulmonary disease by combining chest X-ray images and clinical symptom texts. Our method leverages a Vision Transformer (ViT) to classify chest X-ray images, and a local eXtreme Gradient Boosting (XGB) model trained on the k nearest neighbors of clinical symptom data. The symptom texts are represented using a Bag-of-Words (BoW) model. Experiments on a real-world dataset collected from Chau Doc Hospital (An Giang, Vietnam) show that our multimodal model achieves a classification accuracy of 85.29% on the testset. Furthermore, we introduce the LoXIM (Local eXplanation with Instance-based Modeling), explains the prediction for each test case by extracting important features (words) from the local XGB model. For each test symptom text, words that match the important features are highlighted according to their importance weights. This enhances transparency and enables clinicians to better understand and trust the model’s decisions.