Classificação de emoções em sinais de EEG utilizando técnicas de aprendizado de máquina

Abstract

Emotions play a fundamental role in human daily lives, having influences over decisions and even communications. Understanding how emotions are characterized and how they can be identified are of utmost importance to understand how humans behave. Various methods have been proposed for emotion classification, however researches that correlates discrete emotions and brain activity patterns are still being investigated. Thus this master’s dissertation presents a methodology for discrete emotion classification in EEG signals using machine learning algorithms.During the development of this work various models have been created using machine learning algorithms where accuracy metrics have been noted to be compared and thus select the best classification model. For the development of the models, the DEAP dataset was used. Once the dataset was unbalanced, the balancing algorithms SMOTE and ADASYN were evaluated in their capacity to balance data and contribute to the improvement of the classification models.The main contributions of this dissertation are: the development of the discrete emotion classification models in EEG signals and the evaluation of the machine learning and the data balancing algorithms.After developed the models, the results showed that the Random Forest model with the ADASYN algorithm for data balancing had the best results with an average accuracy of 89.22%. The models built with the SVM algorithm and the ADASYN technique for balancing the data also showed good results with an average accuracy of 87.36%. Finally, the models with Neural Networks and SMOTE algorithms showed the worst results with an average accuracy of 68.56%.Therefore, the results showed that for the discrete emotion classification in EEG signals, the models using the ADASYN algorithm for data balancing with the Random Forest algorithm for classification were superior when compared to the other tested models.