Search papers, labs, and topics across Lattice.
This paper introduces a novel two-stage framework for EEG seizure diagnosis that leverages LLMs to refine graph structures. The method first constructs an initial graph using a Transformer-based edge predictor and MLP, then uses an LLM to validate edges based on textual and statistical features of node pairs. Experiments on the TUSZ dataset show that LLM-refined graphs improve seizure detection accuracy and produce more interpretable graph representations.
LLMs can prune noisy edges in EEG graphs, leading to more accurate and interpretable seizure detection.
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.
