脑电图
情绪识别
模式识别(心理学)
计算机科学
语音识别
人工智能
变压器
心理学
工程类
电气工程
电压
精神科
作者
Yufan Yi,Yiping Xu,Bo Yang,Yan Tian
标识
DOI:10.1109/taffc.2024.3395359
摘要
Emotion recognition based on EEG signals has been a challenging task. The acquisition of EEG signals is complex, time-consuming, and has a high overhead. Artificial Intelligence Generated Content technology has been developing rapidly in image and sound generation, but effective generative models for EEG signal generation have rarely appeared. To address the problem, we propose a weighted co-training framework for emotion recognition using a frequency-spatial diffusion transformer. We propose the EEG signal generation model by utilizing frequency-spatial correlation. In the first step, we use forward diffusion to add noise to the real samples and then use the proposed model to denoise and restore the real EEG signals. It ensures that the trained generative model can generate real EEG signals. Then, we use the denoising process of the proposed model to generate a large amount of data and then use the pseudo-data weighting module to evaluate the generated samples further. Finally, the actual samples and weighted pseudo-data are used to train the classifier for better generalization jointly. We conducted comprehensive experiments on three EEG emotion recognition benchmark datasets. The experimental results show that our method improves by 0.96%-1.91% compared to state-of-the-art methods. In addition, quantitative and qualitative analysis proves the effectiveness of our proposed method.
科研通智能强力驱动
Strongly Powered by AbleSci AI