Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

MedXAI: A Retrieval-Augmented and Self-Verifying Framework for Knowledge-Guided Medical Image Analysis

About

Accurate and interpretable image-based diagnosis remains a fundamental challenge in medical AI, particularly under domain shifts and rare-class conditions. Deep learning models often struggle with real-world distribution changes, exhibit bias against infrequent pathologies, and lack the transparency required for deployment in safety-critical clinical environments. We introduce MedXAI (An Explainable Framework for Medical Imaging Classification), a unified expert knowledge based framework that integrates deep vision models with clinician-derived expert knowledge to improve generalization, reduce rare-class bias, and provide human-understandable explanations by localizing the relevant diagnostic features rather than relying on technical post-hoc methods (e.g., Saliency Maps, LIME). We evaluate MedXAI across heterogeneous modalities on two challenging tasks: (i) Seizure Onset Zone localization from resting-state fMRI, and (ii) Diabetic Retinopathy grading. Ex periments on ten multicenter datasets show consistent gains, including a 3% improvement in cross-domain generalization and a 10% improvmnet in F1 score of rare class, substantially outperforming strong deep learning baselines. Ablations confirm that the symbolic components act as effective clinical priors and regularizers, improving robustness under distribution shift. MedXAI delivers clinically aligned explanations while achieving superior in-domain and cross-domain performance, particularly for rare diseases in multimodal medical AI.

Midhat Urooj, Ayan Banerjee, Farhat Shaikh, Kuntal Thakur, Sandeep Gupta• 2025

Related benchmarks

TaskDatasetResultRank
Diabetic Retinopathy GradingEyepacs (held-out)
Accuracy74.8
5
Diabetic Retinopathy GradingMessidor (held-out)
Accuracy68.3
5
Diabetic Retinopathy GradingMessidor 2 (held-out)
Accuracy75.6
5
Diabetic Retinopathy GradingAptos (held-out)
Accuracy53.1
5
Diabetic Retinopathy GradingAPTOS Source Domain Generalization to other domains
Accuracy59.9
4
Diabetic Retinopathy GradingMESSIDOR2 Source Domain Generalization to other domains
Accuracy65.5
4
Diabetic Retinopathy GradingMESSIDOR Source Domain Generalization to other domains
Accuracy0.671
4
Diabetic Retinopathy GradingEYEPACS Source Domain (Generalization to other domains)
Accuracy61.7
4
Seizure Onset Zone (SOZ) LocalizationPhoenix Child Health Center (PCH) rs-fMRI Independent Components (test)
Accuracy84.6
3
Diabetic Retinopathy ClassificationDiabetic Retinopathy MDG setting
F1 (Grade 3)50.1
2
Showing 10 of 10 rows

Other info

Follow for update