The growing adoption of Artificial Intelligence (AI) in medical imaging has elevated the need for models that balance high diagnostic performance with interpretability. This paper presents TransGlass, a Transformer-based glass-box AI framework for chest X-ray disease classification that ensures transparent decision-making. TransGlass aims at improving both the accuracy and explanations of classification over the Vision Transformer (ViT) and Swin Transformer networks as the main goal. The framework uses attention-based visualization methods, including Grad-CAM and SHAP to justify AI prediction. On the CHEXPERT dataset, TransGlass had an AUC-ROC of 0.91 of pneumonia classification and a high recall and explainability in contrast to conventional CNNs. We demonstrate the clinical feasibility of use of TransGlass in interpretable AI-assisted diagnostics, which makes it reliable in a high-stakes medical setting.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

TransGlass: A Transformer-Based Glass-Box AI Framework for Interpretable Predictions in Critical Domains

  • S. Aarathi,
  • CH. N. Santhosh Kumar,
  • Sumathy Muniamuthu,
  • D. Anil,
  • Preethi,
  • Charul Jain

摘要

The growing adoption of Artificial Intelligence (AI) in medical imaging has elevated the need for models that balance high diagnostic performance with interpretability. This paper presents TransGlass, a Transformer-based glass-box AI framework for chest X-ray disease classification that ensures transparent decision-making. TransGlass aims at improving both the accuracy and explanations of classification over the Vision Transformer (ViT) and Swin Transformer networks as the main goal. The framework uses attention-based visualization methods, including Grad-CAM and SHAP to justify AI prediction. On the CHEXPERT dataset, TransGlass had an AUC-ROC of 0.91 of pneumonia classification and a high recall and explainability in contrast to conventional CNNs. We demonstrate the clinical feasibility of use of TransGlass in interpretable AI-assisted diagnostics, which makes it reliable in a high-stakes medical setting.