The role of data partitioning on the performance of EEG-based deep learning models in supervised cross-subject analysis: a preliminary study

TL;DR

This study evaluates how different data partitioning and cross-validation strategies affect the performance and reliability of EEG-based deep learning models across various neurological and BCI tasks, emphasizing best practices to prevent data leakage.

Contribution

It provides a comprehensive comparison of cross-validation methods in EEG deep learning, offering guidelines to improve model evaluation and prevent data leakage.

Findings

Subject-based cross-validation is essential for reliable evaluation.

Nested cross-validation approaches are more reliable than non-nested methods.

Non-nested methods tend to overfit and favor larger models.

Abstract

Deep learning is significantly advancing the analysis of electroencephalography (EEG) data by effectively discovering highly nonlinear patterns within the signals. Data partitioning and cross-validation are crucial for assessing model performance and ensuring study comparability, as they can produce varied results and data leakage due to specific signal properties (e.g., biometric). Such variability leads to incomparable studies and, increasingly, overestimated performance claims, which are detrimental to the field. Nevertheless, no comprehensive guidelines for proper data partitioning and cross-validation exist in the domain, nor is there a quantitative evaluation of their impact on model accuracy, reliability, and generalizability. To assist researchers in identifying optimal experimental strategies, this paper thoroughly investigates the role of data partitioning and cross-validation in evaluating EEG deep learning models. Five cross-validation settings are compared across three supervised cross-subject classification tasks (BCI, Parkinson's, and Alzheimer's disease detection) and four established architectures of increasing complexity (ShallowConvNet, EEGNet, DeepConvNet, and Temporal-based ResNet). The comparison of over 100,000 trained models underscores, first, the importance of using subject-based cross-validation strategies for evaluating EEG deep learning models, except when within-subject analyses are acceptable (e.g., BCI). Second, it highlights the greater reliability of nested approaches (N-LNSO) compared to non-nested counterparts, which are prone to data leakage and favor larger models overfitting to validation data. In conclusion, this work provides EEG deep learning researchers with an analysis of data partitioning and cross-validation and offers guidelines to avoid data leakage, currently undermining the domain with potentially overestimated performance claims.