基于树模型机器学习的皮肤电信号情绪识别
GSR Signal Emotion Recognition Based on Tree Model Machine Learning
-
摘要: 为了提高维度模型生理信号情绪识别准确率和泛化能力,本文基于DEAP维度情感生理数据集,提取皮肤电信号时域统计特征、功率谱特征、小波包熵特征,分别采用决策树和随机森林算法在唤醒度和效价两个情感维度进行情绪分类.通过选取合适的维度情感标签阈值,有效提高机器情绪识别准确率和稳健性;利用情绪诱发状态与自然状态下的皮肤电信号的差值进行归一化处理,消除个体差异,提高模型的泛化能力;采用多特征融合并基于集成学习的随机森林算法,获得更好的情绪识别性能.在唤醒度维度上的分类准确率Acc和F1值分别为92.0%和0.933,在效价维度上的分类准确率Acc和F1值分别为90.9%和0.925.仿真实验表明,基于树模型的机器学习方法可以实现维度情绪的准确识别,该研究可用于可穿戴设备生理信号情绪自动分析和机器识别.Abstract: In order to improve the accuracy and generalization ability of dimension model physiological signal emotion recognition, this paper extracts the temporal statistical features, power spectral features and wavelet packet entropy features of GSR signal based on the DEAP dimension emotional physiological data set. Decision tree and random forest algorithm were used to classify emotions on two emotional dimensions of arousal and valence. To improve the accuracy and robustness of machine emotion recognition by selecting appropriate dimension emotion label threshold; Using GSR signal difference between induced emotion state and natural emotion state conduct normalize processing, eliminate individual differences, and improve the generalization ability of the model; Applying multi-features fusion and random forest algorithm based on integrated learning to obtain better emotion recognition performance.The classification accuracy Acc and F1 values were 92.0% and 0.933 in the arousal dimension, and 90.9% and 0.925 in the valence dimension.The simulation results showed that the machine learning method based on tree model can realize the accurate recognition of dimension emotion, which can be used for automatic analysis of physiological signal emotion and machine recognition of wearable devices.