Advertisement

Predicting Epileptic Seizures from Intracranial EEG Using LSTM-Based Multi-task Learning

  • Xuelin Ma
  • Shuang Qiu
  • Yuxing Zhang
  • Xiaoqin Lian
  • Huiguang HeEmail author
Conference paper
Part of the Lecture Notes in Computer Science book series (LNCS, volume 11257)

Abstract

Epilepsy afflicts nearly 1% of the world’s population, and is characterized by the occurrence of spontaneous seizures. It’s important to make prediction before seizures, so that epileptic can prevent seizures taking place on some specific occasions to avoid suffering from great damage. The previous work in seizure prediction paid less attention to the time-series information and their performances may also restricted to the small training data. In this study, we proposed a Long Short-Term Memory (LSTM)-based multi-task learning (MTL) framework for seizure prediction. The LSTM unit was used to process the sequential data and the MTL framework was applied to perform prediction and latency regression simultaneously. We evaluated the proposed method in the American Epilepsy Society Seizure Prediction Challenge dataset and obtained an average prediction accuracy of 89.36%, which was 3.41% higher than the reported state-of-the-art. In addition, the input data and output of middle layers were visualized. The visual and experiment results demonstrated the superior performance of our proposed LSTM-MTL method for seizure prediction.

Keywords

Seizure prediction LSTM Multi-task learning Intracranial EEG 

Notes

Acknowledgments

This work was supported by National Natural Science Foundation of China (91520202, 81701785), Youth Innovation Promotion Association CAS, the CAS Scientific Research Equipment Development Project (YJKYYQ20170050) and the Beijing Municipal Science&Technology Commission (Z181100008918010).

References

  1. 1.
    Bashivan, P., Rish, I., Yeasin, M., Codella, N.: Learning representations from EEG with deep recurrent-convolutional neural networks. arXiv preprint arXiv:1511.06448 (2015)
  2. 2.
    Brinkmann, B.H., et al.: Crowdsourcing reproducible seizure forecasting in human and canine epilepsy. Brain 139(6), 1713–1722 (2016)CrossRefGoogle Scholar
  3. 3.
    Caruana, R.: Multitask learning. Mach. Learn. 28(1), 41–75 (1997)MathSciNetCrossRefGoogle Scholar
  4. 4.
    Chen, T., Guestrin, C.: XGBoost: a scalable tree boosting system. In: Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 785–794. ACM (2016)Google Scholar
  5. 5.
    Doersch, C., Zisserman, A.: Multi-task self-supervised visual learning. In: The IEEE International Conference on Computer Vision (ICCV) (2017)Google Scholar
  6. 6.
    Donahue, J., et al.: Long-term recurrent convolutional networks for visual recognition and description. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2625–2634 (2015)Google Scholar
  7. 7.
    Golmohammadi, M., et al.: Deep architectures for automated seizure detection in scalp EEGs. arXiv preprint arXiv:1712.09776 (2017)
  8. 8.
    Graves, A., Liwicki, M., Bunke, H., Schmidhuber, J., Fernández, S.: Unconstrained on-line handwriting recognition with recurrent neural networks. In: Advances in neural information processing systems, pp. 577–584 (2008)Google Scholar
  9. 9.
    Graves, A., Mohamed, A.-R., Hinton, G.: Speech recognition with deep recurrent neural networks. In: 2013 IEEE International Conference on Acoustics Speech and Signal Processing (ICASSP), pp. 6645–6649. IEEE (2013)Google Scholar
  10. 10.
    Hochreiter, S., Schmidhuber, J.: Long short-term memory. Neural Comput. 9(8), 1735–1780 (1997)CrossRefGoogle Scholar
  11. 11.
    Jaffe, A.S.: Long short-term memory recurrent neural networks for classification of acute hypotensive episodes. Ph.D. thesis, Massachusetts Institute of Technology (2017)Google Scholar
  12. 12.
    kaggle: American epilepsy society seizure prediction challenge. https://www.kaggle.com/c/seizure-prediction/data
  13. 13.
    kaggle: Melbourne university aes/mathworks/nih seizure prediction. https://www.kaggle.com/c/melbourne-university-seizure-prediction/data
  14. 14.
    van der Maaten, L., Hinton, G.: Visualizing data using t-SNE. J. Mach. Learn. Res. 9(Nov), 2579–2605 (2008)zbMATHGoogle Scholar
  15. 15.
    Mormann, F., Andrzejak, R.G., Elger, C.E., Lehnertz, K.: Seizure prediction: the long and winding road. Brain 130(2), 314–333 (2006)CrossRefGoogle Scholar
  16. 16.
    O’Regan, S., Faul, S., Marnane, W.: Automatic detection of EEG artefacts arising from head movements using EEG and gyroscope signals. Med. Eng. Phys. 35(7), 867–874 (2013)CrossRefGoogle Scholar
  17. 17.
    Ranjan, R., Patel, V.M., Chellappa, R.: Hyperface: a deep multi-task learning framework for face detection, landmark localization, pose estimation, and gender recognition. IEEE Trans. Pattern Anal. Mach. Intell. (2017)Google Scholar
  18. 18.
    Tahernezhad-Javazm, F., Azimirad, V., Shoaran, M.: A review and experimental study on the application of classifiers and evolutionary algorithms in EEG-based brain-machine interface systems. J. Neural Eng. 15(2), 021007 (2018)CrossRefGoogle Scholar
  19. 19.
    Thodoroff, P., Pineau, J., Lim, A.: Learning robust features using deep learning for automatic seizure detection. In: Machine Learning for Healthcare Conference, pp. 178–190 (2016)Google Scholar
  20. 20.
    Tzallas, A.T., Tsipouras, M.G., Fotiadis, D.I.: Epileptic seizure detection in EEGs using time-frequency analysis. IEEE Trans. Inf. Technol. Biomed. 13(5), 703–710 (2009)CrossRefGoogle Scholar
  21. 21.
    Van Esbroeck, A., Smith, L., Syed, Z., Singh, S., Karam, Z.: Multi-task seizure detection: addressing intra-patient variation in seizure morphologies. Mach. Learn. 102(3), 309–321 (2016)MathSciNetCrossRefGoogle Scholar
  22. 22.
    Wang, Y., et al.: A cauchy-based state-space model for seizure detection in EEG monitoring systems. IEEE Intell. Syst. 30(1), 6–12 (2015)CrossRefGoogle Scholar

Copyright information

© Springer Nature Switzerland AG 2018

Authors and Affiliations

  1. 1.Research Center for Brain-inspired Intelligence and National Laboratory of Pattern Recognition, Institute of AutomationChinese Academy of SciencesBeijingChina
  2. 2.School of Computer and Information EngineeringBeijing Technology and Business UniversityBeijingChina
  3. 3.University of Chinese Academy of SciencesBeijingChina
  4. 4.Center for Excellence in Brain Science and Intelligence TechnologyChinese Academy of SciencesBeijingChina

Personalised recommendations