نوع مقاله : مقاله کامل پژوهشی
نویسندگان
گروه مهندسی پزشکی، دانشکده مهندسی برق و کامپیوتر، دانشگاه صنعتی نوشیروانی بابل، بابل، ایران
کلیدواژهها
موضوعات
عنوان مقاله English
نویسندگان English
Approximately one-third of human life is spent in sleep, and sleep disorders can significantly affect quality of life and cognitive performance; therefore, accurate sleep stage detection is of particular importance. In this study, a lightweight parallel hybrid model is proposed for automatic sleep stage classification using EEG signals. In this model, features obtained by applying the multi-head attention mechanism (MHA) to the spectral content of EEG signals were combined with features extracted from one-dimensional convolutional layers (1D-CNN). Combining the spectral features improved by MHA with temporal features extracted from CNN resulted in a significant increase in classification accuracy. The proposed model achieved an overall accuracy of 81% in sleep stage classification on the Sleep_Edf_20 dataset, indicating its competitive performance over heavier architectures. To demonstrate the superiority of the proposed model, the one-dimensional convolutional neural network was combined with the multi-head attention mechanism in three different structures. The results showed that the proposed structure with a smaller number of parameters has greater accuracy in classifying sleep stages. In addition, to deal with the problem of data imbalance between different classes of sleep stages, a weighting method was used in the cost function, which provided a significant improvement in identifying undersampled stages, especially stage1 of sleep.
کلیدواژهها English