基于注意力机制的多分支LSTM心理负荷评估模型

作者:冯源; 李彦蕾; 李一凡; 张博; 丁锦红; 夏立坤*
来源:计算机应用研究, 2021, 38(11): 3371-3375.
DOI:10.19734/j.issn.1001-3695.2021.04.0142

摘要

充分获取脑电信号的有效特征已成为心理负荷评估亟待解决的问题。提出一种多分支LSTM和注意力机制相结合的多分类网络框架。首先,此网络在对脑电信号做切片处理后,采用多分支LSTM网络提取切片中的时间特征;然后,利用注意力机制对所提取的时间特征进行权重参数优化;最后,通过softmax层输出心理负荷评估结果。通过消融实验和对比实验对模型进行验证。结果表明,此网络无论在二分类任务还是多分类任务中的表现均优于现有先进网络。

全文