arXiv:2607.14314cs.LG2026-07被引 1

用医学知识修正脑电图图结构,提升癫痫诊断准确性和可解释性。

NeuroGRIP: Retrieval-Augmented Graph Refinement for Knowledge-Grounded EEG Seizure Diagnosis

论文配图:NeuroGRIP: Retrieval-Augmented Graph Refinement for Knowledge-Grounded EEG Seizure Diagnosis
图 1 · 摘自论文原文
  • 通过检索临床指南构建知识图谱,为脑电图图结构提供外部医学先验。
  • 在TUSZ和CHB-MIT数据集上,癫痫检测准确率显著提升。
  • 适合关注可解释医疗AI、脑电图分析的研究者与临床工程师。

从脑电信号中进行癫痫诊断是一项关键但持续具有挑战性的任务,因其复杂的神经动力学及通道间建模中的虚假连接。尽管时空图神经网络(STGNN)推动了脑网络表征学习,但其生成的图结构因纯数据驱动而临床合理性低、可解释性差。为此,我们提出NeuroGRIP,一种基于检索增强的图精炼框架,将外部医学知识引入以校正噪声脑电图。我们首先构建了一个大规模、领域专用的知识库,源自权威临床指南;利用大语言模型提取结构化生物医学实体与关系,形成文本知识图(KG),作为临床先验的外部来源。框架通过投影STGNN生成的节点嵌入至KG语义空间,执行对齐感知的查询构造,并使用FAISS进行相似度搜索,从知识三元组中检索关系证据。每个预测边根据检索到的相似度、关系类型和来源可靠性获得置信度评分,从而剔除医学上不合理的边。在TUSZ和CHB-MIT上的大量实验表明,NeuroGRIP不仅提升了癫痫检测准确率,还通过将每项预测锚定在经临床验证的知识上,增强了可解释性。本工作首次实现了通过检索增强推理,将脑活动动态与外部医学专长紧密结合,为知识增强、可解释的临床诊断开辟新路径。代码已开源:https://github.com/LincanLi-X/NeuroGRIP。

原文摘要 · Abstract (English)

Seizure diagnosis from EEG signals is a critical yet persistently challenging task, due to the complicated neural dynamics and the spurious connections in inter-channel modeling. While spatial-temporal graph neural networks (STGNNs) have advanced EEG brain network representation learning, the resulting graph structures suffer from low clinical plausibility and limited interpretability due to their purely data-driven nature. To this end, we introduce NeuroGRIP, a retrieval-augmented graph refinement framework that incorporates external medical knowledge to calibrate noisy EEG graphs. We first construct a large-scale, domain-specific knowledge base derived from authoritative clinical guidelines. Leveraging large language models, we extract structured biomedical entities and relations to form a textual knowledge graph (KG), which serves as external knowledge source of clinical priors. Our framework performs alignment-aware query construction by projecting STGNN-generated EEG node embeddings into the semantic space of KG. Semantic queries are then executed via FAISS-based similarity search over knowledge triplets to retrieve relation evidence. Each predicted edge is assigned a confidence score based on retrieved similarity, relation type, and source reliability, enabling us to prune medically implausible edges from the originally predicted graph. Extensive experiments on TUSZ and CHB-MIT demonstrate that NeuroGRIP not only improves seizure detection accuracy but also enhances interpretability by grounding each prediction in clinically validated knowledge. This work provides the first unified framework that tightly couples brain dynamics with external medical expertise via retrieval-augmented reasoning, paving the way for knowledge-enhanced, explainable clinical diagnosis. The code is available at: https://github.com/LincanLi-X/NeuroGRIP.

脑电图知识图谱可解释性癫痫诊断

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。