让图神经网络自己解释决策过程,提升医疗图像诊断可信度。
X-Node: Self-Explanation is All We Need
- 每个节点生成自身推理依据,实现局部可解释性
- 保持分类精度同时生成忠实的自然语言解释
- 适合医疗等高风险场景,需透明决策的领域
图神经网络在计算机视觉和医学图像分类中表现优异,但其决策过程缺乏透明性,限制了在临床等高风险场景中的应用。现有解释方法多为全局后验分析,难以揭示单个节点的局部推理逻辑。本文提出X-Node,一种自解释图神经网络框架:每个节点在预测过程中生成自身解释。通过构建包含度、中心性、聚类、特征显著性及标签一致性等可解释线索的结构化上下文向量,由轻量级推理模块映射为紧凑解释向量。该向量用于三方面:(1)通过解码器重建节点隐表示以保证忠实性;(2)利用预训练大模型(如Grok、Gemini)生成自然语言解释;(3)通过“文本注入”机制将解释反馈至消息传递流程,指导模型学习。在基于MedMNIST和MorphoMNIST构建的两个图数据集上,结合GCN、GAT、GIN主干网络进行评估,结果表明X-Node在保持竞争力分类准确率的同时,能生成忠实的逐节点解释。
原文摘要 · Abstract (English)
Graph neural networks (GNNs) have achieved state-of-the-art results in computer vision and medical image classification tasks by capturing structural dependencies across data instances. However, their decision-making remains largely opaque, limiting their trustworthiness in high-stakes clinical applications where interpretability is essential. Existing explainability techniques for GNNs are typically post-hoc and global, offering limited insight into individual node decisions or local reasoning. We introduce X-Node, a self-explaining GNN framework in which each node generates its own explanation as part of the prediction process. For every node, we construct a structured context vector encoding interpretable cues such as degree, centrality, clustering, feature saliency, and label agreement within its local topology. A lightweight Reasoner module maps this context into a compact explanation vector, which serves three purposes: (1) reconstructing the node's latent embedding via a decoder to enforce faithfulness, (2) generating a natural language explanation using a pre-trained LLM (e.g., Grok or Gemini), and (3) guiding the GNN itself via a "text-injection" mechanism that feeds explanations back into the message-passing pipeline. We evaluate X-Node on two graph datasets derived from MedMNIST and MorphoMNIST, integrating it with GCN, GAT, and GIN backbones. Our results show that X-Node maintains competitive classification accuracy while producing faithful, per-node explanations. Repository: https://github.com/basiralab/X-Node.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。