Multiclass EEG signal classification utilizing Rényi min-entropy-based feature selection from wavelet packet transformation

Brain Informatics - Tập 7 Số 1 - 2020
Md. Asadur Rahman1, Farzana Khanam2, Mohiuddin Ahmad3, Mohammad Shorif Uddin4
1Department of Biomedical Engineering, Military Institute of Science & Technology (MIST), Mirpur Cantonment, Dhaka, 1216, Bangladesh
2Department of Biomedical Engineering, Jashore University of Science and Technology (JUST), Jashore, 7408, Bangladesh
3Department of Electrical and Electronic Engineering, Khulna University of Engineering & Technology (KUET), Khulna 9203, Bangladesh
4Department of Computer Science and Engineering, Jahangirnagar University, Dhaka, Bangladesh

Tóm tắt

AbstractThis paper proposes a novel feature selection method utilizing Rényi min-entropy-based algorithm for achieving a highly efficient brain–computer interface (BCI). Usually, wavelet packet transformation (WPT) is extensively used for feature extraction from electro-encephalogram (EEG) signals. For the case of multiple-class problem, classification accuracy solely depends on the effective feature selection from the WPT features. In conventional approaches, Shannon entropy and mutual information methods are often used to select the features. In this work, we have shown that our proposed Rényi min-entropy-based approach outperforms the conventional methods for multiple EEG signal classification. The dataset of BCI competition-IV (contains 4-class motor imagery EEG signal) is used for this experiment. The data are preprocessed and separated as the classes and used for the feature extraction using WPT. Then, for feature selection Shannon entropy, mutual information, and Rényi min-entropy methods are applied. With the selected features, four-class motor imagery EEG signals are classified using several machine learning algorithms. The results suggest that the proposed method is better than the conventional approaches for multiple-class BCI.

Từ khóa


Tài liệu tham khảo

Rahman MA, Uddin MS, Ahmad M (2019) Modeling and classification of voluntary and imagery movements for brain–computer interface from fNIR and EEG signals through convolutional neural network. Health Inf Sci Syst 7:22. https://doi.org/10.1007/s13755-019-0081-5

Rahman MA, Rashid MA, Ahmad M (2019) Selecting the optimal conditions of Savitzky-Golay filter for fNIRS signal. Biocyberne Biomed Eng 39(3):624–637. https://doi.org/10.1016/j.bbe.2019.06.004

Nicolas-Alonso LF, Gomez-Gil J (2012) Brain computer interfaces, a review. Sensors 12:1211–1279

Rahman MA, Hossain MK, Khanam F, Alam MK, Ahmad M (2019) Four-class motor imagery EEG signal classification using PCA, wavelet, and two-stage neural network. Int J Adv Comput Sci Appl 10:5. https://doi.org/10.14569/ijacsa.2019.0100562

Weibo Y, Qiu S, Qi H, Zhang L, Wan B, Ming D (2013) EEG feature comparison and classification of simple and compound limb motor imagery. J NeuroEng Rehabil 10:106. https://doi.org/10.1186/1743-0003-10-106

Seok WJ, Muller KR, Lee SW (2015) Classifying directions in continuous arm movement from EEG signals. In: 3rd International winter conference on brain computer interface (BCI), pp. 1–2

Liao K, Xiao R, Gonzalez J, Ding L (2014) Decoding individual finger movements from one hand using human EEG signals. PLoS ONE 9(1)

Ma Y, Ding X, She Q, Luo Z, Potter T, Zhang Y (2016) Classification of motor imagery EEG signals with support vector machines and particle swarm optimization. Comput Math Methods Med 2016:1–8

Tang Z, Li C, Wu J, Liu P, Cheng S ( 2018) Classification of EEG-based single-trial motor imagery tasks using a B-CSP method for BCI. Front Inf Technol Electron Eng

Djemal R, Bazyed AG, Belwafi K, Gannouni S, Kaaniche W (2016) Three-class EEG-based motor imagery classification using phase-space reconstruction technique. Brain Sci 6(36):1–19

Petersen J, Iversen HK, Puthusserypady S (2018) Motor imagery based brain-computer interface paradigm for upper limb stroke rehabilitation. In: International conference of the IEEE Engineering in Medicine and Biology Society (EMBC), Honolulu, HI, pp 1960–1963

Li M, Zhu W, Liu H, Yang J (2017) Adaptive feature extraction of motor imagery EEG with optimal wavelet packets and SE-isomap. Appl Sci 7(390):1–18

Cao L, Xia B, Maysam O, Li J, Xie H, Birbaumer N (2017) A synchronous motor imagery based neural physiological paradigm for brain-computer interface speller. Front Hum Neurosci 11(274):1–9

Mahmood A, Zainab R, Ahmad RB, Saeed M, Kamboh AM (2017) Classification of multi-class motor imagery EEG using four band common spatial pattern. In: annual international conference of the IEEE Engineering in Medicine and Biology Society (EMBC), Seogwipo, pp 1034–1037

Ge S, Wang R, Yu D (2014) Classification of four-class motor imagery employing single-channel electroencephalography. PLoS ONE 9(6):1–7

Wang L, Wu X (2008) Classification of four-class motor imagery EEG data using spatial filtering. In: International conference on bioinformatics and biomedical engineering, Shanghai, pp 2153–2156

Zhang Y, Liu B, Ji X, Huang D (2017) Classification of EEG signals based on autoregressive model and wavelet packet decomposition. Neural Process Lett 45(2):365–378

Zhang Y, Ji X, Zhang Y (2015) Classification of EEG signals based on AR model and approximate entropy. In: International joint conference on neural networks (IJCNN), Killarney, pp 1–6

Saha PK, Rahman MA, Mollah MN (2019) Frequency domain approach in CSP based feature extraction for EEG signal classification. In: International conference on electrical, computer and communication engineering (ECCE), 7–9 February 2019, Cox’s Bazar, Bangladesh

Munoz-Gutierrez PA, Giraldo E, Bueno-Lopez M, Molonas M (2018) Localization of active brain sources from EEG signals using empirical mode decomposition: a comparative study. 12(55):1–14

Sokhal J, Garg B, Aggarwal S, Jain R (2017) Classification of EEG signals using empirical mode decomposition and lifting wavelet transforms. In: international conference on computing, communication, and automation (ICCCA), Greater Noida, pp 1197–1202

Nazneen T, Rahman MA, Mollah MN (2019) Towards the effective intrinsic mode functions for motor imagery EEG signal classification. In: International conference on electrical, computer and communication engineering (ECCE), 7–9 February 2019, Cox’s Bazar, Bangladesh

Gaur P, Pachori R, Wang H, Prasad G (2018) A multi-class EEG-based BCI classification using multivariate empirical mode decomposition based filtering and Riemannian geometry. Expert Syst Appl 95:201–211

Gaur P, Pachori RB, Wang H, Prasad G (2016) A multivariate empirical mode decomposition based filtering for subject independent BCI. In: 2016 27th Irish signals and systems conference (ISSC), Londonderry, pp 1–7

Jing J, Miao Y, Daly I, Zuo C, Huc D, Cichocki A (2019) Correlation-based channel selection and regularized feature optimization for MI-based BCI. Neural Netw 118:262–270

Rahman MA, Hossain MF, Hossain M, Ahmmed R (2019) Employing PCA and t-statistical approach for feature extraction and classification of emotion from multichannel EEG signal. Egypt Inform J. https://doi.org/10.1016/j.eij.2019.10.002

Amin HU, Malik AS, Ahmad RF, Badruddin N, Kamel N, Hussain M, Chooi W-T (2015) Feature extraction and classification for EEG signals using wavelet transform and machine learning techniques. Australas Phys Eng Sci Med 38(1):139–149

Sherwani F, Shanta S, Ibrahim BSKK, Huq MS (2016) Wavelet-based feature extraction for classification of motor imagery signals. In: IEEE EMBS conference on biomedical engineering and sciences (IECBES), Kuala Lumpur, pp 360–364

Sharmila A, Mahalakshmi P (2017) Wavelet-based feature extraction for classification of epileptic seizure EEG signal. J Med Eng Technol 41(8):670–680

Rahman MA, Rashid MMO, Khanam F, Alam MK, Ahmad M (2019) EEG based brain alertness monitoring by statistical and artificial neural network approach. Int J Adv Comput Sci Appl 10:1

Coifman RR, Wickerhauser MV (1992) Entropy-based algorithms for best basis selection. IEEE Trans Inf Theory 38(2):713–718

Khushaba RN, Kodagoda S, Lal S, Dissanayake G (2011) Driver drowsiness classification using fuzzy wavelet-packet-based feature-extraction algorithm. IEEE Trans Biomed Eng 58(1):121–131

Saito N (1994) Local feature extraction and its applications using a library of bases. Ph.D. Thesis, Department of Mathematics, Yale Univ., New Haven, CT

Deqiang L, Pedrycz W, Pizzi NJ (2005) Fuzzy wavelet packet based feature extraction method and its application to biomedical signal classification. IEEE Trans Biomed Eng 52(6):1132–1139

Schalko R (1992) Pattern recognition: statistical, structural and neural approaches. Wiley, New York

Graz University BCI Competition data sets IV. http://www.bbci.de/competition/iv/

Muo UE, Madamedon M, Ball AD, Gu F (2017) Wavelet packet analysis and empirical mode decomposition for the fault diagnosis of reciprocating compressors. In: 23rd international conference on automation and computing (ICAC), Huddersfield, pp 1–6

Wang D, Yue C, Wei S, Lv J (2017) Performance analysis of four decomposition-ensemble models for one-day-ahead agricultural commodity futures price forecasting. Algorithms 10(3):1–25. https://doi.org/10.3390/a10030108

Mammone N, Morabito FC (2014) Enhanced automatic wavelet independent component analysis for electroencephalographic artifact removal. Entropy 16(12):6553–6572. https://doi.org/10.3390/e1612.6553

Wali MK, Murugappan M, Ahmmad B (2013) Wavelet packet transform based driver distraction level classification using EEG. Math Probl Eng 2013, Article ID 297587

Rényi A (1961) On measures of entropy and information. In: Proceedings of the 4th Berkeley symposium on mathematics, statistics, and probability, pp 547–561, https://projecteuclid.org/euclid.bsmsp/1200512181

Smith G (2009) On the foundations of quantitative information flow. In: de Alfaro L (ed) Foundations of software science and computational structures. FoSSaCS 2009. Lecture Notes in Computer Science, vol 5504. Springer, Berlin. https://doi.org/10.1007/978-3-642-00596-1_21

Brown G, Pocock AC, Zhao MJ, Luján M (2012) Conditional likelihood maximisation: a unifying framework for information theoretic feature selection. J Mach Learn Res 13:27–66

Vergara JR, Estévez PA (2014) A review of feature selection methods based on mutual information. Neural Comput Appl 24(1):175–186

Palamidessi C, Romanelli M (2018) Feature selection with Rényi min-entropy. In: Artificial neural networks in pattern recognition—8th IAPR TC3 workshop (ANNPR 2018), Sep 2018, Siena, Italy. Springer, Lecture Notes in Computer Science, 11081, pp 226–239

Khushaba R (2019) Feature extraction using multisignal wavelet transform decomposition. MATLAB Central File Exchange. https://www.mathworks.com/matlabcentral/fileexchange/37950-feature-extraction-using-multisignal-wavelet-transform-decomposition. Accessed 29 Dec 2019