Enhanced EEG-Based Mental State Classification : A novel approach to eliminate data leakage and improve training optimization for Machine Learning
Abstract
In this paper, we explore prior research and introduce a new methodology for classifying mental state levels based on EEG signals utilizing machine learning (ML). Our method proposes an optimized training method by introducing a validation set and a refined standardization process to rectify data leakage shortcomings observed in preceding studies. Furthermore, we establish novel benchmark figures for various models, including random forest and deep neural networks.
- Publication:
-
arXiv e-prints
- Pub Date:
- December 2023
- DOI:
- arXiv:
- arXiv:2312.09379
- Bibcode:
- 2023arXiv231209379G
- Keywords:
-
- Electrical Engineering and Systems Science - Signal Processing;
- I.2.0;
- J.3
- E-Print:
- 5 pages, 2 figures, 1 table