A Hybrid Spatio-Temporal Graph Transformer for EEG-Based ADHD Detection via Network Index Modeling
Abstract
1. Introduction
- A novel formulation of ADHD detection as a network index extraction problem.
- A hybrid spatio-temporal model combining Transformer and GRU for dynamic connectivity analysis.
- A systematic comparison of static and dynamic network index models.
- Demonstration of improved separability and generalization on EEG data. The paper is organized as follows. Section 2 reviews existing research devoted to EEG analysis and functional connectivity in ADHD. Section 3 describes the data used, preprocessing procedures, and methods for forming static and dynamic network indices. Section 4 presents the experimental results and their quantitative analysis. Section 5 discusses the systemic and applied aspects of the obtained results for clinical decision support systems, followed by the main conclusions.
2. Related Work
- A dynamic representation of EEG as sequences of functional graphs;
- A hybrid spatio-temporal model with attention mechanisms, sensitive to the temporal evolution of network interactions;
- The formation of an integrated network index suitable for use in intelligent diagnostic systems.
3. Materials and Methods
3.1. Data Description
3.2. Data Processing and Splitting
3.3. Baseline Network Index Models
3.4. A Proposed Hybrid Multihead Spatio-Temporal Graph Transformer Index
4. Results
4.1. Descriptive Statistics and Quality Control of EEG Data
4.2. Quantitative Assessment of the Diagnostic Value of Network Features
4.3. Network Index Summary
5. Discussion
6. Conclusions
Supplementary Materials
Author Contributions
Funding
Data Availability Statement
Conflicts of Interest
References
- Faraone, S.V.; Bellgrove, M.A.; Brikell, I.; Cortese, S.; Hollis, C.; Newcorn, J.H.; Philipsen, A.; Polanczyk, G.V.; Rubia, K.; Sibley, M.H.; et al. Attention-deficit/hyperactivity disorder. Nat. Rev. Dis. Primers 2024, 10, 11. [Google Scholar] [CrossRef]
- Cortese, S.; Adamo, N.; Del Giovane, C.; Mohr-Jensen, C.; Hayes, A.J.; Carucci, S.; Atkinson, L.Z.; Tessari, L.; Banaschewski, T.; Coghill, D.; et al. Comparative efficacy and tolerability of medications for attention-deficit hyperactivity disorder in children, adolescents, and adults: A systematic review and network meta-analysis. Lancet Psychiatry 2018, 5, 727–738. [Google Scholar] [CrossRef] [PubMed]
- Goodspeed, K.; Armstrong, D.; Dolce, A.; Evans, P.; Said, R.; Tsai, P.; Sirsi, D. Electroencephalographic (EEG) biomarkers in genetic neurodevelopmental disorders. J. Child Neurol. 2023, 38, 466–477. [Google Scholar] [CrossRef]
- Hurjui, I.A. Biomarkers and neuropsychological tools in attention-deficit/hyperactivity disorder: Advances and challenges. Medicina 2025, 61, 1211. [Google Scholar] [CrossRef]
- Preti, M.G.; Van De Ville, D. Dynamic functional connectivity: State of the art and perspectives. NeuroImage 2017, 160, 41–54. [Google Scholar] [CrossRef]
- Zhu, Z.; Wang, H.; Li, X.; Pan, Y.; Li, L. Dynamic functional connectivity changes of resting-state brain network in attention-deficit/hyperactivity disorder. Behav. Brain Res. 2023, 437, 114121. [Google Scholar] [CrossRef] [PubMed]
- Chen, H.; Song, Y.; Li, X. A deep learning framework for identifying children with ADHD using an EEG-based brain network. Neurocomputing 2019, 356, 83–96. [Google Scholar] [CrossRef]
- Klepl, D.; Wu, M.; He, F. Graph neural network–based EEG classification: A survey. IEEE Trans. Neural Syst. Rehabil. Eng. 2024, 32, 1590–1605. [Google Scholar] [CrossRef]
- Sheeraz, M.; Aslam, A.R.; Drakakis, E.M.; Heidari, H.; Altaf, M.A.B.; Saadeh, W. A Closed-Loop Ear-Worn Wearable EEG System with Real-Time Passive Electrode Skin Impedance Measurement for Early Autism Detection. Sensors 2024, 24, 7489. [Google Scholar] [CrossRef]
- Ahmadi Moghadam, E.; Abedinzadeh Torghabeh, F.; Hosseini, S.A.; Moattar, M.H. Improved ADHD diagnosis using EEG connectivity and deep learning through combining Pearson correlation coefficient and phase-locking value. Neuroinformatics 2024, 22, 521–537. [Google Scholar] [CrossRef]
- Wang, T.; Fang, Y.; Zhang, P.; Ji, M.; Shen, M.; Cao, H.; Zhang, J. DSTMNet: A dual spatiotemporal modeling network for EEG-based emotion recognition. Neurocomputing 2026, 671, 132633. [Google Scholar] [CrossRef]
- Amirbay, A.; Baigabylov, N.; Mukhanova, A.; Mukhambetova, K.; Zaitov, E.; Burganova, R.; Khusanova, K.; Akhmedova, F. Autism detection using facial and motor analysis using machine learning. Bull. Electr. Eng. Inform. 2025, 14, 3985–4000. [Google Scholar] [CrossRef]
- Snyder, S.M.; Hall, J.R. A meta-analysis of quantitative EEG power associated with attention-deficit/hyperactivity disorder. J. Clin. Neurophysiol. 2006, 23, 440–455. [Google Scholar] [CrossRef]
- Arns, M.; Conners, C.K.; Kraemer, H.C. A decade of EEG theta/beta ratio research in ADHD: A meta-analysis. J. Atten. Disord. 2013, 17, 374–383. [Google Scholar] [CrossRef]
- Lenartowicz, A.; Loo, S.K. Use of EEG to diagnose ADHD. Curr. Psychiatry Rep. 2014, 16, 498. [Google Scholar] [CrossRef] [PubMed]
- Ghaderi, A.H.; Nazari, M.A.; Shahrokhi, H.; Darooneh, A.H. Functional brain connectivity differences between different ADHD presentations: Impaired functional segregation in ADHD-combined presentation but not in ADHD-inattentive presentation. Basic Clin. Neurosci. 2017, 8, 267–278. [Google Scholar] [CrossRef]
- Wang, L.; Zhu, C.; He, Y.; Zang, Y.; Cao, Q.; Zhang, H.; Zhong, Q.; Wang, Y. Altered small-world brain functional networks in children with attention-deficit/hyperactivity disorder. Hum. Brain Mapp. 2009, 30, 638–649. [Google Scholar] [CrossRef] [PubMed]
- Ekhlasi, A.; Nasrabadi, A.M.; Mohammadi, M.R. Classification of children with ADHD and healthy controls based on EEG directed phase transfer entropy and graph analysis. Front. Biomed. Technol. 2021, 8, 115–122. [Google Scholar] [CrossRef]
- Bashivan, P.; Rish, I.; Yeasin, M.; Codella, N. Learning representations from EEG with deep recurrent-convolutional neural networks. arXiv 2015, arXiv:1511.06448. [Google Scholar]
- Chugh, N.; Aggarwal, S.; Balyan, A. The hybrid deep learning model for identification of attention-deficit/hyperactivity disorder using EEG. Clin. EEG Neurosci. 2024, 55, 22–33. [Google Scholar] [CrossRef]
- Alsharif, N.; Al-Adhaileh, M.H.; Al-Yaari, M. Diagnosis of attention deficit hyperactivity disorder: A deep learning approach. AIMS Math. 2024, 9, 10580–10608. [Google Scholar] [CrossRef]
- Luo, L.; Wang, Y.; Zeng, L.; Li, X.; Hu, Y.; Chen, H. Patterns of brain dynamic functional connectivity are linked with attention-deficit/hyperactivity disorder–related behavioral and cognitive dimensions. Psychol. Med. 2023, 53, 6666–6677. [Google Scholar] [CrossRef]
- Sastry, N.C.; Banerjee, A. Dynamicity of brain network organization & their community architecture as characterizing features for classification of common mental disorders from whole-brain connectome. Transl. Psychiatry 2024, 14, 268. [Google Scholar] [CrossRef]
- Ricchi, I.; Tarun, A.; Petric Maretic, H.; Frossard, P.; Van De Ville, D. Dynamics of functional network organization through graph mixture learning. NeuroImage 2022, 252, 119037. [Google Scholar] [CrossRef]
- Chen, L.; Qiao, C.; Ren, K.; Qu, G.; Calhoun, V.D.; Stephen, J.M.; Wilson, T.W.; Wang, Y.-P. Explainable spatio-temporal graph evolution learning with applications to dynamic brain network analysis during development. NeuroImage 2024, 298, 120771. [Google Scholar] [CrossRef] [PubMed]
- Ekhlasi, A.; Nasrabadi, A.M.; Mohammadi, M.R. Analysis of EEG brain connectivity of children with ADHD using graph theory and directional information transfer. Biomed. Eng. Biomed. Tech. 2023, 68, 423–435. [Google Scholar] [CrossRef] [PubMed]
- Bakhtyari, M.; Mirzaei, S. ADHD detection using dynamic connectivity patterns of EEG data and ConvLSTM with attention framework. Biomed. Signal Process. Control 2022, 76, 103708. [Google Scholar] [CrossRef]
- Nasrabadi, A.M.; Allahverdy, A.; Samavati, M.; Mohammadi, M. EEG Data for ADHD/Control Children. IEEE DataPort. Available online: https://ieee-dataport.org/open-access/eeg-data-adhd-control-children (accessed on 20 May 2026).
- Esas, M.Y.; Latifoğlu, F. Detection of ADHD from EEG Signals Using New Hybrid Decomposition and Deep Learning Techniques. J. Neural Eng. 2023, 20, 036028. [Google Scholar] [CrossRef]
- Alim, A.; Imtiaz, M.H. Automatic Identification of Children with ADHD from EEG Brain Waves. Signals 2023, 4, 193–205. [Google Scholar] [CrossRef]
- Bansal, K.; Singh, A.; Juneja, A.; Chugh, N.; Balyan, A. Scale-Specific Dynamics of High-Amplitude Bursts in EEG Capture Behaviorally Meaningful Variability. NeuroImage 2021, 241, 118425. [Google Scholar] [CrossRef]


















| Field | Value |
|---|---|
| Rows | 2,166,383 |
| Subjects | 121 |
| ADHD subjects | 61 |
| Control subjects | 60 |
| EEG channels | 19 |
| Sampling frequency | 128 Hz |
| Documented age range | 7–12 years |
| Diagnostic framing | ADHD versus typically developing controls |
| Public redistribution | Kaggle EEG Dataset for ADHD |
| Original public-data trace | Nasrabadi, Allahverdy, Samavati, and Mohammadi, EEG data for ADHD/Control children, IEEE DataPort, DOI: 10.21227/rzfh-zn36 [28] |
| Ethics | Public anonymized dataset reused for secondary analysis; no new data collection in this notebook |
| Class_Name | n_Subjects | n_Rows |
|---|---|---|
| ADHD | 61 | 1,207,069 |
| Control | 60 | 959,314 |
| Parameter | Value |
|---|---|
| Sampling frequency | 128 Hz |
| Connectivity window | 512 samples (4.0 s) |
| Window step | 256 samples (0.512 s) |
| Sequence length | 10 windows |
| Sequence stride | 5 windows |
| Adjacency threshold | 0.40 |
| Subject aggregation | mean |
| Fixed decision threshold | 0.0 |
| Proposed-model GRU hidden size | 64 |
| Proposed-model attention heads | 4 |
| Proposed-model encoder layers | 1 |
| Proposed-model width d_model | 64 |
| Static GCN hidden size | 32 |
| Optimizer for proposed model | AdamW |
| Proposed-model learning rate | 4 × 10−4 |
| Batch size | 64 |
| Checkpoint criterion | Best validation child-level composite (0.55 × balanced accuracy + 0.45 × AUC) |
| Early-stopping patience | 5 epochs |
| Maximum epochs | 16 |
| Proposed-model dropout | 0.25 |
| Proposed-model weight decay | 3 × 10−4 |
| Reduced-regularization ablation dropout | 0.15 |
| Reduced-regularization ablation weight decay | 1 × 10−4 |
| Channel | Region | Theta_Beta_Ratio_d | Fusion_Strength_d | Combined_Rank_Score |
|---|---|---|---|---|
| O1 | occipital | −0.3432 | −0.6367 | 0.9799 |
| P7 | parietotemporal | −0.3407 | −0.5858 | 0.9265 |
| Fp1 | frontopolar | −0.3452 | −0.5411 | 0.8863 |
| P3 | parietal | −0.4077 | −0.3840 | 0.7917 |
| Fz | midline frontal | −0.3751 | −0.3720 | 0.7471 |
| T7 | temporal | −0.3739 | −0.3345 | 0.7085 |
| O2 | occipital | −0.3737 | −0.3307 | 0.7044 |
| Pz | midline parietal | −0.3113 | −0.2770 | 0.5883 |
| C3 | central | −0.3243 | −0.2087 | 0.5330 |
| F3 | frontal | −0.3490 | −0.1707 | 0.5197 |
| P4 | parietal | −0.3375 | 0.1665 | 0.5040 |
| P8 | parietotemporal | −0.2844 | −0.1967 | 0.4811 |
| F7 | frontotemporal | −0.2728 | −0.1878 | 0.4606 |
| Fp2 | frontopolar | −0.0433 | −0.4060 | 0.4493 |
| T8 | temporal | −0.3748 | −0.0205 | 0.3952 |
| C4 | central | −0.3701 | 0.0227 | 0.3928 |
| Cz | midline central | −0.2708 | 0.0958 | 0.3666 |
| F4 | frontal | −0.2863 | −0.0090 | 0.2953 |
| F8 | frontotemporal | −0.2245 | −0.0387 | 0.2633 |
| Model | N (Test) | Balanced Accuracy | 95% CI (BA) | AUC-ROC | 95% CI (AUC) | Sensitivity | 95% CI (Sens) | Specificity | 95% CI (Spec) |
|---|---|---|---|---|---|---|---|---|---|
| Proposed Hybrid ST Graph Transformer | 19 | 0.6222 | [0.411, 0.833] | 0.7444 | [0.489, 0.944] | 0.8 | [0.500, 1.000] | 0.4444 | [0.111, 0.778] |
| Proposed model ablation | 19 | 0.6222 | [0.411, 0.833] | 0.7333 | [0.478, 0.944] | 0.8 | [0.500, 1.000] | 0.4444 | [0.111, 0.778] |
| GRU dynamic index | 19 | 0.5722 | [0.356, 0.789] | 0.6556 | [0.378, 0.889] | 0.7 | [0.400, 1.000] | 0.4444 | [0.111, 0.778] |
| Global graph-metric index | 19 | 0.5667 | [0.361, 0.778] | 0.6333 | [0.356, 0.867] | 0.8 | [0.500, 1.000] | 0.3333 | [0.000, 0.667] |
| Linear oscillatory index | 19 | 0.5167 | [0.306, 0.728] | 0.4667 | [0.189, 0.744] | 0.7 | [0.400, 1.000] | 0.3333 | [0.000, 0.667] |
| Static GCN index | 19 | 0.5 | [0.500, 0.500] | 0.4 | [0.156, 0.678] | 1 | [1.000, 1.000] | 0 | [0.000, 0.000] |
| Sparse oscillatory index | 19 | 0.4667 | [0.256, 0.678] | 0.4667 | [0.200, 0.744] | 0.6 | [0.300, 0.900] | 0.3333 | [0.111, 0.667] |
| Modeling Variant | Overall Accuracy | Balanced Accuracy | AUC-ROC | Distribution Overlap (Frequency-Specific Representation) |
|---|---|---|---|---|
| Theta Connectivity Only | 0.6316 | 0.6278 | 0.7333 | 0.6785 |
| Beta Connectivity Only | 0.6316 | 0.6278 | 0.5667 | 0.9346 |
| Joint Theta + Beta Connectivity | 0.6316 | 0.6222 | 0.7444 | 0.7155 |
| Range | Interchannel Edge | ADHD Control Difference | Absolute Shift |
|---|---|---|---|
| Beta | C4-P4 | 0.1674 | 0.1674 |
| Beta | P4-Cz | 0.1318 | 0.1318 |
| Beta | T8-P8 | 0.1217 | 0.1217 |
| Beta | F3-Fz | 0.1158 | 0.1158 |
| Beta | Cz-Pz | 0.1057 | 0.1057 |
| Beta | O1-P7 | −0.1034 | 0.1034 |
| Beta | P4-T8 | 0.1013 | 0.1013 |
| Beta | F4-F8 | 0.0977 | 0.0977 |
| Theta | Fp1-O1 | −0.1298 | 0.1298 |
| Theta | Fp1-P7 | −0.1292 | 0.1292 |
| Theta | P7-Fz | −0.1262 | 0.1262 |
| Theta | O1-Fz | −0.123 | 0.123 |
| Theta | F3-O1 | −0.1217 | 0.1217 |
| Theta | Fp1-P3 | −0.1189 | 0.1189 |
| Theta | O1-P7 | −0.1161 | 0.1161 |
| Theta | F3-C3 | −0.1099 | 0.1099 |
| Clinical Indicator | Value |
|---|---|
| Number of children in the test sample | 19 |
| Distributed network index overlap | 0.45 |
| Sensitivity | 0.8 |
| Specificity | 0.4444 |
| Positive predictive value in the test sample | 0.6154 |
| Balanced accuracy | 0.6222 |
| Display_Model | Accuracy | Balanced_Accuracy | Sensitivity | Specificity | Precision | f1 | auc_roc | Mean_Metric_Rank | Cohens_d | Distribution_Overlap |
|---|---|---|---|---|---|---|---|---|---|---|
| Proposed Hybrid ST Graph Transformer | 0.6316 | 0.6222 | 0.8000 | 0.4444 | 0.6154 | 0.6957 | 0.7444 | 1 | 0.7289 | 0.45 |
| Proposed model ablation | 0.6316 | 0.6222 | 0.8000 | 0.4444 | 0.6154 | 0.6957 | 0.7333 | 1.3333 | 0.6649 | 0.7395 |
| GRU dynamic index | 0.5789 | 0.5722 | 0.7000 | 0.4444 | 0.5833 | 0.6364 | 0.6556 | 2.3333 | 0.6638 | 0.7400 |
| Global graph-metric index | 0.5789 | 0.5667 | 0.8000 | 0.3333 | 0.5714 | 0.6667 | 0.6333 | 3 | 0.5515 | 0.7827 |
| Linear oscillatory index | 0.5263 | 0.5167 | 0.7000 | 0.3333 | 0.5385 | 0.6087 | 0.4667 | 4 | 0.0405 | 0.9839 |
| Static GCN index | 0.5263 | 0.5000 | 1 | 0.0000 | 0.5263 | 0.6897 | 0.4000 | 4.6667 | −0.4990 | 0.8030 |
| Sparse oscillatory index | 0.4737 | 0.4667 | 0.6000 | 0.3333 | 0.5000 | 0.5455 | 0.4667 | 5 | 0.0923 | 0.9632 |
| Display_Model | Accuracy | Balanced_Accuracy | Sensitivity | Specificity | f1 | auc_roc | Mean_Metric_Rank |
|---|---|---|---|---|---|---|---|
| Proposed Hybrid ST Graph Transformer | 0.6316 | 0.6222 | 0.8000 | 0.4444 | 0.6957 | 0.7444 | 1 |
| Proposed model ablation | 0.6316 | 0.6222 | 0.8000 | 0.4444 | 0.6957 | 0.7333 | 1.3333 |
| Model | Mean0 | Mean1 | Std0 | Std1 | Δmean | Separation | Cohen’s d | Pearson r | Overlap | Silhouette |
|---|---|---|---|---|---|---|---|---|---|---|
| Linear_network_index | −3.40 | 4.40 | 7.26 | 4.08 | 7.80 | 0.69 | 1.32 | 0.55 | 0.51 | 0.25 |
| Sparse_linear_network_index | −0.80 | 2.22 | 2.77 | 2.57 | 3.02 | 0.57 | 1.13 | 0.49 | 0.57 | 0.14 |
| Global_graph_metric_network_index | −0.07 | 0.07 | 0.38 | 0.43 | 0.14 | 0.17 | 0.34 | 0.17 | 0.87 | 0.05 |
| RNN_dynamic_connectivity_index | −3.45 | 4.42 | 6.23 | 5.39 | 7.87 | 0.68 | 1.35 | 0.56 | 0.50 | 0.31 |
| Static_GCN_brain_network_index | −0.03 | −0.03 | 0.00 | 0.00 | 0.00 | 0.21 | 0.42 | 0.21 | 0.83 | 0.05 |
| Spatio_temporal_GCN_brain_network_index | 0.10 | 0.22 | 0.25 | 0.27 | 0.12 | 0.23 | 0.47 | 0.23 | 0.81 | 0.05 |
| Multihead_spatio_temporal_graph_transformer_index | −2.91 | 3.30 | 4.28 | 3.97 | 6.21 | 0.75 | 1.51 | 0.60 | 0.45 | 0.32 |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Share and Cite
Baibulova, M.; Mukhanova, A.; Abdukarimova, A.; Abdykerimova, L.; Serimbetov, B.; Akhmetzhanov, M.; Seitakhmetova, Z.; Yeshtayeva, E.; Kassim, M.; Amirbay, A. A Hybrid Spatio-Temporal Graph Transformer for EEG-Based ADHD Detection via Network Index Modeling. Computers 2026, 15, 333. https://doi.org/10.3390/computers15060333
Baibulova M, Mukhanova A, Abdukarimova A, Abdykerimova L, Serimbetov B, Akhmetzhanov M, Seitakhmetova Z, Yeshtayeva E, Kassim M, Amirbay A. A Hybrid Spatio-Temporal Graph Transformer for EEG-Based ADHD Detection via Network Index Modeling. Computers. 2026; 15(6):333. https://doi.org/10.3390/computers15060333
Chicago/Turabian StyleBaibulova, Makbal, Ayagoz Mukhanova, Aliya Abdukarimova, Lazzat Abdykerimova, Bulat Serimbetov, Madi Akhmetzhanov, Zhanat Seitakhmetova, Elmira Yeshtayeva, Murizah Kassim, and Aizat Amirbay. 2026. "A Hybrid Spatio-Temporal Graph Transformer for EEG-Based ADHD Detection via Network Index Modeling" Computers 15, no. 6: 333. https://doi.org/10.3390/computers15060333
APA StyleBaibulova, M., Mukhanova, A., Abdukarimova, A., Abdykerimova, L., Serimbetov, B., Akhmetzhanov, M., Seitakhmetova, Z., Yeshtayeva, E., Kassim, M., & Amirbay, A. (2026). A Hybrid Spatio-Temporal Graph Transformer for EEG-Based ADHD Detection via Network Index Modeling. Computers, 15(6), 333. https://doi.org/10.3390/computers15060333

