This study proposes Contrastive Multi-scale EEGNet (CM-EEGNet) for electroencephalography (EEG)-based classification of Alzheimer’s disease (AD) and cognitively normal (CN) subjects, as well as channel-level contribution analysis. EEG is a useful no...
This study proposes Contrastive Multi-scale EEGNet (CM-EEGNet) for electroencephalography (EEG)-based classification of Alzheimer’s disease (AD) and cognitively normal (CN) subjects, as well as channel-level contribution analysis. EEG is a useful non-invasive signal for assisting dementia diagnosis because it reflects functional brain activity with high temporal resolution. However, EEG-based classification remains challenging due to signal nonstationarity, noise sensitivity, and inter-subject variability. To address these issues, the proposed CM-EEGNet extends the conventional EEGNet structure by applying multi-scale temporal convolutions to capture EEG patterns at different time scales and depthwise spatial convolutions to learn inter-channel spatial relationships.
The proposed model consists of an encoder, a projection head, and a classification head. The classification head predicts AD and CN classes, while the projection head generates embeddings for supervised contrastive learning. The model is trained using a joint loss that combines cross-entropy loss and supervised contrastive loss, thereby improving both classification performance and class separability in the embedding space. Experiments were conducted using the OpenNeuro ADFTD EEG dataset. The frontotemporal dementia group was excluded, and only AD and CN subjects were used for binary classification. A subject-independent 60:20:20 split was applied to prevent subject leakage, and 19-channel EEG signals were segmented into non-overlapping 8-second epochs.
In the subject-independent test set, CM-EEGNet achieved a segment-level accuracy of 97.30%, with F1-scores of 97.22% for AD and 97.37% for CN. The ablation study showed that the joint loss setting with =0.5 and =0.5 produced the best performance. In addition, t-SNE visualization confirmed that the learned encoder features and projection embeddings showed separable distributions between AD and CN. For interpretability, this study applied Contrastive Prototype Attribution (CPA), which estimates channel-wise contributions by measuring changes in contrastive evidence scores after channel masking. The CPA results showed that O2, F7, and P3 contributed mainly to AD classification, while T5, O2, and F4 contributed mainly to CN classification. These findings indicate that the proposed method can not only improve EEG-based AD-CN classification performance but also provide an interpretable channel-level basis for model decisions in the contrastive embedding space.