1
|
Pfeffer MA, Ling SSH, Wong JKW. Exploring the frontier: Transformer-based models in EEG signal analysis for brain-computer interfaces. Comput Biol Med 2024; 178:108705. [PMID: 38865781 DOI: 10.1016/j.compbiomed.2024.108705] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [MESH Headings] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 11/11/2023] [Revised: 05/01/2024] [Accepted: 06/01/2024] [Indexed: 06/14/2024]
Abstract
This review systematically explores the application of transformer-based models in EEG signal processing and brain-computer interface (BCI) development, with a distinct focus on ensuring methodological rigour and adhering to empirical validations within the existing literature. By examining various transformer architectures, such as the Temporal Spatial Transformer Network (TSTN) and EEG Conformer, this review delineates their capabilities in mitigating challenges intrinsic to EEG data, such as noise and artifacts, and their subsequent implications on decoding and classification accuracies across disparate mental tasks. The analytical scope extends to a meticulous examination of attention mechanisms within transformer models, delineating their role in illuminating critical temporal and spatial EEG features and facilitating interpretability in model decision-making processes. The discourse additionally encapsulates emerging works that substantiate the efficacy of transformer models in noise reduction of EEG signals and diversifying applications beyond the conventional motor imagery paradigm. Furthermore, this review elucidates evident gaps and propounds exploratory avenues in the applications of pre-trained transformers in EEG analysis and the potential expansion into real-time and multi-task BCI applications. Collectively, this review distils extant knowledge, navigates through the empirical findings, and puts forward a structured synthesis, thereby serving as a conduit for informed future research endeavours in transformer-enhanced, EEG-based BCI systems.
Collapse
Affiliation(s)
- Maximilian Achim Pfeffer
- Faculty of Engineering and Information Technology, University of Technology Sydney, CB11 81-113, Broadway, Ultimo, 2007, New South Wales, Australia.
| | - Steve Sai Ho Ling
- Faculty of Engineering and Information Technology, University of Technology Sydney, CB11 81-113, Broadway, Ultimo, 2007, New South Wales, Australia.
| | - Johnny Kwok Wai Wong
- Faculty of Design, Architecture and Building, University of Technology Sydney, 15 Broadway, Ultimo, 2007, New South Wales, Australia.
| |
Collapse
|
2
|
Atal DK, Singh M. Effectual seizure detection using MBBF-GPSO with CNN network. Cogn Neurodyn 2024; 18:907-918. [PMID: 38826653 PMCID: PMC11143161 DOI: 10.1007/s11571-023-09943-1] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 06/02/2022] [Revised: 12/23/2022] [Accepted: 02/05/2023] [Indexed: 03/02/2023] Open
Abstract
EEG is the most common test for diagnosing a seizure, where it presents information about the electrical activity of the brain. Automatic Seizure detection is one of the challenging tasks due to limitations of conventional methods with regard to inefficient feature selection, increased computational complexity and time and less accuracy. The situation calls for a practical framework to achieve better performance for detecting the seizure effectively. Hence, this study proposes modified Blackman bandpass filter-greedy particle swarm optimization (MBBF-GPSO) with convolutional neural network (CNN) for effective seizure detection. In this case, unwanted signals (noise) is eliminated by MBBF as it possess better ability in stopband attenuation, and, only the optimized features are selected using GPSO. For enhancing the efficacy of obtaining optimal solutions in GPSO, the time and frequency domain is extracted to complement it. Through this process, an optimized features are attained by MBBF-GPSO. Then, the CNN layer is employed for obtaining the productive classification output using the objective function. Here, CNN is employed due to its ability in automatically learning distinct features for individual class. Such advantages of the proposed system have made it explore better performance in seizure detection that is confirmed through performance and comparative analysis.
Collapse
Affiliation(s)
- Dinesh Kumar Atal
- Department of Electrical Engineering, Delhi Technological University, Bawana Road, Delhi, 110042 India
| | - Mukhtiar Singh
- Department of Electrical Engineering, Delhi Technological University, Bawana Road, Delhi, 110042 India
| |
Collapse
|
3
|
Kantipudi MVVP, Kumar NSP, Aluvalu R, Selvarajan S, Kotecha K. An improved GBSO-TAENN-based EEG signal classification model for epileptic seizure detection. Sci Rep 2024; 14:843. [PMID: 38191643 PMCID: PMC10774431 DOI: 10.1038/s41598-024-51337-8] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [MESH Headings] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 09/08/2023] [Accepted: 01/03/2024] [Indexed: 01/10/2024] Open
Abstract
Detection and classification of epileptic seizures from the EEG signals have gained significant attention in recent decades. Among other signals, EEG signals are extensively used by medical experts for diagnosing purposes. So, most of the existing research works developed automated mechanisms for designing an EEG-based epileptic seizure detection system. Machine learning techniques are highly used for reduced time consumption, high accuracy, and optimal performance. Still, it limits by the issues of high complexity in algorithm design, increased error value, and reduced detection efficacy. Thus, the proposed work intends to develop an automated epileptic seizure detection system with an improved performance rate. Here, the Finite Linear Haar wavelet-based Filtering (FLHF) technique is used to filter the input signals and the relevant set of features are extracted from the normalized output with the help of Fractal Dimension (FD) analysis. Then, the Grasshopper Bio-Inspired Swarm Optimization (GBSO) technique is employed to select the optimal features by computing the best fitness value and the Temporal Activation Expansive Neural Network (TAENN) mechanism is used for classifying the EEG signals to determine whether normal or seizure affected. Numerous intelligence algorithms, such as preprocessing, optimization, and classification, are used in the literature to identify epileptic seizures based on EEG signals. The primary issues facing the majority of optimization approaches are reduced convergence rates and higher computational complexity. Furthermore, the problems with machine learning approaches include a significant method complexity, intricate mathematical calculations, and a decreased training speed. Therefore, the goal of the proposed work is to put into practice efficient algorithms for the recognition and categorization of epileptic seizures based on EEG signals. The combined effect of the proposed FLHF, FD, GBSO, and TAENN models might dramatically improve disease detection accuracy while decreasing complexity of system along with time consumption as compared to the prior techniques. By using the proposed methodology, the overall average epileptic seizure detection performance is increased to 99.6% with f-measure of 99% and G-mean of 98.9% values.
Collapse
Affiliation(s)
- M V V Prasad Kantipudi
- Symbiosis Institute of Technology, Symbiosis International (Deemed) University, Pune, 412115, India
| | - N S Pradeep Kumar
- S.E.A College of Engineering and Technology, Bengaluru, 560049, India
| | - Rajanikanth Aluvalu
- Department of Information Technology, Chaitanya Bharathi Institute of Technology, Hyderabad, 500075, India
| | - Shitharth Selvarajan
- School of Built Environment, Engineering and Computing, Leeds Beckett University, Leeds, LS1 3HE, UK.
- Department of Computer Science, Kebri Dehar University, Somali, Ethiopia.
| | - K Kotecha
- Symbiosis Institute of Technology, Symbiosis International (Deemed) University, Pune, 412115, India
- Symbiosis Centre for Applied Artificial Intelligence (SCAAI), Symbiosis International (Deemed) University, Pune, 412115, India
| |
Collapse
|
4
|
McDougall M, Albaqami H, Mubashar Hassan G, Datta A. Patient Independent Interictal Epileptiform Discharge Detection. ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE AND BIOLOGY SOCIETY. IEEE ENGINEERING IN MEDICINE AND BIOLOGY SOCIETY. ANNUAL INTERNATIONAL CONFERENCE 2023; 2023:1-6. [PMID: 38082573 DOI: 10.1109/embc40787.2023.10341194] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [MESH Headings] [Track Full Text] [Subscribe] [Scholar Register] [Indexed: 12/18/2023]
Abstract
Epilepsy is a highly prevalent brain condition with many serious complications arising from it. The majority of patients which present to a clinic and undergo electroencephalogram (EEG) monitoring would be unlikely to experience seizures during the examination period, thus the presence of interictal epileptiform discharges (IEDs) become effective markers for the diagnosis of epilepsy. Furthermore, IED shapes and patterns are highly variable across individuals, yet trained experts are still able to identify them through EEG recordings - meaning that commonalities exist across IEDs that an algorithm can be trained on to detect and generalise to the larger population. This research proposes an IED detection system for the binary classification of epilepsy using scalp EEG recordings. The proposed system features an ensemble based deep learning method to boost the performance of a residual convolutional neural network, and a bidirectional long short-term memory network. This is implemented using raw EEG data, sourced from Temple University Hospital's EEG Epilepsy Corpus, and is found to outperform the current state of the art model for IED detection across the same dataset. The achieved accuracy and Area Under Curve (AUC) of 94.92% and 97.45% demonstrates the effectiveness of an ensemble method, and that IED detection can be achieved with high performance using raw scalp EEG data, thus showing promise for the proposed approach in clinical settings.
Collapse
|
5
|
Rana M, Bhushan M. Machine learning and deep learning approach for medical image analysis: diagnosis to detection. MULTIMEDIA TOOLS AND APPLICATIONS 2022; 82:1-39. [PMID: 36588765 PMCID: PMC9788870 DOI: 10.1007/s11042-022-14305-w] [Citation(s) in RCA: 4] [Impact Index Per Article: 2.0] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Figures] [Subscribe] [Scholar Register] [Received: 08/01/2022] [Revised: 11/01/2022] [Accepted: 12/10/2022] [Indexed: 06/17/2023]
Abstract
Computer-aided detection using Deep Learning (DL) and Machine Learning (ML) shows tremendous growth in the medical field. Medical images are considered as the actual origin of appropriate information required for diagnosis of disease. Detection of disease at the initial stage, using various modalities, is one of the most important factors to decrease mortality rate occurring due to cancer and tumors. Modalities help radiologists and doctors to study the internal structure of the detected disease for retrieving the required features. ML has limitations with the present modalities due to large amounts of data, whereas DL works efficiently with any amount of data. Hence, DL is considered as the enhanced technique of ML where ML uses the learning techniques and DL acquires details on how machines should react around people. DL uses a multilayered neural network to get more information about the used datasets. This study aims to present a systematic literature review related to applications of ML and DL for the detection along with classification of multiple diseases. A detailed analysis of 40 primary studies acquired from the well-known journals and conferences between Jan 2014-2022 was done. It provides an overview of different approaches based on ML and DL for the detection along with the classification of multiple diseases, modalities for medical imaging, tools and techniques used for the evaluation, description of datasets. Further, experiments are performed using MRI dataset to provide a comparative analysis of ML classifiers and DL models. This study will assist the healthcare community by enabling medical practitioners and researchers to choose an appropriate diagnosis technique for a given disease with reduced time and high accuracy.
Collapse
Affiliation(s)
- Meghavi Rana
- School of Computing, DIT University, Dehradun, India
| | - Megha Bhushan
- School of Computing, DIT University, Dehradun, India
| |
Collapse
|
6
|
Huang X, Sun X, Zhang L, Zhu T, Yang H, Xiong Q, Feng L. A Novel Epilepsy Detection Method Based on Feature Extraction by Deep Autoencoder on EEG Signal. INTERNATIONAL JOURNAL OF ENVIRONMENTAL RESEARCH AND PUBLIC HEALTH 2022; 19:15110. [PMID: 36429845 PMCID: PMC9690147 DOI: 10.3390/ijerph192215110] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [MESH Headings] [Track Full Text] [Figures] [Subscribe] [Scholar Register] [Received: 10/31/2022] [Revised: 11/12/2022] [Accepted: 11/13/2022] [Indexed: 06/16/2023]
Abstract
Electroencephalogram (EEG) signals are the gold standard tool for detecting epileptic seizures. Long-term EEG signal monitoring is a promising method to realize real-time and automatic epilepsy detection with the assistance of computer-aided techniques and the Internet of Medical Things (IoMT) devices. Machine learning (ML) algorithms combined with advanced feature extraction methods have been widely explored to precisely recognize EEG signals, while among which, little attention has been paid to high computing costs and severe information losses. The lack of model interpretability also impedes the wider application and deeper understanding of ML methods in epilepsy detection. In this research, a novel feature extraction method based on an autoencoder (AE) is proposed in the time domain. The architecture and mechanism are elaborated. In this method, specified features are defined and calculated on the basis of signal reconstruction quantification of the AE. The EEG recognition is performed to validate the effectiveness of the proposed detection method, and the prediction accuracy reached 97%. To further investigate the superiority of the proposed AE-based feature extraction method, a widely used feature extraction method, PCA, is allocated for comparison. In order to understand the underlying working mechanism, permutation importance and SHapley Additive exPlanations (SHAP) are conducted for model interpretability, and the results further confirm the reasonability and effectiveness of the extracted features by AE reconstruction. With high computing efficiency in the time domain and an extensively satisfactory accuracy, the proposed epilepsy detection method exhibits great superiority and potential in almost real-time and automatic epilepsy monitoring.
Collapse
Affiliation(s)
- Xiaojie Huang
- School of Basic Medical Sciences, Anhui Medical University, Hefei 230032, China
- Institute of Biopharmaceuticals, Anhui Medical University, Hefei 230032, China
| | - Xiangtao Sun
- Department of Disaster Mitigation for Structures, Tongji University, Shanghai 200092, China
| | - Lijun Zhang
- School of Basic Medical Sciences, Anhui Medical University, Hefei 230032, China
- Institute of Biopharmaceuticals, Anhui Medical University, Hefei 230032, China
| | - Tong Zhu
- School of Basic Medical Sciences, Anhui Medical University, Hefei 230032, China
- Institute of Biopharmaceuticals, Anhui Medical University, Hefei 230032, China
| | - Hao Yang
- School of Basic Medical Sciences, Anhui Medical University, Hefei 230032, China
- Institute of Biopharmaceuticals, Anhui Medical University, Hefei 230032, China
| | - Qingsong Xiong
- Department of Disaster Mitigation for Structures, Tongji University, Shanghai 200092, China
| | - Lijie Feng
- School of Basic Medical Sciences, Anhui Medical University, Hefei 230032, China
- Institute of Biopharmaceuticals, Anhui Medical University, Hefei 230032, China
| |
Collapse
|
7
|
Shoeibi A, Moridian P, Khodatars M, Ghassemi N, Jafari M, Alizadehsani R, Kong Y, Gorriz JM, Ramírez J, Khosravi A, Nahavandi S, Acharya UR. An overview of deep learning techniques for epileptic seizures detection and prediction based on neuroimaging modalities: Methods, challenges, and future works. Comput Biol Med 2022; 149:106053. [DOI: 10.1016/j.compbiomed.2022.106053] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 06/12/2022] [Revised: 08/17/2022] [Accepted: 08/17/2022] [Indexed: 02/01/2023]
|
8
|
Application of Deep Learning and WT-SST in Localization of Epileptogenic Zone Using Epileptic EEG Signals. APPLIED SCIENCES-BASEL 2022. [DOI: 10.3390/app12104879] [Citation(s) in RCA: 1] [Impact Index Per Article: 0.5] [Reference Citation Analysis] [Abstract] [Track Full Text] [Subscribe] [Scholar Register] [Indexed: 02/04/2023]
Abstract
Focal and non-focal Electroencephalogram (EEG) signals have proved to be effective techniques for identifying areas in the brain that are affected by epileptic seizures, known as the epileptogenic zones. The detection of the location of focal EEG signals and the time of seizure occurrence are vital information that help doctors treat focal epileptic seizures using a surgical method. This paper proposed a computer-aided detection (CAD) system for detecting and classifying focal and non-focal EEG signals as the manual process is time-consuming, prone to error, and tedious. The proposed technique employs time-frequency features, statistical, and nonlinear approaches to form a robust features extraction technique. Four detection and classification techniques for focal and non-focal EEG signals were proposed. (1). Combined hybrid features with Support Vector Machine (Hybrid-SVM) (2). Discrete Wavelet Transform with Deep Learning Network (DWT-DNN) (3). Combined hybrid features with DNN (Hybrid-DNN) as an optimized DNN model. Lastly, (4). A newly proposed technique using Wavelet Synchrosqueezing Transform-Deep Convolutional Neural Network (WTSST-DCNN). Prior to feeding the features to classifiers, statistical analyses, including t-tests, were deployed to obtain relevant and significant features at each approach. The proposed feature extraction technique and classification proved effective and suitable for smart Internet of Medical Things (IoMT) devices as performance parameters of accuracy, sensitivity, and specificity are higher than recently related works with a value of 99.7%, 99.5%, and 99.7% respectively.
Collapse
|
9
|
Cherian R, Kanaga EG. Theoretical and Methodological Analysis of EEG based Seizure Detection and Prediction: An Exhaustive Review. J Neurosci Methods 2022; 369:109483. [DOI: 10.1016/j.jneumeth.2022.109483] [Citation(s) in RCA: 1] [Impact Index Per Article: 0.5] [Reference Citation Analysis] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 09/16/2021] [Revised: 01/13/2022] [Accepted: 01/13/2022] [Indexed: 02/07/2023]
|
10
|
Prabhakar SK, Lee SW. SASDL and RBATQ: Sparse Autoencoder with Swarm based Deep Learning and Reinforcement based Q-learning for EEG Classification. IEEE OPEN JOURNAL OF ENGINEERING IN MEDICINE AND BIOLOGY 2022; 3:58-68. [PMID: 35770240 PMCID: PMC9135179 DOI: 10.1109/ojemb.2022.3161837] [Citation(s) in RCA: 5] [Impact Index Per Article: 2.5] [Reference Citation Analysis] [Abstract] [Grants] [Track Full Text] [Download PDF] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 11/02/2021] [Revised: 02/20/2022] [Accepted: 03/17/2022] [Indexed: 11/23/2022] Open
Abstract
The most vital information about the electrical activities of the brain can be obtained with the help of Electroencephalography (EEG) signals. It is quite a powerful tool to analyze the neural activities of the brain and various neurological disorders like epilepsy, schizophrenia, sleep related disorders, parkinson disease etc. can be investigated well with the help of EEG signals. Goal: In this paper, two versatile deep learning methods are proposed for the efficient classification of epilepsy and schizophrenia from EEG datasets. Methods: The main advantage of using deep learning when compared to other machine learning algorithms is that it has the capability to accomplish feature engineering on its own. Swarm intelligence is also a highly useful technique to solve a wide range of real-world, complex, and non-linear problems. Therefore, taking advantage of these factors, the first method proposed is a Sparse Autoencoder (SAE) with swarm based deep learning method and it is named as (SASDL) using Particle Swarm Optimization (PSO) technique, Cuckoo Search Optimization (CSO) technique and Bat Algorithm (BA) technique; and the second technique proposed is the Reinforcement Learning based on Bidirectional Long-Short Term Memory (BiLSTM), Attention Mechanism, Tree LSTM and Q learning, and it is named as (RBATQ) technique. Results and Conclusions: Both these two novel deep learning techniques are tested on epilepsy and schizophrenia EEG datasets and the results are analyzed comprehensively, and a good classification accuracy of more than 93% is obtained for all the datasets.
Collapse
Affiliation(s)
| | - Seong-Whan Lee
- Department of Artificial IntelligenceKorea University Seoul 02841 South Korea
| |
Collapse
|
11
|
Obukhov YV, Kershner IA, Tolmacheva RA, Sinkin MV, Zhavoronkova LA. Wavelet Ridges in EEG Diagnostic Features Extraction: Epilepsy Long-Time Monitoring and Rehabilitation after Traumatic Brain Injury. SENSORS (BASEL, SWITZERLAND) 2021; 21:5989. [PMID: 34577198 PMCID: PMC8468146 DOI: 10.3390/s21185989] [Citation(s) in RCA: 3] [Impact Index Per Article: 1.0] [Reference Citation Analysis] [Abstract] [Key Words] [MESH Headings] [Track Full Text] [Download PDF] [Figures] [Subscribe] [Scholar Register] [Received: 08/13/2021] [Revised: 08/29/2021] [Accepted: 09/03/2021] [Indexed: 01/06/2023]
Abstract
Interchannel EEG synchronization, as well as its violation, is an important diagnostic sign of a number of diseases. In particular, during an epileptic seizure, such synchronization occurs starting from some pairs of channels up to many pairs in a generalized seizure. Additionally, for example, after traumatic brain injury, the destruction of interneuronal connections occurs, which leads to a violation of interchannel synchronization when performing motor or cognitive tests. Within the framework of a unified approach to the analysis of interchannel EEG synchronization using the ridges of wavelet spectra, two problems were solved. First, the segmentation of the initial data of long-term monitoring of scalp EEG with various artifacts into fragments suspicious of epileptic seizures in order to reduce the total duration of the fragments analyzed by the doctor. Second, assessments of recovery after rehabilitation of cognitive functions in patients with moderate traumatic brain injury. In the first task, the initial EEG was segmented into fragments in which at least two channels were synchronized, and by the adaptive threshold method into fragments with a high value of the EEG power spectral density. Overlapping in time synchronized fragments with fragments of high spectral power density was determined. As a result, the total duration of the fragments for analysis by the doctor was reduced by more than 60 times. In the second task, the network of phase-related EEG channels was determined during the cognitive test before and after rehabilitation. Calculation-logical and spatial-pattern cognitive tests were used. The positive dynamics of rehabilitation was determined during the initialization of interhemispheric connections and connections in the frontal cortex of the brain.
Collapse
Affiliation(s)
- Yury Vladimirovich Obukhov
- Kotelnikov Institute of Radio Engineering and Electronics of RAS, Mokhovaya St. 11-7, 125009 Moscow, Russia; (Y.V.O.); (R.A.T.)
| | - Ivan Andreevich Kershner
- Kotelnikov Institute of Radio Engineering and Electronics of RAS, Mokhovaya St. 11-7, 125009 Moscow, Russia; (Y.V.O.); (R.A.T.)
| | - Renata Alekseevna Tolmacheva
- Kotelnikov Institute of Radio Engineering and Electronics of RAS, Mokhovaya St. 11-7, 125009 Moscow, Russia; (Y.V.O.); (R.A.T.)
| | - Mikhail Vladimirovich Sinkin
- Department of Neurosurgery of the Sklifosovsky Research Institute for Emergency Medicine of Moscow Healthcare Department, Bolshaya Sukharevskaya Square 3, 129090 Moscow, Russia; or
- Laboratory of Invasive Neurointerfaces of the Research Institute TechnoBioMed, A.I. Yevdokimov Moscow State University of Medicine and Dentistry, Delegatskaya St. 20 p.1, 127473 Moscow, Russia
| | - Ludmila Alekseevna Zhavoronkova
- Laboratory of General and Clinical Neurophysiology of the Institute of Higher Nervous Activity and Neurophysiology of RAS, Butlerova St. 5a, 117485 Moscow, Russia; or
| |
Collapse
|
12
|
Mukhtar H, Qaisar SM, Zaguia A. Deep Convolutional Neural Network Regularization for Alcoholism Detection Using EEG Signals. SENSORS (BASEL, SWITZERLAND) 2021; 21:5456. [PMID: 34450899 PMCID: PMC8402228 DOI: 10.3390/s21165456] [Citation(s) in RCA: 7] [Impact Index Per Article: 2.3] [Reference Citation Analysis] [Abstract] [Key Words] [MESH Headings] [Grants] [Track Full Text] [Download PDF] [Figures] [Subscribe] [Scholar Register] [Received: 07/19/2021] [Revised: 08/05/2021] [Accepted: 08/09/2021] [Indexed: 12/31/2022]
Abstract
Alcoholism is attributed to regular or excessive drinking of alcohol and leads to the disturbance of the neuronal system in the human brain. This results in certain malfunctioning of neurons that can be detected by an electroencephalogram (EEG) using several electrodes on a human skull at appropriate positions. It is of great interest to be able to classify an EEG activity as that of a normal person or an alcoholic person using data from the minimum possible electrodes (or channels). Due to the complex nature of EEG signals, accurate classification of alcoholism using only a small dataset is a challenging task. Artificial neural networks, specifically convolutional neural networks (CNNs), provide efficient and accurate results in various pattern-based classification problems. In this work, we apply CNN on raw EEG data and demonstrate how we achieved 98% average accuracy by optimizing a baseline CNN model and outperforming its results in a range of performance evaluation metrics on the University of California at Irvine Machine Learning (UCI-ML) EEG dataset. This article explains the stepwise improvement of the baseline model using the dropout, batch normalization, and kernel regularization techniques and provides a comparison of the two models that can be beneficial for aspiring practitioners who aim to develop similar classification models in CNN. A performance comparison is also provided with other approaches using the same dataset.
Collapse
Affiliation(s)
- Hamid Mukhtar
- Department of Computer Science, College of Computers and Information Technology, Taif University, P.O. Box 11099, Taif 21944, Saudi Arabia;
| | - Saeed Mian Qaisar
- Electrical and Computer Engineering Department, College of Engineering, Effat University, Jeddah 22332, Saudi Arabia;
- Communication and Signal Processing Lab, Energy and Technology Research Centre, Effat University, Jeddah 22332, Saudi Arabia
| | - Atef Zaguia
- Department of Computer Science, College of Computers and Information Technology, Taif University, P.O. Box 11099, Taif 21944, Saudi Arabia;
| |
Collapse
|