用大模型精炼脑电图图结构,提升癫痫诊断准确率
LLM as Clinical Graph Structure Refiner: Enhancing Representation Learning in EEG Seizure Diagnosis

- 先用Transformer+MLP生成边概率,再用大模型结合文本与统计特征过滤冗余边
- 在TUSZ数据集上准确率显著提升,图结构更干净可解释
- 适合做脑电信号分析、图神经网络优化的研究者参考
脑电图(EEG)信号对自动癫痫检测至关重要,但其固有噪声使得鲁棒表示学习困难。现有基于相关性或学习的图构建方法常因EEG数据噪声产生冗余或无关边,严重影响图表示质量并限制下游任务性能。受大语言模型(LLM)强大推理与上下文理解能力启发,我们探索将LLM用于图边精炼。提出两阶段框架:首先验证LLM边精炼能有效识别并移除冗余连接,显著提升癫痫检测准确率并生成更合理的图结构。在此基础上,进一步设计鲁棒方案:初始图由Transformer边预测器与多层感知机构建,为潜在边分配概率分数并设阈值决定存在性;随后LLM作为边集精炼器,基于节点对的文本与统计特征做出验证决策。在TUSZ数据集上的大量实验表明,该LLM精炼图学习框架不仅提升任务性能,还生成更清晰、更具可解释性的图表示。
原文摘要 · Abstract (English)
Electroencephalogram (EEG) signals are vital for automated seizure detection, but their inherent noise makes robust representation learning challenging. Existing graph construction methods, whether correlation-based or learning-based, often generate redundant or irrelevant edges due to the noisy nature of EEG data. This significantly impairs the quality of graph representation and limits downstream task performance. Motivated by the remarkable reasoning and contextual understanding capabilities of large language models (LLMs), we explore the idea of using LLMs as graph edge refiners. Specifically, we propose a two-stage framework: we first verify that LLM-based edge refinement can effectively identify and remove redundant connections, leading to significant improvements in seizure detection accuracy and more meaningful graph structures. Building on this insight, we further develop a robust solution where the initial graph is constructed using a Transformer-based edge predictor and multilayer perceptron, assigning probability scores to potential edges and applying a threshold to determine their existence. The LLM then acts as an edge set refiner, making informed decisions based on both textual and statistical features of node pairs to validate the remaining connections. Extensive experiments on TUSZ dataset demonstrate that our LLM-refined graph learning framework not only enhances task performance but also yields cleaner and more interpretable graph representations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。