Deep Learning Based Classification and Combined Transform Based Feature Extraction Approach for Mental Stress Prediction of Human Beings Using EEG

Abstract

Stress is a psychological condition in which a person feels overwhelmed with pressure. Early identification of psychological stress is critical for preventing illness progression and saving lives. Electroencephalography (EEG) is often used to collect psychological information such as brain rhythms in the form of electric waves. Traditional deep learning techniques face limitations like temporal dynamics and feature extraction issues. To address these shortcomings, a deep learning-based classification model was created, combining advanced transform-based feature extraction techniques to more effectively predict mental stress by using EEG signals. The process begins by utilizing physiological parameters extracted from the EEG Psychiatric Disorders Dataset. The raw EEG signals undergo pre-processing to enhance their quality, which includes smoothing, alignment, and addressing non-uniform sampling issues. The signals are then decomposed and their components extracted using the Adaptive Flexible Analytic Wavelet Transform (AFAWT). Short-Term Fourier Transform-Randon Transform (STFT-RT) approach is used to extract the key features of signals. Feature selection is optimized using the Young's Double Slit Experiment Optimizer (YDSE) to ensure only the most relevant features are chosen for classification. Finally, these selected features are fed into the Parallel Neural Networks with Extreme Efficiency (ParNeXt v1-DB) model, which utilizes a drop block mechanism to enhance model generalization and prevent overfitting, ensuring highly effective mental stress prediction. According to simulated research, the proposed approach demonstrated significant improvements over existing algorithms. In Dataset 1, the method achieved an accuracy of 97.8%, selectivity of 95.4%, while Dataset 2 recorded an accuracy of 96.3%, PPV of 93.8%. Thus, the proposed method is the most effective method for predicting human mental stress using EEG.