arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.08081eess.SP

CLDTA:用于跨数据集 EEG 情绪识别的基于对角 Transformer 自编码器的对比学习

CLDTA: Contrastive Learning based on Diagonal Transformer Autoencoder for Cross-Dataset EEG Emotion Recognition

Yuan Liao, Yuhong Zhang, Shenghuan Wang, Xiruo Zhang, Yiling Zhang, Wei Chen, Yuzhe Gu, Liya Huang

AI总结:

针对 EEG 情绪识别中的跨数据集与跨受试者泛化难题,本文提出 CLDTA 框架,通过对角掩码、信息分离、对比学习和少样本校准提升迁移性能与可解释性。

摘要:

非侵入式 EEG 技术的最新进展拓宽了其在情绪识别中的应用,并产生了大量相关数据集。然而,由于采集设备和情绪刺激材料存在差异,深度学习模型难以在这些数据集之间泛化。为满足对一种通用模型的迫切需求,使其能够灵活适应不同 EEG 数据集格式并弥合实验室数据与真实世界数据之间的差距,我们提出了一种新颖的深度学习框架:基于对比学习的对角 Transformer 自编码器(Contrastive Learning based Diagonal Transformer Autoencoder,CLDTA),专为基于 EEG 的情绪识别定制。CLDTA 在其编码器内采用对角掩码策略,以提取全通道 EEG 数据的脑网络知识,促进向通道数较少数据集的迁移。信息分离机制通过实现脑网络的直接可视化提升了模型的可解释性。CLDTA 框架采用对比学习提炼与受试者无关的情绪表征,并使用校准预测过程,使模型能够以极少样本快速适应新受试者,实现准确的情绪识别。我们在 SEED、SEED-IV、SEED-V 和 DEAP 数据集上的分析突显了 CLDTA 的稳定性能,以及其检测 EEG 信号中与情绪相关的任务特定特征和通用特征的能力,强调了其革新情绪识别研究的潜力。

英文摘要:

Recent advances in non-invasive EEG technology have broadened its application in emotion recognition, yielding a multitude of related datasets. Yet, deep learning models struggle to generalize across these datasets due to variations in acquisition equipment and emotional stimulus materials. To address the pressing need for a universal model that fluidly accommodates diverse EEG dataset formats and bridges the gap between laboratory and real-world data, we introduce a novel deep learning framework: the Contrastive Learning based Diagonal Transformer Autoencoder (CLDTA), tailored for EEG-based emotion recognition. The CLDTA employs a diagonal masking strategy within its encoder to extracts full-channel EEG data's brain network knowledge, facilitating transferability to the datasets with fewer channels. And an information separation mechanism improves model interpretability by enabling straightforward visualization of brain networks. The CLDTA framework employs contrastive learning to distill subject-independent emotional representations and uses a calibration prediction process to enable rapid adaptation of the model to new subjects with minimal samples, achieving accurate emotion recognition. Our analysis across the SEED, SEED-IV, SEED-V, and DEAP datasets highlights CLDTA's consistent performance and proficiency in detecting both task-specific and general features of EEG signals related to emotions, underscoring its potential to revolutionize emotion recognition research.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑