Feature extraction and classification for EEG signals using wavelet transform and machine learning techniques

  • Hafeez Ullah Amin
  • Aamir Saeed MalikEmail author
  • Rana Fayyaz Ahmad
  • Nasreen Badruddin
  • Nidal Kamel
  • Muhammad Hussain
  • Weng-Tink Chooi
Scientific Paper


This paper describes a discrete wavelet transform-based feature extraction scheme for the classification of EEG signals. In this scheme, the discrete wavelet transform is applied on EEG signals and the relative wavelet energy is calculated in terms of detailed coefficients and the approximation coefficients of the last decomposition level. The extracted relative wavelet energy features are passed to classifiers for the classification purpose. The EEG dataset employed for the validation of the proposed method consisted of two classes: (1) the EEG signals recorded during the complex cognitive task—Raven’s advance progressive metric test and (2) the EEG signals recorded in rest condition—eyes open. The performance of four different classifiers was evaluated with four performance measures, i.e., accuracy, sensitivity, specificity and precision values. The accuracy was achieved above 98 % by the support vector machine, multi-layer perceptron and the K-nearest neighbor classifiers with approximation (A4) and detailed coefficients (D4), which represent the frequency range of 0.53–3.06 and 3.06–6.12 Hz, respectively. The findings of this study demonstrated that the proposed feature extraction approach has the potential to classify the EEG signals recorded during a complex cognitive task by achieving a high accuracy rate.


Discrete wavelet transform (DWT) Machine learning classifiers Electroencephalography (EEG) Cognitive task 



This research work has been supported by University Research Internal Funding (URIF: 0153AA-B26), Universiti Teknologi PETRONAS; the Fundamental Research Grant Scheme (Ref: FRGS/1/2014/TK03/UTP/02/1), Ministry of Education (MOE), Malaysia and by NSTIP strategic technologies programs, Grant number (12-INF2582-02), in the Kingdom of Saudi Arabia.


  1. 1.
    Übeyli ED (2009) Combined neural network model employing wavelet coefficients for EEG signals classification. Digit Signal Proc 19:297–308CrossRefGoogle Scholar
  2. 2.
    Orhan U, Hekim M, Ozer M (2011) EEG signals classification using the K-means clustering and a multilayer perceptron neural network model. Expert Syst Appl 38:13475–13481CrossRefGoogle Scholar
  3. 3.
    Iscan Z, Dokur Z, Demiralp T (2011) Classification of electroencephalogram signals with combined time and frequency features. Expert Syst Appl 38:10499–10505CrossRefGoogle Scholar
  4. 4.
    Liang N-Y, Saratchandran P, Huang G-B, Sundararajan N (2006) Classification of mental tasks from EEG signals using extreme learning machine. Int J Neural Syst 16:29–38CrossRefPubMedGoogle Scholar
  5. 5.
    Rosso OA, Blanco S, Yordanova J, Kolev V, Figliola A, Schürmann M et al (2001) Wavelet entropy: a new tool for analysis of short duration brain electrical signals. J Neurosci Methods 105:65–75CrossRefPubMedGoogle Scholar
  6. 6.
    Yazdani A, Ebrahimi T, Hoffmann U (2009) Classification of EEG signals using Dempster Shafer theory and a k-nearest neighbor classifier. In: 4th International IEEE/EMBS conference on neural engineering, pp 327–330Google Scholar
  7. 7.
    Garry H, McGinley B, Jones E, Glavin M (2013) An evaluation of the effects of wavelet coefficient quantisation in transform based EEG compression. Comput Biol Med 43:661–669CrossRefPubMedGoogle Scholar
  8. 8.
    Guo L, Wu Y, Zhao L, Cao T, Yan W, Shen X (2011) Classification of mental task from EEG signals using immune feature weighted support vector machines. IEEE Trans Magn 47:866–869CrossRefGoogle Scholar
  9. 9.
    Taghizadeh-Sarabi M, Daliri MR, Niksirat KS (2014) Decoding objects of basic categories from electroencephalographic signals using wavelet transform and support vector machines. Brain Topogr 1–14Google Scholar
  10. 10.
    Acharya UR, Sree SV, Ang PCA, Yanti R, Suri JS (2012) Application of non-linear and wavelet based features for the automated identification of epileptic EEG signals. Int J Neural Syst 22:1250002CrossRefPubMedGoogle Scholar
  11. 11.
    Gandhi T, Panigrahi BK, Anand S (2011) A comparative study of wavelet families for EEG signal classification. Neurocomputing 74:3051–3057CrossRefGoogle Scholar
  12. 12.
    Richman JS, Moorman JR (2000) Physiological time-series analysis using approximate entropy and sample entropy. Am J Physiol Heart Circ Physiol 278:H2039–H2049PubMedGoogle Scholar
  13. 13.
    Vidaurre C, Krämer N, Blankertz B, Schlögl A (2009) Time domain parameters as a feature for EEG-based brain–computer interfaces. Neural Netw 22:1313–1319CrossRefPubMedGoogle Scholar
  14. 14.
    Übeyli ED (2010) Lyapunov exponents/probabilistic neural networks for analysis of EEG signals. Expert Syst Appl 37:985–992CrossRefGoogle Scholar
  15. 15.
    Thatcher RW, North D, Biver C (2005) EEG and intelligence: relations between EEG coherence, EEG phase delay and power. Clin Neurophysiol 116:2129–2141CrossRefPubMedGoogle Scholar
  16. 16.
    Hariharan M, Vijean V, Sindhu R, Divakar P, Saidatul A, Yaacob S (2014) Classification of mental tasks using stockwell transform. Comput Electr Eng 40:1741CrossRefGoogle Scholar
  17. 17.
    Noshadi S, Abootalebi V, Sadeghi MT, Shahvazian MS (2014) Selection of an efficient feature space for EEG-based mental task discrimination. Biocybern Biomed Eng 34:159–168CrossRefGoogle Scholar
  18. 18.
    Zhang L, He W, He C, Wang P (2010) Improving mental task classification by adding high frequency band information. J Med Syst 34:51–60CrossRefPubMedGoogle Scholar
  19. 19.
    Hosni SM, Gadallah ME, Bahgat SF, AbdelWahab MS (2007) Classification of EEG signals using different feature extraction techniques for mental-task BCI. In: International conference on computer engineering & systems, 2007. ICCES’07, pp 220–226Google Scholar
  20. 20.
    Xue J-Z, Zhang H, Zheng C-x, Yan X-G (2003) Wavelet packet transform for feature extraction of EEG during mental tasks. In: International conference on machine learning and cybernetics, 2003, pp 360–363Google Scholar
  21. 21.
    Zhiwei L, Minfen S (2007) Classification of mental task EEG signals using wavelet packet entropy and SVM. In: 8th International conference on electronic measurement and instruments, 2007. ICEMI’07, pp 3-906–3-909Google Scholar
  22. 22.
    Keirn ZA, Aunon JI (1990) A new mode of communication between man and his surroundings. IEEE Trans Biomed Eng 37:1209–1214CrossRefPubMedGoogle Scholar
  23. 23.
    Nai-Jen H, Palaniappan R (2004) Classification of mental tasks using fixed and adaptive autoregressive models of EEG signals. In: 26th Annual international conference of the IEEE engineering in medicine and biology society, 2004. IEMBS’04, pp 507–510Google Scholar
  24. 24.
    Lin C-J, Hsieh M-H (2009) Classification of mental task from EEG data using neural networks based on particle swarm optimization. Neurocomputing 72:1121–1130CrossRefGoogle Scholar
  25. 25.
    Rodríguez-Bermúdez G, García-Laencina PJ, Roca-González J, Roca-Dorda J (2013) Efficient feature selection and linear discrimination of EEG signals. Neurocomputing 115:161–165CrossRefGoogle Scholar
  26. 26.
    Karkare S, Saha G, Bhattacharya J (2009) Investigating long-range correlation properties in EEG during complex cognitive tasks. Chaos Solitons Fractals 42:2067–2073CrossRefGoogle Scholar
  27. 27.
    Jahidin A, Ali MM, Taib MN, Tahir NM, Yassin IM, Lias S (2014) Classification of intelligence quotient via brainwave sub-band power ratio features and artificial neural network. Comput Methods Programs Biomed 114:50–59CrossRefPubMedGoogle Scholar
  28. 28.
    Amin HU, Malik AS, Subhani AR, Badruddin N, Chooi W-T (2013) Dynamics of scalp potential and autonomic nerve activity during intelligence test. In: Lee M et al (eds) Neural information processing, vol 8226. Springer, Berlin, pp 9–16CrossRefGoogle Scholar
  29. 29.
    Raven J (2000) The Raven’s progressive matrices: change and stability over culture and time. Cogn Psychol 41:1–48CrossRefPubMedGoogle Scholar
  30. 30.
    Kunda M, McGreggor K, Goel A (2012) Reasoning on the Raven’s advanced progressive matrices test with iconic visual representations. In: 34th Annual conference of the cognitive science society pp 1828–1833Google Scholar
  31. 31.
    Amin HU, Malik AS, Badruddin N, Chooi W-T (2013) EEG mean power and complexity analysis during complex mental task. In: ICME International conference on complex medical engineering (CME) pp 648–651Google Scholar
  32. 32.
    Jahankhani P, Kodogiannis V, Revett K (2006) EEG signal classification using wavelet feature extraction and neural networks. In: IEEE John Vincent Atanasoff 2006 international symposium on modern computing, 2006. JVA’06. pp 120–124Google Scholar
  33. 33.
    Subasi A (2007) EEG signal classification using wavelet feature extraction and a mixture of expert model. Expert Syst Appl 32:1084–1093CrossRefGoogle Scholar
  34. 34.
    Gonzalez RC, Woods RE (2002) Digital image processing, 2nd edn. Prentice Hall, SLGoogle Scholar
  35. 35.
    Hausfeld L, De Martino F, Bonte M, Formisano E (2012) Pattern analysis of EEG responses to speech and voice: influence of feature grouping. Neuroimage 59:3641–3651CrossRefPubMedGoogle Scholar
  36. 36.
    Ben-Hur A, Weston J (2010) A user’s guide to support vector machines. In: Data mining techniques for the life sciences, Springer, New York, pp 223–239Google Scholar
  37. 37.
    Pereira F, Mitchell T, Botvinick M (2009) Machine learning classifiers and fMRI: a tutorial overview. Neuroimage 45:S199–S209CrossRefPubMedCentralPubMedGoogle Scholar
  38. 38.
    Subasi A, Gursoy MI (2010) EEG signal classification using PCA, ICA, LDA and support vector machines. Expert Syst Appl 37:8659–8666CrossRefGoogle Scholar
  39. 39.
    Witten IH, Frank E (2005) Data mining: practical machine learning tools and techniques. Morgan Kaufmann, BurlingtonGoogle Scholar
  40. 40.
    Daud M, Yunus J (2004) Classification of mental tasks using de-noised EEG signals. In: 7th International conference on signal processing, pp 2206–2209Google Scholar
  41. 41.
    Guo L, Rivero D, Seoane JA, Pazos A (2009) Classification of EEG signals using relative wavelet energy and artificial neural networks. In: Proceedings of the first ACM/SIGEVO summit on genetic and evolutionary computation, pp 177–184Google Scholar
  42. 42.
    Diez PF, Mut V, Laciar E, Torres A, Avila E (2009) Application of the empirical mode decomposition to the extraction of features from EEG signals for mental task classification. In: Annual international conference of the IEEE engineering in medicine and biology society, 2009. EMBC 2009, pp 2579–2582Google Scholar
  43. 43.
    Dimitriadis SI, Laskaris NA, Tsirka V, Vourkas M, Micheloyannis S (2010) What does delta band tell us about cognitive processes: a mental calculation study. Neurosci Lett 483:11–15CrossRefPubMedGoogle Scholar
  44. 44.
    Harper J, Malone SM, Bernat EM (2014) Theta and delta band activity explain N2 and P3 ERP component activity in a go/no-go task. Clin Neurophysiol 125:124–132CrossRefPubMedCentralPubMedGoogle Scholar
  45. 45.
    Amin H, Malik AS (2013) Human memory retention and recall processes: a review of EEG and fMRI studies. Neurosciences 18:330–344PubMedGoogle Scholar
  46. 46.
    Ergen M, Marbach S, Brand A, Başar-Eroğlu C, Demiralp T (2008) P3 and delta band responses in visual oddball paradigm in schizophrenia. Neurosci Lett 440:304–308CrossRefPubMedGoogle Scholar
  47. 47.
    Demiralp T, Ademoglu A, Schürmann M, Basar-Eroglu C, Basar E (1999) Detection of P300 waves in single trials by the wavelet transform (WT). Brain Lang 66:108–128CrossRefPubMedGoogle Scholar
  48. 48.
    Gennady GK (2012) EEG delta oscillations as a correlate of basic homeostatic and motivational processes. Neurosci Biobehav Rev 36:677–695CrossRefGoogle Scholar
  49. 49.
    Baijal S, Srinivasan N (2010) Theta activity and meditative states: spectral changes during concentrative meditation. Cogn Process 11:31–38CrossRefPubMedGoogle Scholar
  50. 50.
    Sakowitz O (2001) The selectively distributed theta system: functions. Int J Psychophysiol 39:197–212CrossRefPubMedGoogle Scholar
  51. 51.
    Nigbur R, Ivanova G, Stürmer B (2011) Theta power as a marker for cognitive interference. Clin Neurophysiol 122:2185–2194CrossRefPubMedGoogle Scholar

Copyright information

© Australasian College of Physical Scientists and Engineers in Medicine 2015

Authors and Affiliations

  • Hafeez Ullah Amin
    • 1
  • Aamir Saeed Malik
    • 1
    Email author
  • Rana Fayyaz Ahmad
    • 1
  • Nasreen Badruddin
    • 1
  • Nidal Kamel
    • 1
  • Muhammad Hussain
    • 2
  • Weng-Tink Chooi
    • 3
  1. 1.Department of Electrical & Electronic Engineering, Centre for Intelligent Signal & Imaging Research (CISIR)Universiti Teknologi PETRONASTronohMalaysia
  2. 2.Department of Computer Science, College of Computer and Information SciencesKing Saud UniversityRiyadhSaudi Arabia
  3. 3.Advanced Medical and Dental Institute (AMDI)Universiti Sains MalaysiaKepala BatasMalaysia

Personalised recommendations