将临床知识融入神经符号框架,提升医学影像中罕见病检测的鲁棒性与可解释性。
XAI-MeD: Explainable Knowledge Guided Neuro-Symbolic Framework for Domain Generalization and Rare Class Detection in Medical Imaging
- 用逻辑规则编码临床知识,生成可验证的诊断规则。
- 跨域泛化性能提升6%,罕见类F1值提高10%。
- 适合需要可解释性和高可靠性医疗AI的场景。
可解释性与域泛化、罕见类可靠性是医疗AI的关键挑战,深度模型在真实世界分布偏移下常失效且对少见病症存在偏差。本文提出XAI-MeD,一个将临床专家知识整合进深度学习的可解释神经符号框架。该框架通过原子医学命题的逻辑连接,构建可机器验证的类别特异性规则,并以加权特征满足度量化其诊断价值,形成补充神经预测的符号推理分支。通过置信度加权融合符号与深度输出,结合受猎人启发的自适应路由机制(基于熵不平衡增益EIG和罕见类吉尼指数),有效缓解类别不平衡、类内高变异性及不确定性问题。在四个挑战任务上评估:基于静息态fMRI的癫痫发作起始区定位、六个多中心数据集上的糖尿病视网膜病变分级。结果表明,跨域泛化性能提升6%,罕见类F1值提高10%,显著优于现有深度学习基线。消融实验确认临床引导的符号组件作为有效正则化器,增强对分布偏移的鲁棒性。XAI-MeD为多模态医疗AI提供了原理清晰、临床可信且可解释的解决方案。
原文摘要 · Abstract (English)
Explainability domain generalization and rare class reliability are critical challenges in medical AI where deep models often fail under real world distribution shifts and exhibit bias against infrequent clinical conditions This paper introduces XAIMeD an explainable medical AI framework that integrates clinically accurate expert knowledge into deep learning through a unified neuro symbolic architecture XAIMeD is designed to improve robustness under distribution shift enhance rare class sensitivity and deliver transparent clinically aligned interpretations The framework encodes clinical expertise as logical connectives over atomic medical propositions transforming them into machine checkable class specific rules Their diagnostic utility is quantified through weighted feature satisfaction scores enabling a symbolic reasoning branch that complements neural predictions A confidence weighted fusion integrates symbolic and deep outputs while a Hunt inspired adaptive routing mechanism guided by Entropy Imbalance Gain EIG and Rare Class Gini mitigates class imbalance high intra class variability and uncertainty We evaluate XAIMeD across diverse modalities on four challenging tasks i Seizure Onset Zone SOZ localization from rs fMRI ii Diabetic Retinopathy grading across 6 multicenter datasets demonstrate substantial performance improvements including 6 percent gains in cross domain generalization and a 10 percent improved rare class F1 score far outperforming state of the art deep learning baselines Ablation studies confirm that the clinically grounded symbolic components act as effective regularizers ensuring robustness to distribution shifts XAIMeD thus provides a principled clinically faithful and interpretable approach to multimodal medical AI.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。