EEG signal classification using LSTM and improved neural network algorithms

被引:99
作者
Nagabushanam, P. [1 ]
George, S. Thomas [2 ]
Radha, S. [3 ]
机构
[1] Karunya Inst Technol & Sci, Dept EEE, Coimbatore, Tamil Nadu, India
[2] Karunya Inst Technol & Sci, Dept EIE, Coimbatore, Tamil Nadu, India
[3] Karunya Inst Technol & Sci, Dept ECE, Coimbatore, Tamil Nadu, India
关键词
LSTM; Neural network (NN); Improved NN; Logistic regression; EEG; Accuracy; ALPHA; ARCHITECTURE; TUMORS;
D O I
10.1007/s00500-019-04515-0
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Neural network (NN) finds role in variety of applications due to combined effect of feature extraction and classification availability in deep learning algorithms. In this paper, we have chosen SVM, logistic regression machine learning algorithms and NN for EEG signal classification. Two-layer LSTM and four-layer improved NN deep learning algorithms are proposed to improve the performance in EEG classification. Novelty lies in one-dimensional gradient descent activation functions with radial basis operations used in the initial layers of improved NN which help in achieving better performance. Statistical features namely mean, standard deviation, kurtosis and skewness are extracted for input EEG collected from Bonn database and then applied for various classification techniques. Accuracy, precision, recall and F1 score are the performance metrics used for analyzing the algorithms. Improved NN and LSTM give better performance compared to all other architectures. The simulations are carried out with variety of activation functions, optimizers and loss models to analyze the performance using Python in keras.
引用
收藏
页码:9981 / 10003
页数:23
相关论文
共 67 条
[1]   Sleep Stage Classification Using EEG Signal Analysis: A Comprehensive Survey and New Investigation [J].
Aboalayon, Khald Ali I. ;
Faezipour, Miad ;
Almuhammadi, Wafaa S. ;
Moslehpour, Saeid .
ENTROPY, 2016, 18 (09)
[2]   Deep convolutional neural network for the automated detection and diagnosis of seizure using EEG signals [J].
Acharya, U. Rajendra ;
Oh, Shu Lih ;
Hagiwara, Yuki ;
Tan, Jen Hong ;
Adeli, Hojjat .
COMPUTERS IN BIOLOGY AND MEDICINE, 2018, 100 :270-278
[3]   Automated EEG-based screening of depression using deep convolutional neural network [J].
Acharya, U. Rajendra ;
Oh, Shu Lih ;
Hagiwara, Yuki ;
Tan, Jen Hong ;
Adeli, Hojjat ;
Subha, D. P. .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2018, 161 :103-113
[4]   Accurate Classification of EEG Signals Using Neural Networks Trained by Hybrid Population-physic-based Algorithm [J].
Afrakhteh, Sajjad ;
Mosavi, Mohammad-Reza ;
Khishe, Mohammad ;
Ayatollahi, Ahmad .
INTERNATIONAL JOURNAL OF AUTOMATION AND COMPUTING, 2020, 17 (01) :108-122
[5]  
Antoniades A, 2017, IEEE T NEUR SYS REH, V25, P1534
[6]   Fully automatic model-based segmentation and classification approach for MRI brain tumor using artificial neural networks [J].
Arunkumar, N. ;
Mohammed, Mazin Abed ;
Mostafa, Salama A. ;
Ibrahim, Dheyaa Ahmed ;
Rodrigues, Joel J. P. C. ;
de Albuquerque, Victor Hugo C. .
CONCURRENCY AND COMPUTATION-PRACTICE & EXPERIENCE, 2020, 32 (01)
[7]   A forecasting tool for prediction of epileptic seizures using a machine learning approach [J].
Asharindavida, Fayas ;
Hossain, M. Shamim ;
Thacham, Azeemsha ;
Khammari, Hedi ;
Ahmed, Irfan ;
Alraddady, Fahad ;
Masud, Mehedi .
CONCURRENCY AND COMPUTATION-PRACTICE & EXPERIENCE, 2020, 32 (01)
[8]  
Asim Y, 2017, INT J IMAG SYST TECH, V2018, P1
[9]  
Bajaj V, 2018, IEEE SENSORS LETT, P2475
[10]  
Bertrand A, 2015, IEEE T NEUR SYS REH, P1534