A novel computer-aided diagnosis framework for EEG-based identification of neural diseases

Abstract

Recent advances in electroencephalogram (EEG) signal classification have primarily focused on domain-specific approaches, which impede algorithm cross-discipline capability. This study introduces a new computer-aided diagnosis (CAD) system for the classification of two distinct EEG domains under a unified sequential framework. The key motivation to consider two neural diseases by one framework is to develop a unified algorithm for EEG classification. The main contributions of this study are five-fold. First, EEG signals are decomposed into 10 intrinsic mode functions (IMFs) with the help of empirical wavelet transform. Second, a novel two-dimensional (2D) modeling of IMFs is plotted to visualize the complexity of EEG signals. Third, several new geometrical features are extracted to analyze the dynamic and chaotic essence. Fourth, significant features are selected by binary particle swarm optimization algorithm (B–PSO). Fifth, selected features are fed to the k-nearest neighbor classifier for EEG signal classification purposes. All the experiments are executed on one depression and two epileptic EEG datasets in a leave one out cross-validation strategy. The proposed CAD system provides an average classification accuracy of 93.35% in depression detection, 99.33% for regular against ictal, and 97.33% for interictal versus ictal respectively. The overall empirical analysis authenticates that the proposed CAD outperforms the existing domain-specific methods in terms of classification accuracies and multirole adaptability, thus, can be endorsed as an effective automated neural rehabilitation system.