Less is More: Reducing Overfitting in Deep Learning for EEG Classification.

Songchi Zhou,Shijia Geng,Jun Li,Deyun Zhang,Ziqian Xie,Chuandong Cheng,Shenda Hong
DOI: https://doi.org/10.22489/CinC.2023.183
2023-01-01
Abstract:Although most of the patients' recordings includes large scale long-term physiological time series, the patient-level quantity is relatively small, posing great challenges for machine learning methods. As part of the George B. Moody PhysioNet Challenge 2023, we aim to propose a series of Reducing Overfitting techniques in Deep learning for EEG (coined as RODE) in this scenario, for neurological recovery prognosis. RODE is a simple yet effective machine-learning method, which is mostly powered by generalizable deep-learning features. Specifically, we first pre-train a convolutional neural network on the segment level with a margin-based and mining-based loss, and extract deep features from it. Then we decrease the deep features' complexity using dimensionality reduction methods, which prove to be quite significant for reducing overfitting in deep learning. Finally, we combine the reduced features with static features in patient level, and put them into an ensemble model for classification. Our team, PKU_NIHDS, receives a Challenge score of 0.821 on the hidden validation set and 0.708 on the hidden test set.
What problem does this paper attempt to address?