Evaluation of Mental State Based on EEG Signals Using Machine Learning Algorithm

被引:0
|
作者
Duta, Stefana [1 ]
Sultana, Alina Elena [1 ]
Banica, Cosmin Karl [2 ]
机构
[1] Natl Univ Sci & Technol, UNSTPB, Appl Elect & Informat Engn, Politehn Bucharest, Bucharest, Romania
[2] Wing Comp Grp SRL, Bucharest, Romania
关键词
Depression; EEG; Multilayer Perceptron; Features;
D O I
10.1007/978-3-031-62520-6_27
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
This paper presents a comprehensive analysis of Multilayer Perceptron (MLP) models for the classification of EEG signals in the context of depression state detection. Experiments were conducted using two separate databases: the Depression Rest Database and the MDD vs. Control Database. For the Depression Rest Database, the MLP model reached an accuracy of 84.65% on the training set but faced challenges with validation, plateauing at 68.79%. Conversely, the MLP model excelled in the MDD vs. Control Database, achieving an accuracy of 89.99% on the training data and 88.97% on the validation data. It displayed high precision and recall values for both healthy and depressed classes, indicating a balanced performance. Additionally, feature selection was explored on a combined database, yielding promising results but with room for further optimizations. The novelty of this study lies in its investigation into whether the combination of two datasets, both oriented toward the common objective of depression, demonstrates superior performance compared to the individual analyses conducted on each dataset.
引用
收藏
页码:230 / 239
页数:10
相关论文
共 50 条
  • [31] Machine Learning for Predicting Epileptic Seizures Using EEG Signals: A Review
    Rasheed, Khansa
    Qayyum, Adnan
    Qadir, Junaid
    Sivathamboo, Shobi
    Kwan, Patrick
    Kuhlmann, Levin
    O'Brien, Terence
    Razi, Adeel
    IEEE REVIEWS IN BIOMEDICAL ENGINEERING, 2021, 14 : 139 - 155
  • [32] Discrimination of Depression Levels Using Machine Learning Methods on EEG Signals
    Mohammadi, Yousef
    Hajian, Mojtaba
    Moradi, Mohammad Hassan
    2019 27TH IRANIAN CONFERENCE ON ELECTRICAL ENGINEERING (ICEE 2019), 2019, : 1765 - 1769
  • [33] An Interpretable Machine Learning Method for the Detection of Schizophrenia Using EEG Signals
    Vazquez, Manuel A.
    Maghsoudi, Arash
    Marino, Ines P.
    FRONTIERS IN SYSTEMS NEUROSCIENCE, 2021, 15
  • [34] A machine learning approach using EEG signals to measure sleep quality
    Ravan M.
    AIMS Electronics and Electrical Engineering, 2019, 3 (04): : 347 - 358
  • [35] EEG Signals Classification Using Machine Learning for The Identification and Diagnosis of Schizophrenia
    Zhang, Lei
    2019 41ST ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE AND BIOLOGY SOCIETY (EMBC), 2019, : 4521 - 4524
  • [36] Cognitive Workload Recognition Using EEG Signals and Machine Learning: A Review
    Zhou, Yueying
    Huang, Shuo
    Xu, Ziming
    Wang, Pengpai
    Wu, Xia
    Zhang, Daoqiang
    IEEE TRANSACTIONS ON COGNITIVE AND DEVELOPMENTAL SYSTEMS, 2022, 14 (03) : 799 - 818
  • [37] Classification of Epileptic EEG Signals Using Synchrosqueezing Transform and Machine Learning
    Cura, Ozlem Karabiber
    Akan, Aydin
    INTERNATIONAL JOURNAL OF NEURAL SYSTEMS, 2021, 31 (05)
  • [38] Machine Learning Approach to Measure Sleep Quality using EEG Signals
    Ravan, M.
    2019 IEEE SIGNAL PROCESSING IN MEDICINE AND BIOLOGY SYMPOSIUM (SPMB), 2019,
  • [39] Using machine learning to reveal the population vector from EEG signals
    Kobler, Reinmar J.
    Almeida, Ines
    Sburlea, Andreea, I
    Mueller-Putz, Gernot R.
    JOURNAL OF NEURAL ENGINEERING, 2020, 17 (02)
  • [40] Prediction of Sleep Apnea Using EEG Signals and Machine Learning Algorithms
    Onargan, Aysu
    Gavcar, Busra
    Caliskan, Gulizar
    Akan, Aydin
    TIP TEKNOLOJILERI KONGRESI (TIPTEKNO'21), 2021,