KIPS Transactions on Software and Data Engineering (정보처리학회논문지:소프트웨어 및 데이터공학)

Korea Information Processing Society (한국정보처리학회)
권호 표지
DOI QR코드

DOI QR Code

Epileptic Seizure Detection Using CNN Ensemble Models Based on Overlapping Segments of EEG Signals

뇌파의 중첩 분할에 기반한 CNN 앙상블 모델을 이용한 뇌전증 발작 검출

  • 김민기 (경상대학교 컴퓨터과학과, 공학연구원(ERI) 자동화.컴퓨터연구센터)
  • Received : 2021.04.28
  • Accepted : 2021.06.17
  • Published : 2021.12.31
Download PDF
⟨ Previous Next ⟩

Abstract

As the diagnosis using encephalography(EEG) has been expanded, various studies have been actively performed for classifying EEG automatically. This paper proposes a CNN model that can effectively classify EEG signals acquired from healthy persons and patients with epilepsy. We segment the EEG signals into sub-signals with smaller dimension to augment the EEG data that is necessary to train the CNN model. Then the sub-signals are segmented again with overlap and they are used for training the CNN model. We also propose ensemble strategy in order to improve the classification accuracy. Experimental result using public Bonn dataset shows that the CNN can detect the epileptic seizure with the accuracy above 99.0%. It also shows that the ensemble method improves the accuracy of 3-class and 5-class EEG classification.

뇌파(electroencephalogram, EEG)를 이용한 진단이 확대되면서 EEG 신호를 자동으로 분류하기 위한 다양한 연구가 활발히 이루어지고 있다. 본 논문은 일반인과 뇌전증 환자에게서 추출한 EEG 신호를 효과적으로 식별할 수 있는 CNN 모델을 제안한다. CNN의 학습에 필요한 데이터를 확장하기 위하여 EEG 신호를 낮은 차원의 신호로 분할하고, 이것을 다시 여러 개의 세그먼트로 중첩 분할하여 CNN 학습에 이용한다. 이와 더불어 CNN의 성능을 개선하기 위하여 CNN 앙상블 전략을 제안한다. 공개된 Bonn 데이터세트로 실험을 수행한 결과 뇌전증 발작을 99.0% 이상의 정확도로 검출하였고, 앙상블 방식에 의해 3-클래스와 5-클래스의 EEG 분류에서 정확도가 향상되었다.

Keywords

References

  1. K. Jung, "Epidemiology of epilepsy in Korea," Epilia: Epilepsy and Community, Vol.2, No.1, pp.17-20, 2020. https://doi.org/10.35615/epilia.2020.00150
  2. R. G. Andrzejak, K. Lehnertz, F. Mormann, C. Rieke, P. David, and C. E. Elger, "Indications of nonlinear deterministic and finite-dimensional structures in time series of brain electrical activity: Dependence on recording region and brain state," Physical Review, Vol.64, No.6, pp.1-8, 2001.
  3. M. R. Mohammadi, A. Khaleghi, A. M. Nasrabadi, S. Rafieivand, M. Begol, and H. Zarafshan, "EEG classification of ADHD and normal children using non-linear featuers and neural network," Biomedical Engineering Letter, Vol.6, No.2, pp.66-73, 2016. https://doi.org/10.1007/s13534-016-0218-2
  4. S. Bavkar, B. Iyer, and S. Deosarkar, "Detection of alcoholism: An EEG hybrid features and ensemble subspace K-NN based approach," In International Conference on Distributed Computing and Internet Technology (pp.161-168), Springer, Cham, 2019.
  5. A. Demerdzieva, "EEG characteristics of generalized anxiety disorder in childhood," Acta Informatica Medica, Vol.19, No.1, pp.9-15, 2011.
  6. D. J. McFarland and J. R. Wolpaw, "EEG-based brain-computer interfaces," Current Opinion in Biomedical Engineering, Vol.4, pp.194-200, 2017. https://doi.org/10.1016/j.cobme.2017.11.004
  7. R. A. Ricardo, O. L. Arturo, and O. P. Ivan, "Analysis of EEG signal processing techniques based on spectrograms," Research in Computing Science, Vol.145, pp.151-162, 2017. https://doi.org/10.13053/rcs-145-1-12
  8. W. Mao, H. I. K. Fathurrahman, Y. Lee, and T. W. Chang, "EEG dataset classification using CNN method," Journal of Physics: Conference Series, Vol.1456, pp.1-7, 2020.
  9. N. Kumar, K, Alam, and A. H. Siddiqi, "Wavelet transform for classification of EEG signal using SVM and ANN," Biomedical & Phamacology Journal, Vol.10, No.4, pp.2061-2069, 2017.
  10. K. C. Hsu and S. N. Yu, "Detection of seizures in EEG using subband nonlinear parameters and genetic algorithm," Computers in Biology and Medicine, Vol.40, No.10, pp.823-830, 2010. https://doi.org/10.1016/j.compbiomed.2010.08.005
  11. M. Savadkoohi, T. Oladunni, and L. Thompson, "A machine learning approach to epileptic seizure prediction using Electroencephalogram (EEG) signal," Biocybernetics and Biomedical Engineering, Vol.40, No.3, pp.1328-1341, 2020. https://doi.org/10.1016/j.bbe.2020.07.004
  12. A. Bhattacharyya and B. Pachori, "A multivariate approach for patient specific EEG seizure detection using empirical wavelet transform," IEEE Transactions on Biomeical Engineering, Vol.64, No.9, pp.2003-2015, 2017. https://doi.org/10.1109/TBME.2017.2650259
  13. R. T. Schirrmeister, et al., "Deep learning with convolutional neural networks for EEG decoding and visualization," in Human Brain Mapping, Vol.38, No.11, pp.5391-5420, 2017. https://doi.org/10.1002/hbm.23730
  14. K. K. Ang, Z. Y. Chin, H. Zhang, and C. Guan. "Filter bank common spatial pattern (FBCSP) in brain-computer interface," in Proceedings of IEEE International Joint Conference on Neural Nwtworks, pp.2390-2397, 2008.
  15. P. Sandheep, S. Vineeth, P. Meljo, and D. P. Subha, "Performance analysis of deep learning CNN in classification of depression EEG signals," in Proceedings of IEEE Region 10 Conference, pp.1339-1344, 2019.
  16. U. R. Acharya, S. L. Oh, Y. Hagiwara, J. H. Tan, and H. Adeli, "Deep convolutional neural network for the automated detection and diagnosis of seizure using EEG signals," Computers in Biology and Medicine, Vol.100, pp.270-278, 2018. https://doi.org/10.1016/j.compbiomed.2017.09.017
  17. I. Ullah, M. Hussain, E. Qazi, and H. Aboalsamh, "An automated system for epilepsy detection using EEG brain signals based on deep learning approach," Expert Systems With Applications, Vol.107, pp.61-71, 2018. https://doi.org/10.1016/j.eswa.2018.04.021
  18. G. Xu, T. Ren, Y. Chen, and W. Che, "A one-dimensional CNN-LSTM model for epileptic seizure recognition using EEG signal analysis," Frontiers in Neuroscience, Vol.14, pp.1-9, 2020. https://doi.org/10.3389/fnins.2020.00001
  19. Epileptic Seizure Recognition Data Set [Internet], https://archive.ics.uci.edu/ml/datasets/Epileptic+Seizure+Recognition.