4.7 Article

A progressive deep wavelet cascade classification model for epilepsy detection

Journal

ARTIFICIAL INTELLIGENCE IN MEDICINE
Volume 118, Issue -, Pages -

Publisher

ELSEVIER
DOI: 10.1016/j.artmed.2021.102117

Keywords

EEG; Epilepsy detection; Discrete wavelet transform; Random forest; Cascade structure

Funding

  1. National Natural Science Foundation of China [61571302]
  2. Project of the Science and Technology Commission of Shanghai Municipality [18070503000]
  3. Key Project of Crossing Innovation of Medicine and Engineering, University of Shanghai for Science and Technology [1020308405]

Ask authors/readers for more resources

The PDWC, based on DWT and RF, mimics the progressive object identification process of human beings with recognition cycles to enhance wavelet energy features and improve recognition accuracy through cascade RF classifiers. It outperforms traditional schemes and deep learning methods with a mean accuracy of 0.9914 for diverse EEG signals.
Automatic epileptic seizure detection according to EEG recordings is helpful for neurologists to identify an epilepsy occurrence in the initial anti-epileptic treatment. To quickly and accurately detect epilepsy, we proposed a progressive deep wavelet cascade classification model (PDWC) based on the discrete wavelet transform (DWT) and Random Forest (RF). Different from current deep networks, the PDWC mimics the progressive object identification process of human beings with recognition cycles. In every cycle, enhanced wavelet energy features at a specific scale were extracted by DWT and input into a set of cascade RF classifiers to realize one recognition. The recognition accuracy of PDWC is gradually improved by the fusion of classification results produced by multiple recognition cycles. Moreover, the cascade structure of PDWC can be automatically determined by the classification accuracy increment between layers. To verify the performance of the PDWC, we respectively applied five traditional schemes and four deep learning schemes to four public datasets. The results show that the PDWC is not only superior than five traditional schemes, including KNN, Bayes, DT, SVM, and RF, but also better than deep learning methods, i.e. convolutional neural network (CNN), Long Short-Term Memory (LSTM), multi Grained Cascade Forest (gcForest) and wavelet cascade model (WCM). The mean accuracy of PDWC for all subjects of all datasets reaches to 0.9914. With a flexible structure and less parameters, the PDWC is more suitable for the epilepsy detection of diverse EEG signals.

Authors

I am an author on this paper
Click your name to claim this paper and add it to your profile.

Reviews

Primary Rating

4.7
Not enough ratings

Secondary Ratings

Novelty
-
Significance
-
Scientific rigor
-
Rate this paper

Recommended

No Data Available
No Data Available