Affiliation:
1. Information Technology Department, College of Computer and Information Sciences, King Saud University, Riyadh, Saudi Arabia
2. Computer Science Department, College of Computer and Information Sciences, Imam Mohammad Ibn Saud Islamic University (IMSIU), Riyadh, Saudi Arabia
Abstract
The performance of electroencephalogram (EEG)-based systems depends on the proper choice of feature extraction and machine learning algorithms. This study highlights the significance of selecting appropriate feature extraction and machine learning algorithms for EEG-based anxiety detection. We explored different annotation/labeling, feature extraction, and classification algorithms. Two measurements, the Hamilton anxiety rating scale (HAM-A) and self-assessment Manikin (SAM), were used to label anxiety states. For EEG feature extraction, we employed the discrete wavelet transform (DWT) and power spectral density (PSD). To improve the accuracy of anxiety detection, we compared ensemble learning methods such as random forest (RF), AdaBoost bagging, and gradient bagging with conventional classification algorithms including linear discriminant analysis (LDA), support vector machine (SVM), and k-nearest neighbor (KNN) classifiers. We also evaluated the performance of the classifiers using different labeling (SAM and HAM-A) and feature extraction algorithms (PSD and DWT). Our findings demonstrated that HAM-A labeling and DWT-based features consistently yielded superior results across all classifiers. Specifically, the RF classifier achieved the highest accuracy of 87.5%, followed by the Ada boost bagging classifier with an accuracy of 79%. The RF classifier outperformed other classifiers in terms of accuracy, precision, and recall.
Funder
Deanship of Scientific Research
Reference24 articles.
1. Predict students’ attention in online learning using EEG data;Al-Nafjan;Sustainability,2022
2. Classification of human emotions from electroencephalogram (EEG) signal using deep neural network;Al-Nafjan;International Journal of Advanced Computer Science and Applications,2017
3. Recognition of consumer preference by analysis and classification EEG signals;Aldayel;Frontiers in Human Neuroscience,2021
4. DASPS: a database for anxious states based on a psychological stimulation;Baghdadi,2019
5. Psychological stimulation for anxious states detection based on EEG-related features;Baghdadi;Journal of Ambient Intelligence and Humanized Computing,2021