M-LINKX: Multiview Graph Learning for Brain Cognitive Disease Detection
Organizations: Dept. of Electrical Engineering and Computer Science, Florida Atlantic University, Boca Raton, FL 33431, USA
Abstract
Electroencephalogram (EEG) is a non-invasive and relatively low-cost procedure that measures brain electricity for the detection of cognitive diseases. EEG-based classification of dementia-related conditions, including Alzheimer's disease (AD), mild cognitive impairment (MCI), and frontotemporal dementia (FTD), remains challenging because EEG signals are noisy, non-stationary, and vary across subjects. Segment-based learning provides a practical way to model long EEG recordings by converting them into fixed-length inputs. For each segment, discriminative information may be explored by using signals within each channel (i.e. electrode), as well as interactions between EEG channels. In this paper, we propose M-LINKX, a multi-view graph learning framework for EEG-based dementia classification. For each segment, we extract channel-level node features and construct multiple functional-connectivity (FC) graph views, where each view is defined by a specific combination of connectivity metric, frequency band, and topology filter, respectively. Instead of relying on message passing over the constructed graphs, M-LINKX follows a simple design in modeling node features and adjacency-based connectivity representations. The graph-view representations are fused using global trainable view weights, and subject-level prediction is obtained by averaging segment-level probabilities. Experiments on two three-class EEG datasets with different diagnostic groups, CAUEEG (HC/MCI/Dementia) and AHEAP (HC/AD/FTD), show that M-LINKX achieves the best subject-level performance under the main experimental settings. Our study suggests that multi-view functional connectivity can improve EEG-based dementia classification when integrated with an appropriate graph-learning architecture. Code and data are available at https://github.com/anphantt/MLINKX.
Figures & tables
| Property | AHEAP | CAUEEG |
|---|---|---|
| # of Subjects | 36 AD, 23 FTD, 29 HC | 291 Dementia, 395 MCI, 436 HC |
| # of Channels | 19 | 19 |
| Condition | Eyes closed | Eyes closed and eyes open |
| Sampling Rate | 500 Hz | 200 Hz |
| Avg. Duration | 13.21 2.32 min | 12.80 2.65 min |
| CAUEEG (10-second window) | AHEAP (4-second window) | ||||||
|---|---|---|---|---|---|---|---|
| 35 HC, 33 MCI, 22 Dementia | 29 HC, 23 FTD, 36 AD | ||||||
| Model | RawEEG | Node features | FC input | Balanced Acc. | Macro-F1 | Balanced Acc. | Macro-F1 |
| 1D ResNet | ✓ | 0.6146 0.0135 | 0.6224 0.0141 | 0.5221 0.0432 | 0.4815 0.0511 | ||
| CNN-LSTM | ✓ | 0.5700 0.0133 | 0.5585 0.0049 | 0.5697 0.0527 | 0.5494 0.0438 | ||
| MLP-node | ✓ | 0.5949 0.0233 | 0.5961 0.0199 | 0.5740 0.0477 | 0.5550 0.0446 | ||
| Ensemble | ✓ | ✓ | 0.6084 0.0285 | 0.6048 0.0303 | 0.5791 0.0258 | 0.5528 0.0273 | |
Electroencephalography
Electroencephalography (EEG) studies brain-generated electrical activity to understand cognition and support diagnosis, monitoring, and brain-computer interfaces.
See all 180 papers