Large Language Models as Clinical Graph Structure Optimizers: Enhancing Representation Learning for EEG Epilepsy Diagnosis

This study innovatively uses the reasoning and context understanding capabilities of large language models (LLMs) as graph edge optimizers to refine graph structures constructed from EEG signals. It significantly improves epilepsy detection accuracy on the TUSZ dataset and generates more interpretable graph representations. This approach provides new insights for the field of medical AI.

Electroencephalogram (EEG) signal analysis is a key technology for epilepsy detection, but its high noise characteristics hinder automated diagnosis. In recent years, graph neural networks (GNNs) have shown potential in EEG analysis, as they can capture functional connections between brain regions. However, existing graph construction methods tend to generate redundant or irrelevant edges: correlation-based methods easily introduce spurious associations, while learning-based methods overfit under high noise, leading to low-quality graph representations and poor robustness.

The research team first verified the feasibility of using LLMs as edge optimizers: LLMs were tasked with analyzing the rationality of candidate edges and deciding whether to retain or remove them. Experimental results show that LLM optimization effectively removes redundant connections, improves epilepsy detection accuracy, and the optimized graph structure is clearer and more reasonable, proving the theoretical and practical effectiveness of LLM optimization.

Based on the proof-of-concept, the team developed an end-to-end solution consisting of three components:

1. Initial Graph Construction: Use a Transformer edge predictor and MLP to build the initial graph, where the Transformer's self-attention captures long-range dependencies;
2. Feature Engineering: Prepare node text descriptions (electrode positions, brain region affiliations) and statistical features (signal intensity, spectral characteristics) for the LLM;
3. LLM Edge Optimization: Intelligently prune noisy edges by integrating factors such as anatomical location, functional connection strength, and signal similarity.

Experiments were conducted on the TUSZ dataset (the largest public EEG epilepsy dataset): The optimized model's epilepsy detection accuracy was significantly higher than the baseline; the optimized graph structure was clearer and aligned with neuroscientific knowledge, enhancing interpretability; ablation experiments showed that removing the LLM optimization module led to a significant performance drop, verifying its necessity.

This study has significant clinical implications: Interpretability is as important as accuracy, and the optimized graph structure allows doctors to trace the diagnostic logic and build trust in AI systems. Technically, it opens up a new direction for using LLM's general intelligence to optimize structured data in specific domains.

In the future, this method can be extended to other medical signal processing tasks such as electrocardiogram (ECG) and electromyogram (EMG); with the improvement of LLM capabilities and advances in multimodal fusion technology, LLMs are expected to become general reasoning engines for various structured data optimization tasks.