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 条
  • [1] EEG-based Evaluation of Mental Fatigue Using Machine Learning Algorithms
    Liu, Yisi
    Lan, Zirui
    Khoo, Han Hua Glenn
    Li, King Ho Holden
    Sourina, Olga
    Mueller-Wittig, Wolfgang
    2018 INTERNATIONAL CONFERENCE ON CYBERWORLDS (CW), 2018, : 276 - 279
  • [2] Classification of mental tasks from EEG signals using extreme learning machine
    Liang, NY
    Saratchandran, P
    Huang, GB
    Sundararajan, N
    INTERNATIONAL JOURNAL OF NEURAL SYSTEMS, 2006, 16 (01) : 29 - 38
  • [3] Mental Workload Estimation from EEG Signals Using Machine Learning Algorithms
    Cheema, Baljeet Singh
    Samima, Shabnam
    Sarma, Monalisa
    Samanta, Debasis
    ENGINEERING PSYCHOLOGY AND COGNITIVE ERGONOMICS (EPCE 2018), 2018, 10906 : 265 - 284
  • [4] Emotion State Detection Using EEG Signals-A Machine Learning Perspective
    Naidu, P. Geethika
    Adhitya, C. M. Jayanth
    Harshita, S.
    Bashpika, T.
    Manikumar, V. S. S. S. R.
    Muthulakshmi, M.
    SMART TRENDS IN COMPUTING AND COMMUNICATIONS, VOL 3, SMARTCOM 2024, 2024, 947 : 471 - 481
  • [5] Evaluation of Machine Learning Algorithms for Classification of EEG Signals
    Javier Ramirez-Arias, Francisco
    Efren Garcia-Guerrero, Enrique
    Tlelo-Cuautle, Esteban
    Miguel Colores-Vargas, Juan
    Garcia-Canseco, Eloisa
    Roberto Lopez-Bonilla, Oscar
    Manuel Galindo-Aldana, Gilberto
    Inzunza-Gonzalez, Everardo
    TECHNOLOGIES, 2022, 10 (04)
  • [6] An Online Teaching Video Evaluation Scheme Based on EEG Signals and Machine Learning
    Huang, Haiping
    Han, Gaorong
    Xiao, Fu
    Wang, Rui
    WIRELESS COMMUNICATIONS & MOBILE COMPUTING, 2022, 2022
  • [7] FPGA implementation of automatic seizure detection in EEG signals using machine learning algorithm
    Sajja, Amrita
    Rooban, S.
    DISCOVER APPLIED SCIENCES, 2024, 6 (08)
  • [8] Emotion Recognition with Machine Learning Using EEG Signals
    Bazgir, Omid
    Mohammadi, Zeynab
    Habibi, Seyed Amir Hassan
    2018 25TH IRANIAN CONFERENCE ON BIOMEDICAL ENGINEERING AND 2018 3RD INTERNATIONAL IRANIAN CONFERENCE ON BIOMEDICAL ENGINEERING (ICBME), 2018, : 149 - 153
  • [9] Machine-Learning-Based Emotion Recognition System Using EEG Signals
    Alhalaseh, Rania
    Alasasfeh, Suzan
    COMPUTERS, 2020, 9 (04) : 1 - 15
  • [10] Classification of EEG signals using a genetic-based machine learning classifier
    Skinner, B. T.
    Nguyen, H. T.
    Liu, D. K.
    2007 ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE AND BIOLOGY SOCIETY, VOLS 1-16, 2007, : 3120 - 3123