EEG datasets: TUAB Abnormal EEG Corpus (2,993 recordings from 2,329 unique patients, age 7 days to 96 years) and NUST-MH-TUKL (NMT) scalp EEG dataset (2,417 recordings from unique participants, South Asian population).
Cross-dataset transfer learning using deep learning models (TCN, Deep4Net, ShallowNet, EEGNet) with discriminative fine-tuning
Baseline models trained from scratch on the target dataset (NMT) or trained on concatenated datasets
Balanced accuracy (BAC) for pathology classification (normal vs abnormal)
Cross-dataset transfer learning with discriminative fine-tuning improves the performance of deep learning models for EEG pathology detection, especially in low-data regimes.
Pathology diagnosis based on EEG signals and decoding brain activity holds immense importance in understanding neurological disorders. With the advancement of artificial intelligence methods and machine learning techniques, the potential for accurate data-driven diagnoses and effective treatments has grown significantly. However, applying machine learning algorithms to real-world datasets presents diverse challenges at multiple levels. The scarcity of labeled data, especially in low regime scenarios with limited availability of real patient cohorts due to high costs of recruitment, underscores the vital deployment of scaling and transfer learning techniques. In this study, we explore a real-world pathology classification task to highlight the effectiveness of data and model scaling and cross-dataset knowledge transfer. As such, we observe varying performance improvements through data scaling, indicating the need for careful evaluation and labeling. Additionally, we identify the challenges of possible negative transfer and emphasize the significance of some key components to overcome distribution shifts and potential spurious correlations and achieve positive transfer. We see improvement in the performance of the target model on the target (NMT) datasets by using the knowledge from the source dataset (TUAB) when a low amount of labeled data was available. Our findings demonstrated that a small and generic model (e.g. ShallowNet) performs well on a single dataset, however, a larger model (e.g. TCN) performs better in transfer learning when leveraging a larger and more diverse dataset.
Building similarity graph...
Analyzing shared references across papers
Loading...
Darvishi-Bayazi et al. (Sat,) studied this question.
synapsesocial.com/papers/6a10a3e001be78fe816126f5 — DOI: https://doi.org/10.1016/j.compbiomed.2023.107893
Mohammad-Javad Darvishi-Bayazi
L'Alliance Boviteq
Mohammad Sajjad Ghaemi
National Research Council Canada
Timothée Lesort
Institut national de recherche en sciences et technologies du numérique
Computers in Biology and Medicine
Université de Montréal
National Research Council Canada
Mila - Quebec Artificial Intelligence Institute
Building similarity graph...
Analyzing shared references across papers
Loading...