Mar, 2024

ARNN:用于识别癫痫发作的多通道脑电信号的注意力循环神经网络

TL;DR我们提出了一种注意力循环神经网络 (ARNN),它在序列上循环应用注意力层,并具有与序列长度线性复杂度的特点。该模型在多通道脑电图信号上操作,利用并行计算。我们的架构受注意力层和长短期记忆 (LSTM) 单元的启发,它使用长短样式门进行扩展,并通过数倍的扩展来并行化处理多通道脑电图信号。经过广泛的实验评估,包括 CHB-MIT、UPenn 和 Mayos Clinic CHB-MIT 数据集,实证结果表明 ARNN 模型在各种任务中表现优于 LSTM、Vision Transformer (ViT)、Compact Convolution Transformer (CCT) 和 R-Transformer (RT) 等基准方法,展示出卓越的性能和更快的处理能力。代码已在 https://github.com/Salim-Lysiun/ARNN 上公开获取。