.png){kind=icon}
返回
AttenEpilepsy: A 2D convolutional network model based on multi-head self-attention
.png){kind=icon}
PRE
AI
DOI：10.1016/j.enganabound.2024.105989.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
The existing epilepsy detection models focus more on local information than the true meaning of long-range dependence when capturing time-frequency image features. This results in imprecise feature vector extraction and room for optimization of detection accuracy. AttenEpilepsy is a novel 2D convolutional network model that uses a multi-head self-attention mechanism to classify epileptic seizure periods, inter-seizure periods, and health states of single-channel EEG signals. The AttenEpilepsy model consists of two parts, namely feature extraction and time-frequency context encoding (STCE). A feature extraction method combining multi-path convolution and adaptive hybrid feature recalibration is proposed, in which multi-path convolution with convolution kernels of different sizes is used to extract relevant multi-scale features from time-frequency images. STCE consists of two modules: multi-head self-attention and causal convolution. A modified multi-head self-attention mechanism is used to model the extracted time-frequency features, and causal convolution is used to analyse the frequency information on the time dependencies. A public dataset from the University of Bonn Epilepsy Research Center is used to evaluate the performance of the AttenEpilepsy model. The experimental results show that the AttenEpilepsy model achieved accuracy (AC), sensitivity (SE), specificity (SP), and F1 score (F1) of 99.81%, 99.82%, 99.89%, and 99.83%, respectively. Further testing of the robustness of the model is conducted by introducing various types of noise into the input data. The proposed AttenEpilepsy network model outperforms the state-of-the-art in terms of various evaluation metrics.
Keyword:
Long-range dependence
Time-frequency image
Feature extraction
Time-frequency context encoding
Multi-head self-attention
Causal convolution
期刊
IF:
4.1
论文数:
5.9K
被引数:
9.4K
机构
引用论文
分享收藏An improved deep convolutional neural network with multi-scale information for bearing fault diagnosis一种改进的多尺度信息深度卷积神经网络在轴承故障诊断中的应用.png)
NEUROCOMPUTING
NEUROCOMPUTINGIF6.5
PREAI
分享收藏分享收藏Energy efficient FPGA implementation of an epileptic seizure detection system using a QDA classifier使用QDA分类器的癫痫发作检测系统的节能FPGA实现
PREAI
分享收藏An automated system for epilepsy detection using EEG brain signals based on deep learning approach基于深度学习方法的脑电脑信号癫痫检测自动化系统
PREAI
分享收藏Personalized attention-based EEG channel selection for epileptic seizure prediction基于个性化注意的脑电通道选择在癫痫发作预测中的应用
PREAI
分享收藏
查看更多内容