arXiv:2602.00670cs.LGeess.SP2026-02

用机器学习对脑电数据做三类情绪分类,随机森林效果最佳。

Three-Way Emotion Classification of EEG-based Signals using Machine Learning

  • 采用逻辑回归、SVM和随机森林三种模型对比分类
  • 随机森林准确率与F1分数最高,优于其他模型
  • 适合想了解脑电情绪识别的初学者或应用开发者

脑电图(EEG)是测量脑活动的常用技术,其信号可反映不同脑区的活动状态,从而揭示人的心理情绪。本文研究如何利用机器学习(ML)模型将有限的脑电信号数据集分为三类情绪:负面、中性或正面。研究展示了完整的处理流程,包括数据预处理和多种机器学习模型的比较。为评估哪种模型更适合此类任务,训练并测试了三种常用模型:逻辑回归(LR)、支持向量机(SVM)和随机森林(RF),并以准确率和F1分数作为评价指标。结果表明,机器学习模型能有效用于脑电信号的三分类情绪识别。在所用数据集上,随机森林模型表现最佳,其更高的准确率和F1分数说明其更有效地捕捉了情绪模式。此外,该模型在准确率方面也优于现有先进分类模型。

原文摘要 · Abstract (English)

Electroencephalography (EEG) is a widely used technique for measuring brain activity. EEG-based signals can reveal a persons emotional state, as they directly reflect activity in different brain regions. Emotion-aware systems and EEG-based emotion recognition are a growing research area. This paper presents how machine learning (ML) models categorize a limited dataset of EEG signals into three different classes, namely Negative, Neutral, or Positive. It also presents the complete workflow, including data preprocessing and comparison of ML models. To understand which ML classification model works best for this kind of problem, we train and test the following three commonly used models: logistic regression (LR), support vector machine (SVM), and random forest (RF). The performance of each is evaluated with respect to accuracy and F1-score. The results indicate that ML models can be effectively utilized for three-way emotion classification of EEG signals. Among the three ML models trained on the available dataset, the RF model gave the best results. Its higher accuracy and F1-score suggest that it is able to capture the emotional patterns more accurately and effectively than the other two models. The RF model also outperformed the existing state-of-the-art classification models in terms of the accuracy parameter.

情绪识别脑电分析机器学习分类模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。