Abstract
EEG signal classification play an important role in recognition of epilepsy. Recently, dictionary learning algorithms have shown the effectiveness in this field. When designing dictionaries, due to highly non-stationary of EEG signals, and collecting signals existing in different stimulus and drug modes, training and testing scenarios may be different. Thus, the performance of classical dictionary learning algorithms is unsatisfactory. In this paper, a transfer discriminative dictionary learning with label consistency (called TDDLLC) algorithm is proposed for EEG signal classification. Since each EEG signal can be represented as a linear combination of dictionary atoms, and some atoms are dataset independent, two dictionaries are learned simultaneously in source domain (SD) and target domain (TD) respectively where the discrepancy between two dictionaries is minimized. Meanwhile, utilizing the label information of samples in SD and a small number of labeled samples in TD, these dictionaries are learned with the aim of achieving discriminative abilities. To avoid the NP-hard problem, \(\ell_{1}\)-norm regularization term is used in TDDLLC, and objective function is solved by block-coordinate descent method. Extensive experiments have been performed on Bonn dataset and show the validity of the TDDLLC algorithm.
Similar content being viewed by others
References
Abolghasemi V, Ferdowsi S (2015) EEG-fMRI: dictionary learning for removal of ballistocardiogram artifact from EEG. Biomed Signal Process Control 18(4):186–194
Akhavan A, Moradi MH (2018) Detection of concealed information using multichannel discriminative dictionary and spatial filter learning. IEEE Trans Inform Forensics Secur 13(10):2616–2627
Ameri R, Pouyan A, Abolghasemi V (2016) Projective dictionary pair learning for EEG signal classification in brain computer interface applications. Neurocomputing 218(12):382–389
Amin HU, Yusoff MZ, Ahmad RF (2020) A novel approach based on wavelet analysis and arithmetic coding for automated detection and diagnosis of epileptic seizure in EEG signals using machine learning techniques. Biomed Signal Process Control Biomed Signal Process Control 56(2):1–10
Andrzejak RG, Lehnertz K, Rieke C, Mormann F, David P, Elger CE (2001) Indications of nonlinear deterministic and finite dimensional structures in time series of brain electrical activity: dependence on recording region and brain state. Phys Rev E 64:061907. https://doi.org/10.1103/PhysRevE.64.061907
Chen G, Xiong C, Corso J J (2012). Dictionary transfer for image denoising via domain adaptation, in Proceedings of 2012 19th IEEE International Conference on Image Processing ICIP 2012, Orlando, USA, 1189–1192, doi: https://doi.org/10.1109/ICIP.2012.6467078
Deng Z, Xu P, Xie L, Choi KS, Wang S (2018) Transductive joint-knowledge-transfer TSK FS for recognition of epileptic EEG signals. IEEE Trans Neural Syst Rehab Eng 26(8):1481–1494
Gu X, Chung F, Wang S (2017) Bayesian Takagi–Sugeno–Kang fuzzy classifier. IEEE Trans Fuzzy Syst 25(6):1655–1671
Gu X, Zhang C, Ni T (2020) A hierarchical discriminative sparse representation classifier for EEG signal detection. IEEE/ACM Trans Comput Biol Bioinform. https://doi.org/10.1109/TCBB.2020.2973978
Hu W, Cao J, Lai X, Liu J (2019) Mean amplitude spectrum based epileptic state classification for seizure prediction using convolutional neural networks. J Ambient Intell Hum Comput. https://doi.org/10.1007/s12652-019-01220-6
Jiang Z, Lin Z, Davis LS (2013) Label consistent K-SVD: learning a discriminative dictionary for recognition. IEEE Trans Pattern Anal Mach Intell 35(11):2651–2664
Jiang Y, Deng Z, Chung F, Wang G, Qian P, Choi KS, Wang S (2017) Recognition of epileptic EEG signals using a novel multiview TSK fuzzy system. IEEE Trans Fuzzy Syst 25(1):3–20
Jiang Y, Wu D, Deng Z, Qian P, Wang J, Wang G, Chung F, Sze CK, Wang S (2017) Seizure classification from EEG signals using transfer learning, semi-supervised learning and TSK fuzzy system. IEEE Trans Neural Syst Rehab Eng 25(12):2270–2284
Jiang Z, Chung F, Wang S (2019) Recognition of multiclass epileptic EEG signals based on knowledge and label space inductive transfer. IEEE Trans Neural Syst Rehab Eng 27(4):630–642
Kevric J, Subasi A (2017) Comparison of signal decomposition methods in classification of EEG signals for motor-imagery BCI system. Biomed Signal Process Control 31(1):398–406
Kundishora AJ, Gummadavelli A, Ma C, Liu M, Mccafferty C, Schiff ND, Willie JT, Gross RE, Gerrard J, Hal B (2017) Restoring conscious arousal during focal limbic seizures with deep brain stimulation. Cereb Cortex 27(3):1964–1975
Liu B D, Wang Y X, Shen B, Zhang Y J, Wang Y J (2014). Blockwise coordinate descent schemes for sparse representation, in Proceedings of 2014 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), Florence, Italy, 5267–5271. doi:https://doi.org/10.1109/ICASSP.2014.6854608
Mendoza PF, Menezes ML, Santanna A, Ortiz BM, Samara A, Galway L (2019) Affective recognition from EEG signals: an integrated data-mining approach. J. Ambient Intell. Hum. Comput. 10:3955–3974
Ni TG, Gu XQ, Wang HY, Zhang Z, Chen SB, Jin C (2018) Discriminative deep transfer metric learning for cross-scenario person reidentification. J Electron Imaging 27(4):043026. https://doi.org/10.1117/1.JEI.27.4.043026
Ni TG, Gu XQ, Wang J, Zheng YH, Wang HY (2018) Scalable transfer support vector machine with group probabilities. Neurocomputing 273(17):570–582
Pan SJ, Yang Q (2010) A survey on transfer learning. IEEE Trans Knowl Data Eng 22(10):1345–1359
Quanz B, Huan, J (2009) Large margin transductive transfer learning, in Proceedings of the 18th ACM conference on Information and knowledge management, Hong Kong, China: 1327–1336. doi: https://doi.org/10.1145/1645953.1646121
Seo J, Laine TH, Sohn K (2019) Machine learning approaches for boredom classification using EEG. J Ambient Intell Hum Comput 10:3831–3846
Shao S, Xu R, Liu W, Di LB, Wang YJ (2020) Label embedded dictionary learning for image classification. Neurocomputing 385(14):122–131
Sreej SR, Samanta D (2019) Classification of multiclass motor imagery EEG signal using sparsity approach. Neurocomputing 368(11):133–145
Sreej SR, Samanta D, Sarma M (2019a) Weighted sparse representation for classification of motor imagery EEG signals, in Proceedings of the 2019 41st Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), Berlin, Germany: 23–27. doi: https://doi.org/10.1109/EMBC.2019.8857496
Sridhar KP, Baskar S, Shakeel PM, Dhulipala VR (2019) Developing brain abnormality recognize system using multi-objective pattern producing neural network. J Ambient Intell Hum Comput 10:3287–3295
Wang SH, Zhang Y, Li YJ, Jia WJ, Liu FY, Yang MM, Zhang YD (2018) Single slice based detection for Alzheimer’s disease via wavelet entropy and multilayer perceptron trained by biogeography-based optimization. Multimed Tools Appl 77(5):10393–10417
Wang SH, Zhang YD, Yang M, Liu B, Ramirez JG, Juan M (2019) Unilateral sensorineural hearing loss identification based on double-density dual-tree complex wavelet transform and multinomial logistic regression. Integr Comput-Aided Eng 26(4):411–426
Wang SH, Xie SP, Guttery DS, Tang CS, Sun JD (2019) Zhang Y D (2019b) Alcoholism identification based on an AlexNet transfer learning model. Front Psychiatry 10:205. https://doi.org/10.3389/fpsyt.2019.00205
Xia K, Ni T, Yin H, Chen B (2020) Cross-domain classification model with knowledge utilization maximization for recognition of epileptic EEG signals. IEEE/ACM Trans Comput Biol Bioinform. https://doi.org/10.1109/TCBB.2020.2973978
Yang C, Deng Z, Choi KS, Jiang Y, Wang S (2014) Transductive domain adaptive learning for epileptic electroencephalogram recognition. Artif Intell Med 62(3):165–177
Zhang Q, Li B (2010) Discriminative K-SVD for dictionary learning in face recognition, in Proceedings of the Conference on Computer Vision and Pattern Recognition (CVPR), San Francisco, USA: 2691–2698. Doi: https://doi.org/10.1109/CVPR.2010.5539989
Zhang YD, Wang SH, Sui YX, Yang M, Liu B, Cheng H, Sun JD, Jia WJ, Phillips P, Gorriz JM (2018) Multivariate approach for Alzheimer’s disease detection using stationary wavelet entropy and predator-prey particle swarm optimization. J. Alzheimer’s Dis 65(3):855–869
Zhang Z, Jiang W, Qin J, Zhang L, Yan S (2018) Jointly learning structured analysis discriminative dictionary and analysis multiclass classifier. IEEE Trans Neural Netw Learn Syst 29(8):3798–3814
Zhang YD, Govindaraj VV, Tang CS, Zhu WG, Sun JD (2019) High performance multiple sclerosis classification by data augmentation and AlexNet transfer learning model. J Med Imaging Health Inform 9(9):2012–2021
Zheng J, Jiang Z, Chellappa R (2016) Cross-view action recognition via transferable dictionary learning. IEEE Trans Image Process 25(6):2542–2556
Zhu F, Shao L, Fang Y (2016) Boosted cross-domain dictionary learning for visual categorization. IEEE Intell Syst 31(3):6–18
Acknowledgements
This work was supported in part by the National Natural Science Foundation of China under Grant 61806026, by the Natural Science Foundation of Jiangsu Province under Grant BK 20180956, by the 2018 Six Talent Peaks Project of Jiangsu Province under Grant XYDXX-127.
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interest
The author declares no conflict of interest.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
Ni, T., Gu, X. & Jiang, Y. Transfer discriminative dictionary learning with label consistency for classification of EEG signals of epilepsy. J Ambient Intell Human Comput 14, 5529–5540 (2023). https://doi.org/10.1007/s12652-020-02620-9
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s12652-020-02620-9