Proceedings

LLM as Clinical Graph Structure Refiner: Enhancing Representation Learning in EEG Seizure Diagnosis

Download BibTeX Cite

Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence. AI and Health. Pages 6770-6778.

https://doi.org/10.24963/ijcai.2026/753

Abstract

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 representations 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.

  1. [Abeysinghe et al., 2025] Rashmie Abeysinghe, Shiqiang Tao, Samden D Lhatoo, Guo-Qiang Zhang, and Licong Cui. Leveraging pretrained language models for seizure frequency extraction from epilepsy evaluation reports. npj Digital Medicine, 8(1):208, 2025.
  2. [Aldahr et al., 2022] Raghdah Saem Aldahr, Munid Alanazi, and Mohammad Ilyas. Addressing inter-patient variability in eeg: Diversity-enhanced data augmentation and few- shot learning-based epilepsy detection. In 2022 Inter- national Conference on Healthcare Engineering (ICHE), pages 1-7, 2022.
  3. [Barmpas et al., 2025] Konstantinos Barmpas, Na Lee, Alexandros Koliousis, Yannis Panagakis, Dimitrios A Adamos, Nikolaos Laskaris, and Stefanos Zafeiriou. Neu- rorvq: Multi-scale eeg tokenization for generative large brainwave models. arXiv preprint arXiv:2510.13068, 2025.
  4. [Beaulieu-Jones et al., 2023] Brett K Beaulieu-Jones, Mauricio F Villamar, Phil Scordis, Ana Paula Bartmann, Waqar Ali, Benjamin D Wissel, Emily Alsentzer, Johann de Jong, Arijit Patra, and Isaac Kohane. Predicting seizure recurrence after an initial seizure-like episode from routine clinical notes using large language models: a retrospective cohort study. The Lancet Digital Health, 5(12):e882-e894, 2023.
  5. [Begley et al., 2022] Charles Begley, Ryan G Wagner, An- nette Abraham, Ettore Beghi, Charles Newton, Churl-Su Kwon, David Labiner, and Andrea S Winkler. The global cost of epilepsy: a systematic review and extrapolation. Epilepsia, 63(4):892-903, 2022.
  6. [Benbadis et al., 2020] Selim Benbadis, Sándor Beniczky, Edward Bertram, Stephanie MacIver, and Solomon L Moshé. The role of eeg in patients with suspected epilepsy. Epileptic Disorders, (2):143-155, 2020.
  7. [Bhandari et al., 2024] Harish C Bhandari, Yagya R Pandeya, Kanhaiya Jha, Sudan Jha, and Sultan Ahmad. Exploring non-euclidean approaches: A comprehensive survey on graph-based techniques for eeg signal anal- ysis. Journal of Advances in Information Technology, 15(10):1089-1105, 2024.
  8. [Chen et al., 2022] Junru Chen, Yang Yang, Tao Yu, Yingy- ing Fan, Xiaolong Mo, and Carl Yang. Brainnet: Epileptic wave detection from seeg with hierarchical graph diffusion learning. In Proceedings of the 28th ACM SIGKDD Con- ference on Knowledge Discovery and Data Mining, pages 2741-2751, 2022.
  9. [Cini et al., 2023] Andrea Cini, Daniele Zambon, and Cesare Alippi. Sparse graph learning from spatiotemporal time series. Journal of Machine Learning Research, 24(242):1- 36, 2023.
  10. [Cui et al., 2024] Wenhui Cui, Woojae Jeong, Philipp Thölke, Takfarinas Medani, Karim Jerbi, Anand A Joshi, and Richard M Leahy. Neuro-gpt: Towards a foundation model for eeg. In 2024 IEEE International Symposium on Biomedical Imaging (ISBI), pages 1-5, 2024.
  11. [Dong and Kluger, 2023] Mingze Dong and Yuval Kluger. Towards understanding and reducing graph structural noise for gnns. In International Conference on Machine Learning, pages 8202-8226. PMLR, 2023.
  12. [Ford et al., 2025] Joseph Ford, Nathan Pevy, Richard Grunewald, Stephen Howell, and Markus Reuber. Can artificial intelligence diagnose seizures based on patients' descriptions? a study of gpt-4. Epilepsia, 2025.
  13. [Ho and Armanfard, 2023] Thi Kieu Khanh Ho and Narges Armanfard. Self-supervised learning for anomalous chan- nel detection in eeg graphs: Application to seizure anal- ysis. In Proceedings of the AAAI conference on artificial intelligence, volume 37, pages 7866-7874, 2023.
  14. [Huang et al., 2025] Zhentao Huang, Yuyao Yang, Yahong Ma, Qi Dong, Jianyun Su, Hangyu Shi, Shanwen Zhang, and Liangliang Hu. Eeg detection and recognition model for epilepsy based on dual attention mechanism. Scientific Reports, 15(1):9404, 2025.
  15. [Jiang et al., 2024a] Wei-Bang Jiang, Yansen Wang, Bao- Liang Lu, and Dongsheng Li. Neurolm: A universal multi- task foundation model for bridging the gap between lan- guage and eeg signals. arXiv preprint arXiv:2409.00101, 2024.
  16. [Jiang et al., 2024b] Wei-Bang Jiang, Li-Ming Zhao, and Bao-Liang Lu. Large brain model for learning generic representations with tremendous eeg data in bci. arXiv preprint arXiv:2405.18765, 2024.
  17. [Kerr et al., 2025] Wesley T Kerr, Katherine N McFarlane, Gabriela Figueiredo Pucci, Danielle R Carns, Alex Is- rael, Lianne Vighetti, Page B Pennell, John M Stern, Zongqi Xia, and Yanshan Wang. Supervised machine learning compared to large language models for identify- ing functional seizures from medical records. Epilepsia, 66(4):1155-1164, 2025.
  18. [Kim et al., 2024] Jonathan W Kim, Ahmed Alaa, and Danilo Bernardo. Eeg-gpt: exploring capabilities of large language models for eeg classification and interpretation. arXiv preprint arXiv:2401.18006, 2024.
  19. [Klepl et al., 2024] Dominik Klepl, Min Wu, and Fei He. Graph neural network-based eeg classification: A survey. IEEE Transactions on Neural Systems and Rehabilitation Engineering, 32:493-503, 2024.
  20. [Li et al., 2024] Yang Li, Yang Yang, Qinghe Zheng, Yunxia Liu, Hongjun Wang, Shangling Song, and Penghui Zhao. Dynamical graph neural network with attention mecha- nism for epilepsy detection using single channel eeg. Med- ical & Biological Engineering & Computing, 62(1):307- 326, 2024.
  21. [Li et al., 2026] Lincan Li, Rikuto Kotoge, Xihao Piao, Zheng Chen, and Yushun Dong. Optimizing eeg graph structure for seizure detection: An information bottleneck and self-supervised learning approach. arXiv preprint arXiv:2604.01595, 2026.
  22. [Mishra et al., 2025] Abhijit Mishra, Shreya Shukla, Jose Torres, Jacek Gwizdka, and Shounak Roychowdhury. Thought2text: text generation from eeg signal using large language models (llms). In Findings of the Association for Computational Linguistics: NAACL 2025, pages 3747- 3759, 2025.
  23. [Rani et al., 2024] N Geetha Rani, Nayakanti Praveen Ku- mar, et al. Attcnnnet: Attention based cnn network to detect seizures from eeg subjects. In 2024 IEEE 16th In- ternational Conference on Computational Intelligence and Communication Networks (CICN), pages 800-804, 2024.
  24. [Saba-Sadiya et al., 2021] Sari Saba-Sadiya, Eric Chantland, Tuka Alhanai, Taosheng Liu, and Mohammad M Ghas- semi. Unsupervised eeg artifact detection and correction. Frontiers in digital health, 2:608920, 2021.
  25. [Shah et al., 2018] Vinit Shah, Eva Von Weltin, Silvia Lopez, James Riley McHugh, Lillian Veloso, Meysam Golmohammadi, Iyad Obeid, and Joseph Picone. The tem- ple university hospital seizure detection corpus. Frontiers in neuroinformatics, 12:83, 2018.
  26. [Shankar et al., 2021] A. Shankar, S. Dandapat, and S. Barma. Classification of seizure types based on statistical variants and machine learning. 2021 IEEE 18th India Council International Conference, pages 1-6, 2021.
  27. [Sharmila and Geethanjali, 2020] A. Sharmila and P. Geethanjali. Evaluation of time domain features on detection of epileptic seizure from eeg signals. Health and Technology, 10:711-722, 2020.
  28. [Shu et al., 2024] Kai Shu, Le Wu, Yuchang Zhao, Aiping Liu, Ruobing Qian, and Xun Chen. Data augmentation for seizure prediction with generative diffusion model. IEEE Transactions on Cognitive and Developmental Systems, 2024.
  29. [Tang et al., 2022] Siyi Tang, Jared Dunnmon, Khaled Ka- mal Saab, Xuan Zhang, Qianying Huang, Florian Du- bost, Daniel Rubin, and Christopher Lee-Messer. Self- supervised graph neural networks for improved electroen- cephalographic seizure analysis. In International Confer- ence on Learning Representations, 2022.
  30. [Tang et al., 2023a] Siyi Tang, Jared Dunnmon, Liangqiong Qu, et al. Modeling multivariate biosignals with graph neural networks and structured state space. In ICLR 2023 Workshop on Time Series Representation Learning for Health, 2023.
  31. [Tang et al., 2023b] Siyi Tang, Jared A Dunnmon, Qu Liangqiong, Khaled K Saab, Tina Baykaner, Christopher Lee-Messer, and Daniel L Rubin. Mod- eling multivariate biosignals with graph neural networks and structured state space models. In Proceedings of the Conference on Health, Inference, and Learning, pages 50-71, 2023.
  32. [Tao et al., 2022] Tian-li Tao, Liang-hu Guo, Qiang He, Han Zhang, and Lin Xu. Seizure detection by brain- connectivity analysis using dynamic graph isomorphism network. In 2022 44th Annual International Conference of the IEEE Engineering in Medicine & Biology Society (EMBC), pages 2302-2305, 2022.
  33. [Wang and Yuan, 2025] Zuo Wang and Ye Yuan. Jensen- shannon divergence message-passing for rich-text graph representation learning. arXiv preprint arXiv:2512.20094, 2025.
  34. [Wang et al., 2020] Jialin Wang, Shen Liang, Dake He, Ye Wang, Yingpei Wu, and Yanchun Zhang. A sequen- tial graph convolutional network with frequency-domain complex network of eeg signals for epilepsy detection. In 2020 IEEE International Conference on Bioinformatics and Biomedicine (BIBM), pages 785-792. IEEE, 2020.
  35. [Weng et al., 2024] Weining Weng, Yang Gu, Shuai Guo, Yuan Ma, Zhaohua Yang, Yuchen Liu, and Yiqiang Chen. Self-supervised learning for electroencephalogram: A sys- tematic survey. arXiv preprint arXiv:2401.05446, 2024.
  36. [Xu et al., 2024] Jie Xu, Shasha Yuan, Junliang Shang, Juan Wang, Kuiting Yan, and Yankai Yang. Spatiotempo- ral network based on gcn and bigru for seizure detec- tion. IEEE Journal of Biomedical and Health Informatics, 28(4):2037-2046, 2024.
  37. [Xu et al., 2026] Rongchao Xu, Kunlin Cai, Lin Jiang, Zhiqing Hong, Yuan Tian, and Guang Wang. Geogen: A two-stage coarse-to-fine framework for fine-grained syn- thetic location-based social network trajectory generation. In Proceedings of the AAAI Conference on Artificial Intel- ligence, 2026.
  38. [Yang et al., 2024] Shihao Yang, Yaxi Luo, Meng Jiao, Neel Fotedar, Vikram R Rao, Xinglong Ju, Shasha Wu, Xi- aochen Xian, Hai Sun, Ioannis Karakis, et al. Episemollm: A fine-tuned large language model for epileptogenic zone localization based on seizure semiology with a perfor- mance comparable to epileptologists. MedRxiv, pages 2024-09, 2024.
  39. [Yu et al., 2026] Dahai Yu, Lin Jiang, Rongchao Xu, and Guang Wang. Healthmamba: An uncertainty-aware spa- tiotemporal graph state space model for effective and re- liable healthcare facility visit prediction. arXiv preprint arXiv:2602.05286, 2026. IJCAI
  40. [Zhang et al., 2022] Shang Zhang, Guangda Liu, Ruolan Xiao, Wenjie Cui, Jing Cai, Xinlei Hu, Yubing Sun, Jiqing Qiu, and Yuan Qi. A combination of statistical parame- ters for epileptic seizure detection and classification using vmd and nltwsvm. Biocybernetics and Biomedical Engi- neering, 2022.
  41. [Zhao et al., 2021] Yanna Zhao, Changxu Dong, Gaobo Zhang, Yaru Wang, Xin Chen, Weikuan Jia, Qi Yuan, Fangzhou Xu, and Yuanjie Zheng. Eeg-based seizure de- tection using linear graph convolution network with focal loss. Computer Methods and Programs in Biomedicine, 208:106277, 2021.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards