Next Article in Journal
Estimation of the Genome Size and Complete Chloroplast Genome in Adenophora remotiflora: Genome Structures, Comparative Genomics, and Phylogenetic Diversity
Previous Article in Journal
Analysis of Lateral Displacement of Pile Foundation Caused by Large-Diameter Shield Tunneling
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Identifying Patterns for Neurological Disabilities by Integrating Discrete Wavelet Transform and Visualization

1
Department of Computer Science, Bowie State University, Bowie, MD 20715, USA
2
Department of Computer Science, Old Dominion University, Norfolk, VA 23529, USA
3
Department of Human Movement Sciences, Old Dominion University, Norfolk, VA 23529, USA
4
Department of Psychology, Bowie State University, Bowie, MD 20715, USA
5
Department of Computer Science and Information Technology, University of the District of Columbia, Washington, DC 20008, USA
*
Authors to whom correspondence should be addressed.
Appl. Sci. 2024, 14(1), 273; https://doi.org/10.3390/app14010273
Submission received: 17 November 2023 / Revised: 15 December 2023 / Accepted: 24 December 2023 / Published: 28 December 2023
(This article belongs to the Section Biomedical Engineering)

Abstract

:
Neurological disabilities cause diverse health and mental challenges, impacting quality of life and imposing financial burdens on both the individuals diagnosed with these conditions and their caregivers. Abnormal brain activity, stemming from malfunctions in the human nervous system, characterizes neurological disorders. Therefore, the early identification of these abnormalities is crucial for devising suitable treatments and interventions aimed at promoting and sustaining quality of life. Electroencephalogram (EEG), a non-invasive method for monitoring brain activity, is frequently employed to detect abnormal brain activity in neurological and mental disorders. This study introduces an approach that extends the understanding and identification of neurological disabilities by integrating feature extraction, machine learning, and visual analysis based on EEG signals collected from individuals with neurological and mental disorders. The classification performance of four feature approaches—EEG frequency band, raw data, power spectral density, and wavelet transform—is assessed using machine learning techniques to evaluate their capability to differentiate neurological disabilities in short EEG segmentations (one second and two seconds). In detail, the classification analysis is conducted under two conditions: single-channel-based classification and region-based classification. While a clear demarcation between normal (healthy) and abnormal (neurological disabilities) EEG metrics may not be evident, their similarities and distinctions are observed through visualization, employing wavelet features. Notably, the frontal brain region (frontal lobe) emerges as a crucial area for distinguishing abnormalities among different brain regions. Also, the integration of wavelet features and visual analysis proves effective in identifying and understanding neurological disabilities.

1. Introduction

In neuroscience, electroencephalography (EEG) has commonly been used as an early diagnostic tool for examining various abnormal neurological or psychiatric conditions, encompassing attention-deficit/hyperactivity disorder (ADHD), chronic pain, major depressive disorder, and obsessive–compulsive disorder. ADHD is one of the most common neurodevelopmental disabilities affecting both children (approximately 5 % ) and adults (approximately 2.5 % ) [1]. Individuals who have ADHD can experience various symptoms, such as social interaction difficulties, depression, anxiety, learning challenges, and sleep disruptions [2]. Insomnia, a sleep disorder, can contribute to emotional issues, frustration, and learning difficulties [3]. It also causes problems with decision making and poses a risk for heart-related diseases [4]. Chronic pain, persisting for over 12 weeks despite treatment, significantly impacts patients’ lives, leading to depression, presenteeism, and financial burdens on families or caregivers [5]. Major depressive disorder (MDD), the most prevalent mental disorder, adversely affects physical health, sleep, mood, and social activities, potentially leading to suicidal thoughts or attempts [6]. Obsessive–compulsive disorder (OCD), another common mental disorder, diminishes quality of life through persistent intrusive thoughts and repetitive behaviors [7,8]. The direct economic cost of mental or neurological disorders in the United States surpassed USD 210 billion in 2010, with the global economic cost of mental disorders estimated at USD 2.5 trillion [9,10].
Since neurological and mental disorders are increasingly prevalent and closely related to various mental and physical health conditions and deficits in social interaction, the early detection of abnormal conditions is crucial for effective care planning, intervention, and advocacy. EEG is capable of early diagnosis identification and is reliable and valid in accurately detecting neurological and mental disorders. While previous studies have adeptly utilized EEG data for early detection, these studies primarily focused on analyzing and computing techniques for isolating single disorders. Moreover, despite the existence of more than six hundred neurological diseases, limited research has been conducted to identify neurological disorders using a unified framework. This study shifts the focus to comprehending the patterns associated with multiple neurological disabilities by extracting significant features and subjecting them to analysis to enhance our understanding of these disabilities through visualization techniques. We introduce an automated pattern recognition approach that integrates visual analysis and machine learning to analyze neurological disabilities. It is important to note that, in this study, the term ’neurological disability’ encompasses both neurological disorders and mental disorders. To the best of our knowledge, there is no previous study that analyzes multiple disorders based on EEG data while integrating multiple visualization techniques.
EEG signals are recognized as multivariate time series data, comprising a sequence of spatial and temporal information ordered over time. The signals encompass diverse local and global information, enabling the identification of distinctive patterns for various health conditions and the tracking of trends in individual health conditions. This information is particularly valuable for capturing unique characteristics related to neurological disabilities. However, extracting significant features from EEG signals is an essential yet challenging task in neurological studies. Although a few previous studies have demonstrated the potential to differentiate neurological disabilities using classification algorithms, there exists room for performance enhancement and the advancement of knowledge to accurately differentiate disorders from one another. Due to similar symptoms between ADHD and other psychiatric disorders, identifying ADHD through the analysis of neurophysiological signals remains a challenging task [11]. Therefore, this study introduces an approach that integrates machine learning and visualization to determine differences among patients with neurological disabilities. We emphasize the importance of identifying critical features from EEG signals associated with individual neurological disabilities and determining their unique patterns, as this enhances our ability to understand the disabilities. In summary, the primary contributions of this study are outlined as follows:
  • We designed an efficient feature extraction approach, utilizing signal processing techniques to analyze EEG data associated with multiple neurological and mental disorders with short-term EEG segmentation, such as 1 s and 2 s segmentation intervals.
  • Our work is the first study utilizing visualization techniques to enhance understanding of the extracted features to distinguish the disorders.
  • We conducted multiple data classification analyses utilizing machine learning (ML) algorithms, including support vector machine (SVM), random forest (RF), k-nearest neighbors (KNN), and logistic regression (LR).
  • We also performed an extensive evaluation to determine the optimal wavelets and decomposition levels for analyzing and differentiating neurological disabilities.
The rest of the paper is organized as follows: First, we discuss prior research in neurological disorders utilizing EEG in the Previous Works section. Then, the proposed approach is explained in greater detail, emphasizing the importance of utilizing wavelet transform and integrating visualization. After presenting the research findings and implications, we conclude this paper with possible future works.

2. Previous Works

Various studies have been performed to understand neurodevelopmental disorders and analyze neurological diseases. Musser and Nigg [12] conducted an analysis on electrocardiogram (ECG) and impedance cardiography data in individuals with ADHD to explore its connection with emotions. Out of a total of 100 datasets, 50 pertained to individuals with ADHD. They revealed that children, both with and without ADHD, exhibited similar facial affection behaviors. However, those with ADHD demonstrated reduced coherence between facial affection behavior and their parasympathetic function. In comparison to their counterparts without ADHD, children with ADHD may have experienced conflicting emotional signals. While it is acknowledged that genetic factors may strongly contribute to the onset of ADHD [13], researchers emphasize the need for numerous ongoing studies to gain a more accurate understanding of symptoms and facilitate more effective medical treatments. Martin et al. [14] emphasized the existence of gender differences in ADHD. They studied to explore gender differences through quantitative and qualitative analyses of genetic factors. Their research indicated a higher risk of ADHD among siblings of females. In another study, Maniruzzaman et al. [15] investigated an optimal channel selection approach for predicting ADHD using machine learning techniques, including decision tree (DT), KNN, RF, LR, and SVM. The study employed a hybrid approach of SVM and an independent t-test to identify optimal channels, revealing six crucial channels (Fz, F8, F3, C4, C3, and F7) used for predicting ADHD.
Most neurological studies using EEG have focused on building a predictive model with a single neurological disease. For instance, Khare and Acharya [16] examined EEG data to find appropriate brain regions. Specifically, they focused on determining significant EEG channels by employing entropy, statistical measures, and nonlinear features to identify ADHD. They identified the crucial role of frontal regions in ADHD detection and highlighted specific EEG channels (Fz, F7, Pz, P7, and Cz) as significant for ADHD analysis. Coelho et al. [17] evaluated EEG Hjorth features extracted from the brain regions, i.e., the parietal, frontal, central, and occipital lobes, to identify Parkinson’s disease patients. To evaluate the performance of differentiating the patients with Parkinson’s disease and healthy individuals, they utilized classification methods such as SVM, KNN, and RF. Koh et al. [18] introduced an approach to detect ADHD and conduct disorder (CD) using ECG signals by integrating empirical wavelet transform (EWT). They applied feature extraction by measuring entropy and performed feature selection via the analysis of variance (ANOVA) test. They assessed the selected features to determine ADHA and CD using machine learning techniques such as SVM, decision tree (DT), and KNN. Yasin et al. [19] reviewed previous EEG studies focused on MDD and observed a growing demand for applying deep learning (DL) in EEG analysis. The study also underscored the challenges in interpreting results from neural network-based approaches. Mulaffer et al. [20] highlighted the significance of EEG features in insomnia detection by comparing them with hypnogram-based features using SVM. The study employed C3 and C4 channels to assess these features. Although numerous research studies have been performed to identify neurodevelopmental disorders, there is still a need to identify essential features for accurate identification. SVM has been underscored as a crucial technique for analyzing EEG signals in dyslexia detection [21,22]. Despite extensive research, it is still necessary to find essential features to identify them accurately. Seshadri et al. [23] proposed an approach to detect learning disabilities (LD), a neurodevelopmental disorder that severely impacts children’s lives. They assessed classification performance using various ML classifiers, including DT, KNN, SVM, ensemble classifiers, Naive Bayes, linear discriminant analysis (LDA), and LR, and neural network (shallow and deep) models. They segmented ten seconds of non-overlapping windows to extract features from EEG data. In detail, they used discrete wavelet transform (DWT) to decompose EEG data into different frequency levels with diverse EEG frequency bands (i.e., α , β , δ , θ ) to extract features such as mean, standard deviation, median absolute deviation, variance, interquartile range, kurtosis, and skewness. They showed the effectiveness of DWT in analyzing EEG data by evaluating the performance differences of multiple predictive models using combined features from nineteen channels.
Mumtaz et al. [24] presented a method for diagnosing MDD by utilizing EEG power features with various machine learning algorithms, such as SVM, Naive Bayes (NB), and logistic regression (LR). They conducted data standardization through z-scores to identify critical features by eliminating potential outliers. The study revealed that SVM yielded high classification results in the analysis of MDD. Suuronen et al. [25] used data collected from multiple academic institutions (i.e., the University of Turku, the University of New Mexico, and the University of Iowa) to investigate Parkinson’s disease. Sample entropy was used to extract features from different EEG frequency bands, delta ([0.5–4] Hz), theta ([4–8] Hz), low alpha ([8–10] Hz), high alpha ([10–13] Hz), and beta ([13–30] Hz). They developed a greedy-based feature selection procedure to determine the optimal channels for disease analysis. The relevance of the channel was determined by measuring the area under the ROC curve. Consequently, the performance using one channel and five channels with the eyes open (EO) datasets was determined as 67 % and 76 % , respectively. Additionally, crucial channels were identified as frontal, left-temporal, and midline-occipital.
Shilaskar et al. [26] proposed an approach for dyslexia detection using principal component analysis (PCA). To address the data imbalance problem, they applied synthetic minority oversampling techniques (SMOTEs). Then, classification performance results to detect dyslexia were evaluated using various machine learning algorithms, including SVM, LR, RF, NB, decision trees, and KNN. The results indicated that the combination of SMOTE and RF accurately identified dyslexia. In a comprehensive review, Ahire et al. [27] explored diverse approaches utilizing machine learning algorithms for dyslexia identification based on EEG signals. Their findings suggested that SVM outperformed other machine learning algorithms in this context. Zolezzi et al. [28] concentrated on categorizing chronic pain into three groups (high, moderate, and low) by employing approximate entropy (ApEn) and absolute band power from five frequency bands: delta ([0.1–4] Hz), theta ([4–8] Hz), alpha ([8–12] Hz), beta ([12–30] Hz), and gamma ([30–100] Hz). By analyzing multiple datasets with chronic neuropathic pain, they found that combining three bands (theta, alpha, and beta) with the ApEn feature separated the pain group well with the approximate entropy. Aydin et al. [29] performed a classification of patients with obsessive–compulsive disorder (OCD) using SVM with EEG segments of 2 s. By employing three entropy measurements (approximate entropy, sample entropy, and permutation entropy) after segmenting EEG in 2 s, they determined that permutation entropy yielded the highest accuracy in classifying OCD.
Researchers have also performed various multi-class classification studies for analyzing neurological disorders with EEG data. Alturki et al. [30] conducted a study to analyze neurological disorders by extracting features from EEG sub-bands and applying multiple classification algorithms, such as LDA, SVM, KNN, and ANN. They used five statistical methods, including logarithmic band power (LBP), standard deviation, variance, kurtosis, and Shannon entropy (SE), from DWT with level four decomposition, ‘db4’ wavelet, and EEG segments of length 50 s, to extract the features. They performed an extensive analysis of evaluating classification accuracies in both two- and three-class scenarios. They also conducted a study with three types of EEG datasets, considering both single-channel and multi-channel modes of diagnosis. They identified that SVM produced high classification accuracy, especially when combined with the logarithmic band power feature. In a different approach, Tawhid et al. [31] proposed a method for analyzing EEG signals to classify various neurological disabilities, such as autism, epilepsy, Parkinson’s disease, and schizophrenia, utilizing Convolutional Neural Networks (CNNs). As part of the pre-processing steps, EEG signals were segmented into 3 s intervals and transformed into 2D time–frequency–spectrogram images using a short-time Fourier transform. The classification analysis, incorporating five-fold cross validation, aimed to distinguish neurological disorders from normal subjects. The results indicated that the proposed CNN model achieved a notable performance accuracy of 98.33% in classifying the disorders.
Although numerous studies have been performed to identify neurological disorders or mental disorders by analyzing EEG data, the majority have centered on understanding individual disorders by simply applying diverse classification algorithms. The critical examination of each channel associated with neurological disabilities is essential for discerning both commonalities and distinctions among them. Despite this importance, limited studies have been performed to analyze multiple neurological disorders or mental disorders by evaluating their shared characteristics and variations. This study addresses this gap by undertaking a comparative analysis of various features, including EEG frequency bands, power spectral density, raw data, and wavelet features. Our study gauges the efficacy of these features in analyzing neurological disorders, utilizing a publicly available dataset containing diverse neurophysiological information encompassing conditions, such as ADHD, MDD, OCD, and Parkinson’s disease. Multiple features were extracted from the dataset using various EEG segmentation techniques, including short-term EEG segmentation based on DWT. EG signals are irregular and non-stationary with changing magnitude values over time [31]. So, presenting information from a specific duration satisfying the stationarity assumption in EEG signals is important [32,33]. Thus, the choice of a segmentation length in analyzing EEG signals is always challenging. Using longer segmentation lengths may produce misleading results [32] due to increased non-stationarity. Also, utilizing longer segmentation lengths typically results in an elevated computational cost [34,35]. Numerous studies are conducted on a single neurological disorder with various segmentation lengths empirically as 1 s, 2 s, 3 s, and 5 s [16,31,35,36,37,38]. In this study, we examined two segmentation lengths (1 s and 2 s) to identify neurological disabilities by comparing their features, classification performance results, and visual representations. To the best of our knowledge, no previous studies have provided detailed comparisons illustrating feature extraction, classification performance results, and visual representation in different segmentation lengths with various neurological disorders. To advance our understanding of the distinctions among neurological disorders, our study advocates adopting a sophisticated analysis technique—visual analytics. In detail, multiple visualization methods were employed to generate graphical representations of data, facilitating interactive visual analysis for users. Additionally, an extensive region-based analysis was performed to address the limitation of comprehending neurological disorders solely through a broad perspective, initiating a detailed exploration of specific regions. A detailed explanation of the conducted procedures and steps is provided in the following sections.

3. Methods

As mentioned above, the primary objective of this study is to discern neurological disabilities by combining wavelet-based feature extraction and visual analysis. Since the identification of crucial features that encapsulate the distinctive characteristics of neurological disabilities is pivotal in differentiating them, our approach involves an examination of the most pertinent channels associated with each disability and an assessment of the performance variances between employing single-channel- and region-based predictive models. The incorporation of visual representations of features enhances the capacity to comprehend both the extracted features and EEG data, facilitating the differentiation of various disabilities. Our proposed approach contains four steps: pre-processing, extracting and selecting features, generating predictive models, and conducting visual analysis. Detailed elucidations for each of these steps are added in the following subsections.

3.1. Data Description

We used a publicly available TDBRAIN dataset [39]. The dataset contains resting state raw EEG signals with relevant clinical and demographic data collected from psychiatric patients between 2001 and 2021. The dataset contains 1274 EEG signals (with a sampling frequency of 500 Hz) for psychiatric patients (aged 5 to 89 years). It includes both formal diagnoses and referral indications (unofficial diagnoses). The raw EEG signals have 26-channel (10–10 Ag/AgCl electrodes) recordings with eyes open (EO) and eyes closed (EC) for 2 min each. The data contain both healthy and several disorders such as major depressive disorder (MDD), attention deficit hyperactivity disorder (ADHD), subjective memory complaints, obsessive–compulsive disorder, burnout, dyslexia, chronic pain, Parkinson’s, tinnitus, insomnia, and migraine as indications. In this study, we only analyzed the subjects that matched between each indication and formal diagnosis. Collected during eyes open (EO) and eyes closed (EC) tasks, the data in different task conditions are referred to as EO-data and EC-data throughout the paper for clarity. We analyzed 23 channels (C3, C4, CP3, CP4, CPz, F3, F4, F7, Cz, F8, P3, P4, FC3, FC4, FCz, Fp2, Fz, O1, O2, Oz, P8, T7, and T8). Given the prevalence of subjects categorized as ‘Unknown’ for formal diagnoses, we only considered the data associated with seven disabilities: ‘burnout’, ‘dyslexia’, ‘chronic pain’, ‘MDD’, ‘OCD’, ‘ADHD’, ‘Parkinson’s disease’, and ‘healthy’ condition (see Table 1).
Due to the existence of numerous disorder types in the dataset, we analyzed the data by categorizing them into two groups: ‘normal (healthy)’ and ‘abnormal (disabilities)’.

3.2. Feature Extraction and Selection

Data pre-processing plays a crucial role in eliminating noise and obtaining clean data for EEG analysis. In this study, we employed notch and band-pass filters to enhance the data quality. Since the EEG data contain a 50 Hz frequency associated with power line noise, the notch filter was applied for frequency removal. Then, the band-pass filter within the frequency range of 0.5 to 100 Hz was employed for further refinement.
Feature extraction is an essential step influencing the overall classification performance. To analyze the EEG data, a non-overlapping segmentation is utilized to split EEG signals into distinct segments. For the multi-channel EEG data of subject k { 1 , 2 , , N } , where N indicates the total number of the subjects, including healthy and unhealthy subjects, each channel is divided into segments with a pre-defined segmentation length. Specifically, D i = { X 1 , X 2 , , X N k } , where N k = l s represents the total number of segments for the subject k, and l denotes the total length of the EEG data. The i-th segment of a channel is represented as C i containing T instances, where 1 i N k and T = m l × f s , where m l is the pre-defined segmentation length and f s is a sampling rate. As the suitable segmentation length for EEG data analysis is unknown, we evaluated the EEG data with two segmentation lengths (i.e., m l = 1 second, 2 seconds). The class label of the i-th channel is determined by C i { 0 , 1 } , and 0 and 1 represent “normal” and “abnormal”, respectively.
We employed four different feature extraction approaches to extract features from each segment to compare and examine the capability of identifying the abnormal condition. The extracted features encompass raw features, EEG frequency band information, power spectral density (PSD), and wavelet features. The raw features represent the features extracted directly from each segment, calculated by determining the average of the z-score, mean, variance, kurtosis, and entropy. The EEG frequency band information is broadly used in analyzing EEG data. To extract EEG frequency band features, commonly used EEG frequency band information, such as α , θ , γ , β , and δ , was identified in each segment. Then, the average and standard deviation of the frequency band were computed. For the extraction of PSD features, Welch’s periodogram [40] was employed to compute the frequency spectrum of EEG signals using fast Fourier transform (FFT). The resulting PSD features comprised five attributes: frequency, energy, variance, sample entropy, and permutation entropy. While FFT is effective in identifying the underlying frequency information within the EEG data, it struggles to detect local transient changes at specific frequencies. Therefore, wavelet transform (WT) is considered due to its suitability for analyzing non-stationary data, offering the capability to extract both frequency (scales) and time information. Wavelets are characterized by small waves with limited duration and zero average values [41]. WT is good for analyzing data at a specific time and frequency or revealing information with different scales [42]. It extracts local information with different frequencies to reveal trends, discontinuities, and repeated patterns underlying the data. Through the analysis of low- and high-frequency information, data trends and deviations (representing low and high fluctuations) can be identified accordingly [43]. DWT decomposes the EEG data at a predefined level by splitting each level into two sub-bands containing different frequency ranges (i.e., low and high frequencies) using the formula
ϕ a , τ ( X t , c ) = 1 a ( X t τ a t , c )
where X t , c represents the EEG data at the segment t , 1 t T of the channel c, and ϕ a , τ . ( X t , c ) is a scaling function, called wavelet basis functions, generated by the mother wavelet ϕ ( X t , c ) . a and τ indicate the scale and translation factor, respectively, controlling the scaling and translation of the wavelet. In our study, DWT was employed to extract features by decomposing EEG data into high- and low-frequency information at each level. High-frequency information produces detail coefficients, presenting any rapid or sudden changes in the data, while low-frequency information produces approximate coefficients, showing slow changes in the data. The detail coefficients are particularly useful for detecting rapid changes, such as discontinuities or sudden shifts. Using the coefficients, the following wavelet features are computed at each level:
ω 1 j = 1 n d i , j n , ω 2 j = i = 1 n ( d i , j d i , j ¯ ) 2 n , ω 3 j = ( n μ × ( 2 l o g ( n ) ) )
ω 4 j = 1 n ( δ i , j ) 2 n , ω 5 = μ d i , 5 μ a , ω 6 = μ d i , 4 μ a
where d i , j = { d 1 , j , d 2 , j , , d i , j } represent the wavelet coefficients at the jth level ( j = 1 , 2 , , ( l + 1 ) ), l denotes a pre-defined decomposition level, i indicates the elements of the coefficients ( i = 1 , 2 , , n ), d i , j ¯ indicates the mean of d i , j , μ a represents the sum of the coefficients in level l, and n denotes the length of the coefficients. n μ , representing the noise variance, can be computed as n μ = ϵ d i , j T d , where ϵ d i , j indicates the median of the coefficients, and T d = 0.6745 presents the wideband neuronal signal for Gaussian noise [44]. We evaluated various mother wavelets and decomposition levels to examine their similarities and differences for analyzing neurological disabilities.
Once all features were extracted, we performed feature selection to remove any redundancies. In detail, we employed one-way analysis of variance (ANOVA) [45] to determine statistically significant features ( p < 0.05 ). Only the statistically significant features were utilized for classification and visual analysis.

3.3. Data Classifications

To assess the effectiveness of the selected features in identifying unhealthy conditions, we conducted a classification analysis using machine learning (ML) algorithms. Specifically, four classification algorithms—SVM, RF, KNN, and LR—were employed to measure the performance differences. The four features, i.e., wavelet, raw, PSD, and EEG frequency band, were tested to explore their distinctiveness in classifying “normal” and “abnormal”. In detail, we conducted the classification analysis under two conditions: single channel based and region based. The single-channel-based analysis involved the independent analysis of data from each channel. The region-based analysis focused on the examination of five brain regions, i.e., frontal, temporal, central, parietal, and centroparietal, whose channel information is detailed in Table 2. All features in each brain region were combined to investigate the capability of identifying unhealthy subjects. Various performance metrics, including precision, recall, F1 score, and the area under the receiver operating characteristic curve (AUC), were compared using k-fold cross validation to evaluate the classification performance and identify unhealthy subjects. With k-fold cross validation, the feature set in channel c, denoted as F c , was divided into n subsets, represented as f 1 , f 2 f n , f i f j . In detail, n 1 subsets were used for training, and the remaining one was employed for testing the trained model. This process was repeated n times, and the average performance scores were determined.

3.4. Visual Analysis

Conducting data classifications with different machine learning algorithms is valuable for assessing their effectiveness in distinguishing between “normal” and “abnormal”. However, the clarity in differentiating between them is hindered by the imbalanced data. Imbalanced data may cause challenges in comprehending the performance of a classification model due to its inherent unreliability. To address this issue, we considered leveraging visualization techniques to represent the extracted features by mapping them into visual glyphs. Visual representations can help in understanding the patterns and structures of the data by revealing underlying patterns. Since the data utilized in this study consist of numerous variables, dimension reduction is applied to show the data in a lower-dimensional space (i.e., 2D space). In the visualization domain, various dimension reduction techniques have been employed for presenting high-dimensional data, including PCA, LDA, MDS (multi-dimensional scaling), t-SNE (t-distributed stochastic neighbor embedding), UMAP (Uniform Manifold Approximation and Projection), and others [46]. Among them, t-SNE and UMAP have been broadly used recently by researchers in representing data because they show a better separation of the classes than others. However, these methods come with computational costs to approximate the optimal positions. Therefore, we considered using PCA to determine the points representing the high-dimensional data. It computes eigenvectors and eigenvalues to identify multiple principal components. By utilizing the 1st and 2nd principal components, the data can be effectively represented in a 2D space [47].
We also employed parallel coordinates visualization [48] to represent all data instances. It generates polylines through vertically arranged variables (i.e., parallel axes), representing individual instances. Despite its effectiveness in presenting multivariate data, researchers have proposed numerous variations of parallel coordinates visualization [49,50]. However, parallel coordinates visualization has a major limitation when handling large data instances, occurring a visual cluttering problem where multiple polylines overlap. To address this problem, researchers considered reducing the number of polylines or rearranging (or reordering) the axes of the parallel coordinates. Since we utilized the parallel coordinates visualization to understand the patterns between classes, we did not consider addressing the visual cluttering problem in this study. Instead, we considered showing data distribution in each axis because it helps users understand clusters in each group (e.g., normal vs. abnormal). To represent data distributions in each variable, angular histograms [51] were generated to illustrate the density and slopes of underlying polylines overlaid onto the parallel coordinates. Angular histogram is a technique that presents the frequency distribution of underlying data within each parallel axis by measuring the frequency of the data and the directional information of polylines in each axis. Figure 1 shows an example of applying angular histograms to parallel coordinates. The histogram highlights density using different colors (red for high density and green for low density) based on the measured distributions. To measure the distributions, various statistical approaches have been proposed to determine an appropriate bin size for histograms [52]. However, traditional histograms only present univariate data. In detail, a single histogram can only represent the frequency of the data plots within the selected vertical axis. Therefore, no unique statistical approach can be applied to determine an appropriate bin size for handling all variables in the parallel coordinates visualization. Instead, a user-driven approach is commonly utilized to determine the bin size of histograms in parallel coordinates [51,53,54]. To generate density distributions, the angular histogram utilizes a user-defined binning approach to determine the denseness of each distribution. Each bin is defined as a direction vector representing the slope ( α ) of each polyline toward another parallel axis. The angular frequency distance ( F ( α ) ) is computed by measuring the frequency in each bin and evaluating the distance to the next parallel axis.

4. Results

As mentioned above, we applied four different feature extraction approaches with two EEG segmentations (1 s and 2 s) to examine the effect of the segmentation length in distinguishing abnormal conditions. For applying DWT, we employed a pre-defined wavelet decomposition level of six ( l = 6 ) to extract wavelet features. Then, statistically significant features were determined by applying ANOVA with the p-value (<0.05). Varying numbers of significant features were determined in each channel. By analyzing the EEG frequency band features, we found three to nine and four to nine significant features from the EC- and EO-data, respectively. In detail, we found nine significant features with the 1 s EEG segmentation in the O2 channel. But, with the 2 s EEG segmentation, only three significant features were determined. For the FC3 channel, we found three and nine significant features with the 1 s and 2 s EEG segmentations, respectively. With the 2 s EEG segmentation, nine significant features were identified in the channels (i.e., CP4, Cz, F3, Fc3, Fz, O1, Oz, P3, P4, CP4, and T7), seven features in the channels (i.e., C3, C4, CP3, CPz, F4, F7, and Fp1), six features in the channels (i.e., FCz, and T8), and four features in the FC4. We also found the same number of features from the channels with the 1 s and 2 s segmentations: nine features—CP4, F3, and P7; eight features—F8; and six features—T8. We determined nine features from the channels (Cpz, Cz, and Fp2) in both the 1 s and 2 s segmentations. Among the various features, we found that the measured standard deviation was commonly identified as a significant feature. The measured average value was also identified as a significant feature in some channels. These features might be critical for analyzing the conditions precisely because they hold unique characteristics for differentiating them. The extracted features from β and γ bands in all channels of the EC- and EO-data were also recognized as significant features.
By evaluating the PSD features from the EO-data, we identified that all features extracted from 60.6 % of the channels were determined as significant features. But for the features from the EC-data, only 27.3 % and 51.1 % were determined to produce all significant features with the 1 s segmentation and the 2 s segmentation accordingly. We also found that sample entropy (SmpE) and permutation entropy (PE) were identified as significant features from all channels in the EC-data. Three raw features (kurtosis, variance, and entropy) were selected as significant features from 90 % of the channels with the 1 s segmentation and 75.7 % with the 2 s segmentation. The two features (kurtosis and variance) were determined as significant features in most EC- and EO-data channels. But the feature (z-score) was not identified as a significant feature in any of the channels. The feature (kurtosis) was determined as a significant feature in the two channels (C3 and Oz) with the 1 s segmentation and in the four channels (C3, F3, FCz3, and Oz) with the 2 s segmentation with the EO-data. For the EC-data, only the P7 channel presented significant features with the 1 s segmentation. When evaluating the wavelet features, we determined the features from the coefficient levels (3, 4, and 5) corresponding to 4 Hz∼32 Hz to be significant features in the 1 s and 2 s segmentations. In detail, the features ( ω 1 3 , ω 2 3 , ω 5 3 , ω 6 3 ) corresponding to 16 Hz∼32 Hz were selected as significant features in level three detail coefficients. Only the features ( ω 2 1 , ω 3 1 , ω 4 1 ) were identified as significant features in level one detail coefficients. The feature ( ω 2 1 ) was also identified as a significant feature in the EC- and EO-data with the 2 s segmentation.

4.1. Single-Channel Analysis

Figure 2 shows examples of the four features extracted from the channels (O2, C4, Fp1, and T7) using the EC-data with the ‘db4’ wavelet and level six decomposition. The raw features exhibited subtle distinctions with minimal overlaps among the various disabilities. Specifically, the T7 channel presented slight differences in patterns between OCD, dyslexia, and Parkinson’s (see Figure 2M). In both the O2 and T7 channels, OCD was distinguishable, presenting an isolated pattern (high values) compared to others (see Figure 2A,M).
Using the PSD features, we found distinct patterns between the two disabilities (dyslexia and ADHD) and the healthy condition, particularly in the Fp1 and O2 channels. Notably, the C4 and T7 channels exhibited nearly identical patterns between dyslexia and ADHD. When analyzing EEG frequency band features, a clear separation among dyslexia, Parkinson’s disease, and MDD was determined. When employing wavelet features, the Fp1 channel emerged as effective in differentiating between dyslexia, ADHD, Parkinson’s disease, chronic pain, and the healthy condition. For the O2 channel, we observed differences, especially when comparing features of OCD, ADHD, and Parkinson’s disease with others. Results for the disabilities (burnout and MDD) were almost identical across most channels. Additionally, chronic pain and Parkinson’s disease showed almost similar patterns in the C4 channel. We also found a high similarity between MDD and Parkinson’s disease in the Fp1 channel. But, we noticed completely different patterns between them in some of the channels (O2, T7, and C4). MDD and burnout displayed high similarity across most channels, except for the T7 channel. Among the four features, the wavelet features were determined as superior in distinguishing the disabilities from healthy. Specifically, we identified that the wavelet features in the Fp1 channel could be a good indicator for differentiating Parkinson’s disease, OCD, ADHD, dyslexia, and healthy because of the clear separation between them. With the 2 s segmentation, we observed the wavelet features in the C4 channel showed three distinctively separable groups: dyslexia and ADHD, chronic pain, and the rest of the disabilities.
Utilizing the level six decomposition ( l = 6 ), we observed nearly identical patterns between ADHD and dyslexia in the O2 channel. However, a clear differentiation among the disabilities within the same channel became evident when employing the level seven decomposition ( l = 7 ). Furthermore, with the level seven decomposition, we identified a distinct separation between MDD and the healthy condition, which was not apparent with the level six decomposition. Distinctions between Parkinson’s disease and burnout were highlighted through features extracted from level four and six detail coefficients and approximate coefficients in the T7 channel. We observed a clear separation among burnout, healthy, and other disabilities with the level seven decomposition, particularly when utilizing the features from the approximate coefficient corresponding to 1∼4 Hz. We also found that the features with detail coefficient levels five, six, and seven presented distinctive separations across the disabilities: dyslexia and ADHD, and other disabilities. In the Fp1 channel, chronic pain showed a clearly distinguishable pattern from other disabilities, similar to the results obtained using the level six decomposition. The wavelet features derived from the level seven decomposition revealed two to three divisible groups among the disabilities, whereas the level six decomposition showed individual differences.
Figure 3 illustrates the comprehensive results of single-channel classification for the EC-data with both 1 s and 2 s segmentations. We found that the wavelet features showed better classification performance results than those using other features. Among the classification algorithms, RF showed better classification performance than the others. While the 2 s segmentation showed slightly better classification results than the 1 s segmentation, variations in classification performance were noted across the channels depending on the segmentation (either 1 s or 2 s segmentation). Specifically, the channels F3, F4, F8, and CP4 showed enhanced performance results with the 1 s segmentation. From the evaluation of the classification performance results, we found an improvement in AUC with increased segmentation length. For the single-channel classification evaluation, we used the wavelet (‘db4’). Alternatively, different wavelets can be utilized to extract wavelet features. Based on our evaluation with another wavelet (‘coif’), we found slightly different classification performance results. For instance, the F1 classification scores for analyzing the EC-data were determined as RF ( 0.85 ± 0.0002 ), SVM ( 0.79 ± 0.0001 ), KNN ( 0.83 ± 0.0003 ), and LR ( 0.80 ± 0.0001 ). When analyzing the EO-data, the F1 scores were identified as RF ( 0.85 ± 0.0002 ), SVM ( 0.80 ± 0.0002 ) , KNN ( 0.83 ± 0.0002 ) , and LR ( 0.80 ± 0.0002 ) ). We found the AUC scores to be RF ( 0.60 ± 0.0005 ), SVM ( 0.51 ± 00.0003 ), KNN ( 0.59 ± 0.0004 ), and LR ( 0.51 ± 0.0003 ). Although slightly different classification results can be observed depending on the applied wavelet, we determined that the wavelet features produced better classification performance results than other features.
By evaluating the classification performance results for the features with the 2 s segmentation using the level six and seven decompositions ( l = 6 and 7), we found higher F1 scores with l = 6 . But similar AUC scores were observed between them. When using the wavelet features extracted from the EC-data with ‘db4’, we found the overall average F1 scores for all channels to be RF ( 0.880 ± 0.0002 ) , SVM ( 0.860 ± 0.0001 ) , KNN ( 0.856 ± 0.0003 ) , and LR ( 0.860 ± 0.0001 ) . For evaluating the wavelet features from the EO-data, the F1 scores were determined as RF ( 0.851 ± 0.0014 ) , SVM ( 0.800 ± 0.0011 ) , KNN ( 0.831 ± 0.0012 ) , and LR ( 0.801 ± 0.0008 ) . The overall average AUC of all channels from the EC-data was measured as RF ( 0.601 ± 0.0001 ) , SVM ( 0.510 ± 0.0002 ) , KNN ( 0.574 ± 0.005 ) , and LR ( 0.513 ± 0.0012 ) . For the EO-data, the performance results were RF ( 0.610 ± 0.0001 ) , SVM ( 0.510 ± 0.0001 ) , KNN ( 0.582 ± 0.0003 ) , and LR ( 0.510 ± 0.0002 ) . Overall, we found slightly better performance reults with l = 6 when comparing the F1 and AUC scores with 2 s segmentation. We also found slightly better performance in multiple channels (Fp1, Fp2, F3, F4, and FcZ) with l = 7 .

4.2. Region-Based Analysis

We also conducted a region-based channel analysis on the five regions (frontal, temporal, central, parietal, and occipital). Figure 4 shows the F1 scores for the applied classification algorithms. When evaluating the performance results between 1 s and 2 s segmentations, the 2 s segmentation showed slightly better results. We also found that the wavelet features performed better with the region-based classification. Because of numerous performance results from our study, three region-based classification results (accuracy, precision, and AUC) are listed in Appendix A. Notably, we observed high accuracy and precision scores for the regions utilizing the wavelet features from the region-based classification analysis on the EC- and EO-data. We also found that the wavelet features showed better AUC scores. Among the classification algorithms, both KNN and RF showed better classification performance results. More importantly, we determined that RF demonstrated high classification performance results with the wavelet features (see the red glyphs in Figure 4B). High F1 scores were identified in the frontal region with ML techniques. The wavelet features also showed high F1 scores with SVM, RF, and LR in all regions except KNN. When using KNN, minor differences were observed between the wavelet and EEG frequency band features in the frontal brain region (see Figure 4C). The EEG frequency band features showed comparable performance with the wavelet features in the frontal region. Except for the frontal region, all other regions performed better with the wavelet features.
Additionally, we performed a comparison of the classification results employing different wavelets (w) and decomposition levels (l). Table 3 presents the F1 scores for the region-based classification with the 2 s segmentation using two wavelets ( w = ‘coif3’ and ‘db4’) and two decomposition levels ( l = 6 and 7). We observed similar F1 scores between the two wavelets, with minor differences depending on the decomposition levels and the specific classification algorithms. Overall, the frontal region showed higher performance results across the regions. RF and KNN showed higher performance results with the EC- and EO-data. Specifically, the occipital and frontal regions showed enhanced performance results with l = 6 and w = ‘coif3’, while the central region exhibited better performance with l = 7 and w = ‘coif3’. However, considering the frequency of achieving higher performance by each classification algorithm, ‘db4’ was selected for l = 6 ( 55 % ) with the EO-data. With l = 7 , the ‘coif’ was determined as the best wavelet when analyzing both the EC- and EO-data.
By measuring AUC scores for the frontal region in the EC-data with l = 6 and w = ‘db4’, we determined the performance scores as SVM ( 0.903 ) , RF ( 0.921 ), KNN ( 0.968 ), and LR ( 0.627 ). For analyzing the EO-data, we found the scores to be SVM ( 0.975 ) , RF ( 0.976 ), KNN ( 0.986 ), and LR ( 0.682 ). When examining the same region with l = 6 and w = ‘coif3’, the AUC scores for the EC-data were identified as SVM ( 0.928 ), RF ( 0.913 ), KNN ( 0.977 ), and LR ( 0.664 ) and for the EO-data as SVM ( 0.916 ), RF ( 0.925 ), KNN ( 0.965 ), and LR ( 0.594 ). With l = 7 and w = ‘db4’, the AUC scores for the EC-data were found to be SVM ( 0.882 ) , RF ( 0.929 ), KNN ( 0.975 ), and LR ( 0.627 ), and those for the EO-data were SVM ( 0.885 ) , RF ( 0.924 ), KNN ( 0.978 ), and LR ( 0.615 ). With l = 7 and w = ‘coif3’, we found the scores for the EO-data to be SVM ( 0.885 ) , RF ( 0.924 ), KNN ( 0.980 ), and LR ( 0.673 ), and for the EC-data, SVM ( 0.903 ) , RF ( 0.953 ), KNN ( 0.980 ), and LR ( 0.629 ). Overall, KNN showed better results when comparing the AUC scores. By evaluating the average classification scores for all classification algorithms, we observed that the AUC scores with l = 6 and w = ‘db4’ were higher than with l = 6 and w = ‘coif3’ except for the central region in the EC-data. With the EO data, l = 6 and w = ‘db4’ was higher in the frontal, occipital, and parietal regions. With the level seven decomposition ( l = 7 ), the ‘db4’ performed better for the regions except for the frontal region with the EC data. With the EO data, the ‘db4’ was superior for all regions except the frontal and occipital regions.

4.3. Visual Analysis

To understand the patterns and structures of the different features, PCA-based visualization is performed. Figure 5 shows examples of the PCA projections of different features. To generate the PCA projections, the first- and second-principal components are determined and used to show all instances in 2D scatterplots, mapped to the x- and y-axis, respectively. Although different visualizations were generated depending on the features, we could identify similarities among PCA projections. For instance, the visual representations with the EEG frequency band and wavelet features from the occipital region showed similar widely spread patterns (see Figure 5A,D). Triangular shapes were also observed when analyzing the EEG frequency band and wavelet features from the O2 channel (see Figure 5E,H). For the PSD features, we could not find any similarities compared to the visualizations with other features. But, we noticed completely different patterns when analyzing the visualizations with the raw features (see Figure 5C,G). This might happen because of the small number of attributes and their high similarities. The visualizations with the features from the O2 channel showed various possible outliers (see the arrow). The raw features showed a dense region in the bottom left corner. Two mostly extreme outliers, representing ADHD and Parkinson’s instances, appeared in each corner of the PCA space. The PCA-based visualization is good for understanding the overall pattern of the features because PCA is useful for identifying major variances by determining the principal components from the data. However, we found that it was difficult to determine the unique difference between normal and abnormal groups with PCA. This might be because no major difference between them can be determined due to their high similarity. To address this limitation, we performed an additional analysis with parallel coordinates visualization.
To help users perform a comparative analysis, group-based visualizations are generated to represent the normal and abnormal groups in each separated parallel coordinates visualizations. Since we found that it was not easy to identify major differences between the groups within the PCA projections, multiple parallel coordinates visualizations were generated to determine the differences. In the parallel coordinates, the variables are arranged vertically to represent parallel axes. Angular histograms are overlaid on parallel axes to show the distribution of each variable. Figure 6 shows multiple parallel coordinates visualizations with different features. The difference was not apparent in the parallel coodinates visualizations using the raw features because they may have similar characteristics. But, with the angular histograms, minor differences were observed. Specifically, the angular histograms of the variables O201 and O202 showed the high-density areas at the lower region in the normal group (see the arrows in Figure 6C). The visualizations with the PSD and raw features showed similar results between the groups because there was no significant difference in the distributions of the features (see the zoomed representations of the angular histograms in the figure). The visualizations with the EEG frequency band and wavelet showed distinctive representations because of many statistically significant features. In detail, they showed various distinctive data distributions of variables (see the angular histograms highlighted by arrows).

5. Discussion

By evaluating the classification performance results, we found that the segmentation length (2 s) is suitable for analyzing the disabilities. We also identified that RF showed better performance results than the other algorithms. When evaluating the difference using the decomposition levels ( l = 6 or 7), we found a clear distinction between chronic pain and healthy with l = 6 . With the level l = 6 , we also observed slightly better performance results in most channels except the four channels (F3, Fp1, F4, Fcz, and P4). The four channels showed opposite results with the level l = 7 . These results indicated that a higher decomposition level should be applied for analyzing certain disabilities. However, this statement requires an extensive analysis of identifying the relationship between decomposition levels and EEG channels in producing classification performance results for analyzing disabilities. Since this analysis is not a primary consideration of this study, we leave this as our future work. Comparing the two wavelets (‘db4’ and ‘coif3’), we found similar classification performance results, even if different decomposition levels were applied. For the single-channel-based classification, we found a higher performance result with RF ( 1 % difference) when using either wavelet. However, slightly better performance results with the 2 s segmentation and ‘coif3’ were determined when analyzing the EO-data with l = 6 . Furthermore, for the region-based classification, ‘coif3’ performed better than ‘db4.’ Since DWT is suitable for detecting local events with different frequency scales, combining multiple wavelet features from different channels could synergistically improve the overall performance of identifying disabilities.
As discussed above, the most influencing features were determined with the ANOVA test to distinguish the disabilities in the channels. The cardinality of determining the subset of the features as significant features varied depending on the channel, segmentation length, and feature approach. For example, four EEG frequency band features were selected in the O2 channel with the 2 s segmentation, producing the recall performance as SVM (0.787), RF (0.807), KNN (0.780), and LR (0.784) for analyzing the EO-data. For the EC-data, nine features were selected, producing the performance results of SVM (0.859), RF (0.857), KNN (0.843), and LR (0.858). When using the wavelet features, the same cardinality of features was selected in the O2 channel, even when using different segmentation lengths. Instead, the C4 channel showed the different cardinality as important features between the two segmentation lengths. The recall scores with the 2 s segmentation were determined as the EC-data of SVM (0.883), RF (0.898), KNN (0.871), and LR (0.879), and the EO-data of SVM (0.852), RF (0.868), KNN (0.844), and LR (0.852). The same metric score performance with the 1 s segmentation was SVM (0.813), RF (0.859), KNN (0.844), and LR (0.814) with the EC-data and SVM (0.714), RF (0.806), KNN (0.782), and LR (0.725) with the EC-data. Overall, the EO-data with the 2 s segmentation provide better performance results. We found that the higher the cardinality, the better the performance presented.
For understanding the neurological disabilities depending on their age groups, multiple parallel coordinates visualizations were generated as shown in Figure 7. With the traditional parallel coordinates visualization (i.e., representing each instance as polyline), we could not determine the difference between the age groups when analyzing the data. However, we identified the difference between the groups by evaluating the angular histograms (bin size = 25) that present the distribution of each variable. For instance, the variable (‘Cz07’) represents fat-tailed distributions (i.e., skewed distributions). But distinctive distributions were observed depending on the age groups. In detail, the high peak appeared near the centerline of the angular histograms in the age 20s. But, the high peak existed at the bottom edge of the angular histograms in the age 70s. The differences, depending on the age groups, were more clearly visible in the variable ‘C417’. Specifically, the distribution in the age 40s showed skew normal distribution, having normal distribution but leaning toward one side of the distribution. Although identifying the primary cause of such distinctive distributions depending on the age groups cannot be determined from the visualization, it highlights the need to conduct extensive analysis to understand the effectiveness of each attribute in differentiating the distributions in the age groups. We performed Tukey’s Honest Significant Difference (HSD) test to examine the differences between the age groups. We found most variables showed significant differences ( p < 0.05 ) between the age groups. For instance, the variable (‘Cz07’) in the Cz channel showed significant differences ( p < 0.05 ) except for the age groups between 0 and 50 ( p = 0.9 ), 0 and 70 ( p = 0.0584 ), 20 and 70 ( p = 0.894 ), and 30 and 60 ( p = 0.925 ). The variable (‘C417’) in the C4 channel presented significant differences ( p < 0.05 ) except for the age groups between 20 and 30 ( p = 0.8348 ), and 40 and 50 ( p = 0.9 ).
We conducted an additional analysis to understand the difference between the disorders by generating multiple parallel coordinates visualizations depending on the conditions (seven neurological disorders and one healthy condition). Figure 8 shows eight parallel coordinates visualizations with angular histograms (bin size = 25) arranged horizontally. From the figure, we determined distinctive patterns depending on the conditions. For instance, the healthy condition showed similar frequency distributions compared to the disorders (i.e., Parkinson’s disease and MDD) in the variables (‘C300’∼‘Cz06’—from left to right in the visualization). But the rest of the other variables represented distinctive distributions between them. For example, the variables (‘Cz10’ and ‘Cz11’) in the healthy condition showed highly skewed distributions (i.e., asymmetric distributions) (see Figure 8A). Although the disorders MDD and Parkinson’s disease also represented asymmetric distributions, their distributions were different (see Figure 8B and Figure 8C, respectively). More interestingly, Parkinson’s disease shows peaks at the bottom edge of the distributions (see Figure 8C). These findings can be used to extract more critical features from data to differentiate the disabilities. The disorders chronic pain and dyslexia showed similar frequency distributions in the variables (‘C300’∼‘Cz06’—from left to right in the visualization). The disorders burnout and OCD represented sparse frequency distributions (small-sized histograms appeared on one side of the distribution) compared to other conditions. This may have occurred because fewer instances exist in the dataset.
The limitations of this study are as follows. First, this study used a relatively imbalanced sample size of individual neurological disabilities. Second, we did not perform extensive analyses of identifying individual neurological disability comparisons with healthy subjects. At last, we only provided comparisons of two decomposition levels (i.e., six and seven) with short-length segmentation (1 s and 2 s). To have a full understanding of neurological disabilities, extensive analysis with different datasets needs to be performed. Various segmentation sizes should be tested to determine the effectiveness of different segmentations for analyzing neurological disabilities.

6. Conclusions and Future Works

This study presents a novel approach to analyzing EEG data with neurological disabilities, incorporating DWT, ML classification algorithms, and visualization techniques. Despite various existing methods utilizing feature extraction and machine learning, previous studies have faced challenges in accurately identifying disabilities and enhancing understanding of the features and data. To address these limitations, we introduced a feature extraction approach using DWT to differentiate disabilities with short-length segmentations (e.g., 1 s and 2 s). We evaluated the effectiveness of this approach by employing multiple feature extraction methods (extracting the EEG frequency band, power spectral density, and raw features) and comparing their classification performance with wavelet features. Specifically, we conducted single-channel-based and region-based classifications. Then, further extensive analysis was performed through visual analysis using multiple visualization techniques. The comprehensive analysis revealed that wavelet features with the region-based classification exhibited strong performance results in differentiating disabilities, with the frontal brain region proving particularly effective. Among the classification algorithms, RF and KNN presented good performance results in the region-based classification, with RF demonstrating higher performance than other algorithms such as SVM, KNN, and LR. While evaluating different decomposition levels and wavelets for analyzing disabilities, no explicit patterns were observed. However, we noted that the ‘db4’ wavelet with 2 s segmentation and level six decomposition showed high classification performance results in the single-channel-based classification. Additionally, we found that the ‘coif’ wavelet yielded high performance with level seven decomposition, and the ‘db4’ with level six decomposition demonstrated high performance in the region-based classification.
For future works, we will expand our study through extensive analyses, exploring individual disabilities compared to healthy conditions across channels and region-based classifications while considering age and gender differences. Additionally, we will conduct multiclass classifications of neurological disabilities across individual channels. The integration of deep learning techniques will also be explored as a means to distinguish between disabilities using the proposed features. Furthermore, we plan to test longer EEG segmentation lengths, such as 5 s, 10 s, 20 s, 30 s, and 60 s, with various wavelets to determine their effectiveness for analyzing neurological disabilities. This will enable us to examine the relationship between wavelets and decomposition levels, uncovering the distinctive characteristics associated with different disabilities.

Author Contributions

Conceptualization, S.Y.J., S.J., A.M.P., K.K. and D.H.J.; methodology, S.Y.J., S.J., A.M.P., K.K. and D.H.J.; software, S.Y.J. and D.H.J.; validation, S.Y.J., D.H.J. and S.J.; formal analysis, S.Y.J., S.J., A.M.P., K.K. and D.H.J.; writing—original draft preparation S.Y.J. and D.H.J.; writing—review and editing, S.Y.J., S.J., A.M.P., K.K. and D.H.J.; visualization, D.H.J. and S.Y.J.; funding acquisition, S.Y.J., S.J., A.M.P. and K.K. All authors have read and agreed to the published version of the manuscript.

Funding

This material is based upon work supported by the National Science Foundation under Grant No. (2219634).

Institutional Review Board Statement

Not applicable.

Informed Consent Statement

Not applicable.

Data Availability Statement

Publicly available datasets were analyzed in this study. This data can be found here: https://brainclinics.com/resources/ (accessed on 14 November 2023).

Conflicts of Interest

The authors declare no conflicts of interest. The funders had no role in the design of the study; in the collection, analyses, or interpretation of data; in the writing of the manuscript; or in the decision to publish the results.

Appendix A. Region-Based Classification Performance Comparisons

We performed an extensive classification analysis to determine the effectiveness of using the wavelet features compared to other feature extraction approaches. Since the performed classification results include too many values, detailed information about the results is listed here.
Table A1 presents the ML performance comparisons of the regions in 1 s and 2 s segmentations. Among the features, wavelet features performed better in the regions for the metrics AUC, accuracy, and precision with the 1 s segmentation with the EC-data. With the 2 s segmentation, we found that the wavelet features performed higher in the regions. All regions performed better with the wavelet features for accuracy except for the temporal region. It also showed slightly better performance compared to the EEG frequency band features with SVM, KNN, and LR.
Table A2 presents the ML performance comparisons of the regions in 1 s and 2 s segmentation with the EO-data. With EC- and EO-data, we found that wavelet features performed better in all the regions for the metrics (AUC, accuracy, and precision) in both 1 s and 2 s segmentation. Please note that the results of the F1 scores are included in the main paper.
Table A1. Region-based classification performance comparisons for the EC-data. S represents the segmentation length, F indicates the features, and R presents the brain regions. The letters in column F denote E—EEG frequency band; P—PSD; R—raw; and W—wavelet. The letters in column R indicate C—central; F—frontal; P—parietal; O—occipital; and T—temporal.
Table A1. Region-based classification performance comparisons for the EC-data. S represents the segmentation length, F indicates the features, and R presents the brain regions. The letters in column F denote E—EEG frequency band; P—PSD; R—raw; and W—wavelet. The letters in column R indicate C—central; F—frontal; P—parietal; O—occipital; and T—temporal.
Metric ACCPrecisionAUC
S F R SVMRFKNNLRSVMRFKNNLRSVMRFKNNLR
1 sEC0.8970.8960.9140.8520.8990.9030.9090.8030.6610.6530.7730.501
F0.9730.9250.9880.8520.9740.9310.9880.8040.9170.7470.9730.514
O0.8930.8880.9150.8520.8960.90.9090.7890.6510.6220.7650.503
P0.8940.8960.9170.8520.8970.9060.9120.7260.6520.6510.7730.5
T0.8640.8770.8760.8520.8750.8830.860.7420.5410.590.6630.5
PC0.870.8890.890.8520.8780.8910.8790.7360.5640.6350.7080.5
F0.9350.9240.9630.8510.9360.9290.9630.7360.7950.7450.9060.5
O0.8610.8830.8840.8510.8680.8850.8710.7310.5320.6180.6940.5
P0.8570.8910.8910.8520.870.890.880.7260.5170.6480.710.5
T0.8520.8740.8610.8520.7660.8640.8380.7260.50.6010.6230.5
RC0.8560.8690.8480.8520.8590.8550.8120.7810.5150.5840.5760.501
F0.8720.9140.90.8540.8810.9180.8920.8270.5720.7190.740.513
O0.8560.870.8530.8550.8490.8610.820.8540.5160.5790.5840.511
P0.8570.8710.8540.8520.8720.8590.8220.7260.5160.5850.5870.5
T0.8630.8640.8460.8490.8540.8450.8040.7570.5460.5620.5580.501
WC0.9130.9490.940.860.9130.950.9380.8320.7250.840.8450.564
F0.9840.9850.9910.8770.9840.9850.9910.8610.9520.9490.9820.646
O0.9150.9470.9420.8540.9150.9480.940.8150.7310.8260.8490.541
P0.9250.9540.9450.8530.9250.9550.9430.8120.760.8530.8630.547
T0.8730.9160.8920.850.8750.9150.8820.8050.5780.7380.7220.542
2 sEC0.8970.8960.9140.8520.9130.9160.9360.7380.6710.6710.830.5
F0.9730.9250.9880.8520.9750.9450.9880.8180.9230.7950.9720.514
O0.8930.8880.9150.8520.890.8990.920.7920.5790.6060.7830.503
P0.8940.8960.9170.8520.9030.9120.9430.7910.6510.660.8480.503
T0.9060.9070.9390.8590.8820.8930.8810.810.570.6110.6960.503
PC0.8930.9110.9310.8590.9010.9150.9270.8290.6240.6910.8080.502
F0.9430.9390.9750.860.9440.9420.9750.8490.8050.7860.9370.504
O0.8650.8950.9050.8590.8720.8980.8970.8010.5240.6360.7440.501
P0.8740.9130.9290.8580.8730.9140.9250.7880.5620.7040.8160.501
T0.8630.8890.8860.8580.8650.8820.8730.7610.5180.6350.6890.5
RC0.8690.8740.8530.8620.8780.8690.8150.8320.5380.5650.5650.52
F0.880.9140.9090.860.8830.9170.9020.8310.580.7020.7570.504
O0.8590.8630.8450.8590.7380.8360.7870.7660.50.5270.5270.5
P0.8610.8660.8450.8590.8710.8380.7970.7380.5060.550.5430.5
T0.8640.8590.8490.8590.8680.8140.7960.7470.5220.530.5340.5
WC0.9020.9430.9450.8640.910.9460.9430.8360.6570.8020.8480.538
F0.970.9770.9870.8820.9710.9770.9870.8670.9040.9210.9680.63
O0.8930.9350.9420.8640.90.9380.9390.8350.6260.7720.8420.535
P0.8870.9420.9450.8650.890.9440.9430.840.6070.7970.8480.542
T0.8680.9130.8940.8630.8730.9140.8830.8320.5360.7060.6980.536
Table A2. Classification performance comparisons with region-based for the EO-data. S represents segmentation length, F indicates features, and R presents brain regions. The letters in column F denote E—EEG frequency band; P—PSD; R—raw; and W—wavelet. The letters in column R indicate C—central; F—frontal; P—parietal; O—occipital; and T:—temporal.
Table A2. Classification performance comparisons with region-based for the EO-data. S represents segmentation length, F indicates features, and R presents brain regions. The letters in column F denote E—EEG frequency band; P—PSD; R—raw; and W—wavelet. The letters in column R indicate C—central; F—frontal; P—parietal; O—occipital; and T:—temporal.
MetricACCPrecisionAUC
S F R SVMRFKNNLRSVMRFKNNLRSVMRFKNNLR
1 sEC0.8880.8890.9090.8530.8840.8950.9030.8060.6450.6310.7670.51
F0.9850.970.9930.8550.9850.9710.9930.8190.9550.90.9810.54
O0.9060.9190.9280.8520.9070.9250.9250.7950.6980.7310.8080.501
P0.9220.9240.940.8530.9210.9280.9380.8080.7560.7480.8380.507
T0.8680.8830.8780.8530.870.8830.8620.8060.5580.6150.660.512
PC0.8550.890.8910.8520.8460.8860.8820.7610.5110.6510.730.5
F0.9620.9650.9760.8560.9620.9660.9760.8220.8820.8860.940.551
O0.8680.9150.9070.850.8680.9150.9010.780.5610.7280.760.504
P0.8830.9170.9140.8520.8820.9150.9090.8280.6210.7410.7780.502
T0.8680.8830.8780.8530.870.8830.8620.8060.510.6160.6390.503
RC0.8550.8650.8480.8520.8480.8430.8170.7790.5120.5790.5890.502
F0.910.9550.9460.8610.9140.9550.9440.8590.7060.8570.8680.535
O0.8580.890.8670.8520.8540.8810.8460.7560.5250.6640.6390.5
P0.8750.8950.8790.850.8680.8860.8650.7510.5980.6910.6890.501
T0.8520.8510.8350.8520.7260.7970.7740.7850.50.5150.5220.5
WC0.8940.9440.9330.8570.8890.9450.930.8240.670.8240.8360.552
F0.9890.9880.9930.8870.9890.9890.9920.8750.9670.9620.9830.703
O0.9060.9530.9480.8580.9070.9550.9460.8290.6980.8480.8640.548
P0.9240.960.9540.8630.9250.9610.9530.8390.7590.8720.8770.568
T0.8690.920.8890.8570.8710.920.8780.8230.5620.7440.7050.57
2 sEC0.8290.8440.8430.7850.8290.8440.8430.7850.6160.6540.7140.514
F0.9810.9670.9890.7960.9810.9670.9890.7960.9630.9250.9820.578
O0.8950.9030.9210.7880.8950.9030.9210.7880.7770.7810.8570.515
P0.8820.8980.9050.7840.8820.8980.9050.7840.7350.770.8270.501
T0.8190.8550.8290.7850.8190.8550.8290.7850.5960.6810.6820.505
PC0.8260.8640.8710.7830.8330.8630.8650.6910.6090.7140.7790.501
F0.9480.9560.970.8070.9490.9570.970.7820.8930.9010.9460.611
O0.8330.8670.8620.7930.8330.8640.8550.7610.6330.7250.7540.544
P0.8340.8860.8860.7850.8410.8870.8810.7880.6290.7560.7930.501
T0.7920.8340.810.7840.80.8210.7930.6530.520.670.6650.5
RC0.7880.8030.7790.7840.8020.7780.7410.6940.5090.5770.580.501
F0.8560.9380.9130.7820.8640.9390.9110.7010.6780.870.8460.506
O0.7950.8060.7790.7910.7960.7810.7470.7810.530.6030.5940.52
P0.8080.8420.8090.7840.8190.8360.7880.7210.5610.6660.6460.511
T0.7850.8010.7670.7850.7450.7730.7230.6150.5010.5860.5610.5
WC0.8920.9460.9180.7980.8890.9460.9160.7660.790.8910.8590.583
F0.9870.9890.9920.8720.9870.9890.9920.8660.9750.9760.9860.773
O0.9230.9610.9390.8210.9230.9610.9380.8030.8440.9180.8910.653
P0.9060.9560.9390.8030.910.9570.9380.7750.7950.9040.8910.588
T0.8340.9180.8650.7960.8390.9170.8580.7630.8390.9170.8580.763

References

  1. Scandurra, V.; Emberti Gialloreti, L.; Barbanera, F.; Scordo, M.R.; Pierini, A.; Canitano, R. Neurodevelopmental disorders and adaptive functions: A study of children with autism spectrum disorders (ASD) and/or attention deficit and hyperactivity disorder (ADHD). Front. Psychiatry 2019, 10, 673. [Google Scholar] [CrossRef] [PubMed]
  2. Pingali, S.; Sunderajan, J. A study of comorbidities in attention deficit hyperactivity disorder: A retrospective analysis of case records. Arch. Ment. Health 2014, 15, 206. [Google Scholar]
  3. Baglioni, C.; Spiegelhalder, K.; Lombardo, C.; Riemann, D. Sleep and emotions: A focus on insomnia. Sleep Med. Rev. 2010, 1, 227–238. [Google Scholar] [CrossRef] [PubMed]
  4. Schwartz, S.; Anderson, W.M.; Cole, S.R.; Cornoni-Huntley, J.; Hays, J.C.; Blazer, D. Insomnia and heart disease: A review of epidemiologic studies. J. Psychosom. Res. 1999, 47, 313–333. [Google Scholar] [CrossRef] [PubMed]
  5. Polanía, R.; Nitsche, M.A.; Ruff, C.C. Studying and modifying brain function with non-invasive brain stimulation. Nat. Neurosci. 2018, 21, 174–187. [Google Scholar] [CrossRef] [PubMed]
  6. Liu, S.; Duan, M.; Sun, Y.; Wang, L.; An, L.; Ming, D. Neural responses to social decision-making in suicide attempters with mental disorders. BMC Psychiatry 2023, 23, 19. [Google Scholar] [CrossRef] [PubMed]
  7. Subramaniam, M.; Soh, P.; Vaingankar, J.A.; Picco, L.; Chong, S.A. Quality of life in obsessive-compulsive disorder: Impact of the disorder and of treatment. CNS Drugs 2013, 27, 367–383. [Google Scholar] [CrossRef] [PubMed]
  8. American Psychiatric Association. Diagnostic and Statistical Manual of Mental Disorders: DSM-5; American Psychiatric Association: Washington, DC, USA, 2013; Volume 5. [Google Scholar]
  9. Chow, W.; Doane, M.; Sheehan, J.; Alphs, L.; Le, H. Economic burden among patients with major depressive disorder: An analysis of healthcare resource use, work productivity, and direct and indirect costs by depression severity. Am. J. Manag. Care 2019, 16, e188–e196. [Google Scholar]
  10. Trautmann, S.; Rehm, J.; Wittchen, H.U. The economic costs of mental disorders: Do our societies react appropriately to the burden of mental disorders? EMBO Rep. 2016, 17, 1245–1249. [Google Scholar] [CrossRef]
  11. Saif, M.G.M. Clinical Efficacy of Neurofeedback Protocols in Treatment of Attention Deficit/Hyperactivity Disorder (ADHD): A Systematic Review. Psychiatry Res. Neuroimaging 2023, 335, 111723. [Google Scholar] [CrossRef]
  12. Musser, E.D.; Nigg, J.T. Emotion dysregulation across emotion systems in attention deficit/hyperactivity disorder. J. Clin. Child Adolesc. Psychol. 2019, 48, 153–165. [Google Scholar] [CrossRef] [PubMed]
  13. Chang, Z.; Lichtenstein, P.; Asherson, P.J.; Larsson, H. Developmental twin study of attention problems: High heritabilities throughout development. JAMA Psychiatry 2013, 70, 311–318. [Google Scholar] [CrossRef] [PubMed]
  14. Martin, J.; Walters, R.K.; Demontis, D.; Mattheisen, M.; Lee, S.H.; Robinson, E.; Brikell, I.; Ghirardi, L.; Larsson, H.; Lichtenstein, P.; et al. A genetic investigation of sex bias in the prevalence of attention-deficit/hyperactivity disorder. Biol. Psychiatry 2018, 83, 1044–1053. [Google Scholar] [CrossRef]
  15. Maniruzzaman, M.; Hasan, M.A.M.; Asai, N.; Shin, J. Optimal Channels and Features Selection based ADHD Detection from EEG Signal using Statistical and Machine Learning Techniques. IEEE Access 2023, 11, 33570–33583. [Google Scholar] [CrossRef]
  16. Khare, S.K.; Acharya, U.R. An explainable and interpretable model for attention deficit hyperactivity disorder in children using EEG signals. Comput. Biol. Med. 2023, 155, 106676. [Google Scholar] [CrossRef] [PubMed]
  17. Coelho, B.F.O.; Massaranduba, A.B.R.; dos Santos Souza, C.A.; Viana, G.G.; Brys, I.; Ramos, R.P. Parkinson’s disease effective biomarkers based on Hjorth features improved by machine learning. Expert Syst. Appl. 2023, 212, 118772. [Google Scholar] [CrossRef]
  18. Koh, J.E.; Ooi, C.P.; Lim-Ashworth, N.S.; Vicnesh, J.; Tor, H.T.; Lih, O.S.; Tan, R.S.; Acharya, U.R.; Fung, D.S.S. Automated classification of attention deficit hyperactivity disorder and conduct disorder using entropy features with ECG signals. Comput. Biol. Med. 2022, 140, 105120. [Google Scholar] [CrossRef]
  19. Yasin, S.; Hussain, S.A.; Aslan, S.; Raza, I.; Muzammel, M.; Othmani, A. EEG based Major Depressive disorder and Bipolar disorder detection using Neural Networks: A review. Comput. Methods Programs Biomed. 2021, 202, 106007. [Google Scholar] [CrossRef]
  20. Mulaffer, L.; Shahin, M.; Glos, M.; Penzel, T.; Ahmed, B. Comparing two insomnia detection models of clinical diagnosis techniques. In Proceedings of the 2017 39th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), Jeju, Republic of Korea, 11–15 July 2017; IEEE: Piscataway, NJ, USA, 2017; pp. 3749–3752. [Google Scholar]
  21. Perera, H.; Shiratuddin, M.F.; Wong, K.W. A review of electroencephalogram-based analysis and classification frameworks for dyslexia. In Proceedings of the International Conference on Neural Information Processing, Kyoto, Japan, 16–21 October 2016; Springer: Cham, Switzerland, 2016; pp. 626–635. [Google Scholar]
  22. Perera, P.; Harshani, H.; Shiratuddin, M.F.; Wong, K.W.; Fullarton, K. EEG signal analysis of writing and typing between adults with dyslexia and normal controls. Int. J. Interact. Multimed. Artif. Intell. 2018, 5, 62–67. [Google Scholar] [CrossRef]
  23. Seshadri, N.G.; Agrawal, S.; Singh, B.K.; Geethanjali, B.; Mahesh, V.; Pachori, R.B. EEG based classification of children with learning disabilities using shallow and deep neural network. Biomed. Signal Process. Control 2023, 82, 104553. [Google Scholar] [CrossRef]
  24. Mumtaz, W.; Xia, L.; Ali, S.S.A.; Yasin, M.A.M.; Hussain, M.; Malik, A.S. Electroencephalogram (EEG)-based computer-aided technique to diagnose major depressive disorder (MDD). Biomed. Signal Process. Control 2017, 31, 108–115. [Google Scholar] [CrossRef]
  25. Suuronen, I.; Airola, A.; Pahikkala, T.; Murtojärvi, M.; Kaasinen, V.; Railo, H. Budget-based classification of Parkinson’s disease from resting state EEG. IEEE J. Biomed. Health Inform. 2023, 27, 3740–3747. [Google Scholar] [CrossRef]
  26. Shilaskar, S.; Bhatlawande, S.; Deshmukh, S.; Dhande, H. Prediction of Autism and Dyslexia Using Machine Learning and Clinical Data Balancing. In Proceedings of the 2023 International Conference on Advances in Intelligent Computing and Applications (AICAPS), Kochi, India, 1–3 February 2023; IEEE: Piscataway, NJ, USA, 2023; pp. 1–11. [Google Scholar]
  27. Ahire, N.; Awale, R.; Patnaik, S.; Wagh, A. A comprehensive review of machine learning approaches for dyslexia diagnosis. Multimed. Tools Appl. 2023, 82, 13557–13577. [Google Scholar] [CrossRef]
  28. Zolezzi, D.M.; Alonso-Valerdi, L.M.; Ibarra-Zarate, D.I. EEG frequency band analysis in chronic neuropathic pain: A linear and nonlinear approach to classify pain severity. Comput. Methods Programs Biomed. 2023, 230, 107349. [Google Scholar] [CrossRef] [PubMed]
  29. Aydin, S.; Arica, N.; Ergul, E.; Tan, O. Classification of obsessive-compulsive disorder by EEG complexity and hemispheric dependency measurements. Int. J. Neural Syst. 2015, 25, 1550010. [Google Scholar] [CrossRef]
  30. Alturki, F.A.; AlSharabi, K.; Abdurraqeeb, A.M.; Aljalal, M. EEG signal analysis for diagnosing neurological disorders using discrete wavelet transform and intelligent techniques. Sensors 2020, 20, 2505. [Google Scholar] [CrossRef] [PubMed]
  31. Tawhid, M.N.A.; Siuly, S.; Wang, K.; Wang, H. Automatic and Efficient Framework for Identifying Multiple Neurological Disorders From EEG Signals. IEEE Trans. Technol. Soc. 2023, 4, 76–86. [Google Scholar] [CrossRef]
  32. Siuly, S.; Alçin, Ö.F.; Kabir, E.; Şengür, A.; Wang, H.; Zhang, Y.; Whittaker, F. A new framework for automatic detection of patients with mild cognitive impairment using resting-state EEG signals. IEEE Trans. Neural Syst. Rehabil. Eng. 2020, 28, 1966–1976. [Google Scholar] [CrossRef]
  33. Andrzejak, R.G.; Lehnertz, K.; Mormann, F.; Rieke, C.; David, P.; Elger, C.E. 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 2001, 64, 061907. [Google Scholar] [CrossRef]
  34. Motamedi-Fakhr, S.; Moshrefi-Torbati, M.; Hill, M.; Hill, C.M.; White, P.R. Signal processing techniques applied to human sleep EEG signals—A review. Biomed. Signal Process. Control 2014, 10, 21–33. [Google Scholar] [CrossRef]
  35. Aljalal, M.; Molinas, M.; Aldosari, S.A.; AlSharabi, K.; Abdurraqeeb, A.M.; Alturki, F.A. Mild cognitive impairment detection with optimally selected EEG channels based on variational mode decomposition and supervised machine learning. Biomed. Signal Process. Control 2024, 87, 105462. [Google Scholar] [CrossRef]
  36. Tawhid, M.; Siuly, S.; Wang, H. Diagnosis of autism spectrum disorder from EEG using a time–frequency spectrogram image-based approach. Electron. Lett. 2020, 56, 1372–1375. [Google Scholar] [CrossRef]
  37. Mir, W.A.; Anjum, M.; Shahab, S. Deep-EEG: An optimized and robust framework and method for EEG-based diagnosis of epileptic seizure. Diagnostics 2023, 13, 773. [Google Scholar] [CrossRef] [PubMed]
  38. Aslan, Z.; Akin, M. Automatic Detection of Schizophrenia by Applying Deep Learning over Spectrogram Images of EEG Signals. Trait. Du Signal 2020, 37. [Google Scholar] [CrossRef]
  39. van Dijk, H.; van Wingen, G.; Denys, D.; Olbrich, S.; van Ruth, R.; Arns, M. The two decades brainclinics research archive for insights in neurophysiology (TDBRAIN) database. Sci. Data 2022, 9, 333. [Google Scholar] [CrossRef] [PubMed]
  40. Welch, P. The use of fast Fourier transform for the estimation of power spectra: A method based on time averaging over short, modified periodograms. IEEE Trans. Audio Electroacoust. 1967, 15, 70–73. [Google Scholar] [CrossRef]
  41. Daubechies, I. Ten Lectures on Wavelets; SIAM: Philadelphia, PA, USA, 1992. [Google Scholar]
  42. Samant, A.; Adeli, H. Feature extraction for traffic incident detection using wavelet transform and linear discriminant analysis. Comput. Aided Civ. Infrastruct. Eng. 2000, 15, 241–250. [Google Scholar] [CrossRef]
  43. Stolojescu, C.; Railean, I.; Moga, S.; Isar, A. Comparison of wavelet families with application to WiMAX traffic forecasting. In Proceedings of the 2010 12th International Conference on Optimization of Electrical and Electronic Equipment, Brasov, Romania, 20–22 May 2010; IEEE: Piscataway, NJ, USA, 2010; pp. 932–937. [Google Scholar]
  44. Mahajan, R.; Morshed, B.I. Unsupervised eye blink artifact denoising of EEG data with modified multiscale sample entropy, kurtosis, and wavelet-ICA. IEEE Biomed. Health Inform. 2014, 19, 158–165. [Google Scholar] [CrossRef]
  45. Degirmenci, M.; Yuce, Y.K.; Perc, M.; Isler, Y. Statistically significant features improve binary and multiple Motor Imagery task predictions from EEGs. Front. Hum. Neurosci. 2023, 17. [Google Scholar] [CrossRef]
  46. Van Der Maaten, L.; Postma, E.; Van den Herik, J. Dimensionality reduction: A comparative review. J. Mach. Learn. Res. 2009, 10, 66–71. [Google Scholar]
  47. Gewers, F.L.; Ferreira, G.R.; Arruda, H.F.D.; Silva, F.N.; Comin, C.H.; Amancio, D.R.; Costa, L.D.F. Principal Component Analysis: A Natural Approach to Data Exploration. ACM Comput. Surv. 2021, 54, 1–34. [Google Scholar] [CrossRef]
  48. Inselberg, A. The plane with parallel coordinates. Vis. Comput. 1985, 1, 69–91. [Google Scholar] [CrossRef]
  49. Johansson, J.; Forsell, C. Evaluation of Parallel Coordinates: Overview, Categorization and Guidelines for Future Research. IEEE Trans. Vis. Comput. Graph. 2016, 22, 579–588. [Google Scholar] [CrossRef] [PubMed]
  50. Heinrich, J.; Weiskopf, D. State of the Art of Parallel Coordinates. In Proceedings of the Eurographics 2013—State of the Art Reports, Girona, Spain, 6–10 May 2013; Sbert, M., Szirmay-Kalos, L., Eds.; The Eurographics Association: Eindhoven, The Netherlands, 2013. [Google Scholar] [CrossRef]
  51. Geng, Z.; Peng, Z.; Laramee, R.S.; Roberts, J.C.; Walker, R. Angular Histograms: Frequency-Based Visualizations for Large, High Dimensional Data. IEEE Trans. Vis. Comput. Graph. 2011, 17, 2572–2580. [Google Scholar] [CrossRef]
  52. Silverman, B.W. Density Estimation for Statistics and Data Analysis; Chapman & Hall: London, UK, 1986. [Google Scholar]
  53. Hauser, H.; Ledermann, F.; Doleisch, H. Angular Brushing of Extended Parallel Coordinates. In Proceedings of the IEEE Symposium on Information Visualization (InfoVis’02), Boston, MA, USA, 28–29 October 2002; p. 127. [Google Scholar]
  54. Bok, J.; Kim, B.; Seo, J. Augmenting Parallel Coordinates Plots With Color-Coded Stacked Histograms. IEEE Trans. Vis. Comput. Graph. 2022, 28, 2563–2576. [Google Scholar] [CrossRef]
Figure 1. An example overview of applying an angular histogram to parallel coordinates. The angular histogram shows the distributions of data with different colors overlaid on top of parallel coordinates. In the histogram, the red color indicates a high-density region, and the green color represents a low-density region. F ( α ) indicates the angular frequency distance. (A) Parallel coordinates of the variables of O102∼O104 in the occipital region in the PSD features, (B) schematic design of the angular histogram, and (C) applying angular histograms to the parallel coordinates. In the parallel coordinates (A,C), the values located on the top and bottom of the plots indicate the minimum and maximum values of each variable.
Figure 1. An example overview of applying an angular histogram to parallel coordinates. The angular histogram shows the distributions of data with different colors overlaid on top of parallel coordinates. In the histogram, the red color indicates a high-density region, and the green color represents a low-density region. F ( α ) indicates the angular frequency distance. (A) Parallel coordinates of the variables of O102∼O104 in the occipital region in the PSD features, (B) schematic design of the angular histogram, and (C) applying angular histograms to the parallel coordinates. In the parallel coordinates (A,C), the values located on the top and bottom of the plots indicate the minimum and maximum values of each variable.
Applsci 14 00273 g001
Figure 2. The comparisons of the channels (O2, C4, Fp1, and T7) with the EC-data using the selected features with the segmentation length (=1 s). The ‘db4’ wavelet and the level six decomposition were used. The x-axis indicates each feature, and the y-axis denotes the mean ( μ ) ± SEM. The plots are arranged by channels (O2: (AD); C4: (EH); Fp1: (IL); and T7: (MP)), and by feature type (Raw: (A,E,I,M); PSD: (B,F,J,N); EEGband: (C,G,K,O); Wavelet: (D,H,L,P)).
Figure 2. The comparisons of the channels (O2, C4, Fp1, and T7) with the EC-data using the selected features with the segmentation length (=1 s). The ‘db4’ wavelet and the level six decomposition were used. The x-axis indicates each feature, and the y-axis denotes the mean ( μ ) ± SEM. The plots are arranged by channels (O2: (AD); C4: (EH); Fp1: (IL); and T7: (MP)), and by feature type (Raw: (A,E,I,M); PSD: (B,F,J,N); EEGband: (C,G,K,O); Wavelet: (D,H,L,P)).
Applsci 14 00273 g002
Figure 3. Average classification performance comparisons of four extracted features (i.e., EEG frequency band, PSD, raw, and wavelet) for all channels in the EC-data. The classification of the four extracted features using the 1 s and 2 s segmentations was evaluated.
Figure 3. Average classification performance comparisons of four extracted features (i.e., EEG frequency band, PSD, raw, and wavelet) for all channels in the EC-data. The classification of the four extracted features using the 1 s and 2 s segmentations was evaluated.
Applsci 14 00273 g003
Figure 4. Region-based classification performance results with 1 s and 2 s segmentations using the four features (EEGband, PSD, raw, and wavelet). (AD) show box plots of the F1 scores with SVM, RF, KNN, and LR, respectively. Regions are arranged along the x-axis—C: central, F: frontal, O: occipital, P: parietal, T: temporal.
Figure 4. Region-based classification performance results with 1 s and 2 s segmentations using the four features (EEGband, PSD, raw, and wavelet). (AD) show box plots of the F1 scores with SVM, RF, KNN, and LR, respectively. Regions are arranged along the x-axis—C: central, F: frontal, O: occipital, P: parietal, T: temporal.
Applsci 14 00273 g004
Figure 5. PCA projections of the (A,E) EEG frequency band, (B,F) PSD, (C,G) raw, and (D,H) wavelet features. (A,B) use the features from the occipital region using a one-second window size. (EH) represent the features from the O2 channel using two-second segmentation. In the visualizations, the blue- and red-colored glyphs denote normal and abnormal groups, respectively. The arrows indicate possible outliers that do not follow the patterns of other instances within the PCA projection space. In the projections, the x- and y-axis indicate the first (PC1)- and second (PC2)-principal components, respectively.
Figure 5. PCA projections of the (A,E) EEG frequency band, (B,F) PSD, (C,G) raw, and (D,H) wavelet features. (A,B) use the features from the occipital region using a one-second window size. (EH) represent the features from the O2 channel using two-second segmentation. In the visualizations, the blue- and red-colored glyphs denote normal and abnormal groups, respectively. The arrows indicate possible outliers that do not follow the patterns of other instances within the PCA projection space. In the projections, the x- and y-axis indicate the first (PC1)- and second (PC2)-principal components, respectively.
Applsci 14 00273 g005
Figure 6. Parallel coordinates with angular histograms of the (A) EEG frequency band, (B) PSD, (C) raw, and (D) wavelet features from the occipital region using the window size (1 s). Significant differences between normal and abnormal groups are emphasized with arrows. Zoomed representations of the highlighted regions (indicated with rectangular outlines) are arranged on the right side of the paralleled coordinates. Arrows indicate distinctive patterns compared to other data distributions. In detail, high-density regions (colored red in angular histograms) have appeared in multiple locations in the histograms.
Figure 6. Parallel coordinates with angular histograms of the (A) EEG frequency band, (B) PSD, (C) raw, and (D) wavelet features from the occipital region using the window size (1 s). Significant differences between normal and abnormal groups are emphasized with arrows. Zoomed representations of the highlighted regions (indicated with rectangular outlines) are arranged on the right side of the paralleled coordinates. Arrows indicate distinctive patterns compared to other data distributions. In detail, high-density regions (colored red in angular histograms) have appeared in multiple locations in the histograms.
Applsci 14 00273 g006aApplsci 14 00273 g006b
Figure 7. Analyzing patterns by age groups with parallel coordinates visualizations. Statistically significant wavelet features from the central region (window size = 1 s and bin size = 25) are used to generate the visualization. Each horizontally positioned parallel coordinates visualization represents an individual age group (i.e., 0: 0∼9, 10: 10∼19, , 70: 70∼79). Various background color attributes are used to represent data patterns in different age groups.
Figure 7. Analyzing patterns by age groups with parallel coordinates visualizations. Statistically significant wavelet features from the central region (window size = 1 s and bin size = 25) are used to generate the visualization. Each horizontally positioned parallel coordinates visualization represents an individual age group (i.e., 0: 0∼9, 10: 10∼19, , 70: 70∼79). Various background color attributes are used to represent data patterns in different age groups.
Applsci 14 00273 g007
Figure 8. Analyzing features based on eight conditions (seven neurological disabilities and one healthy) with parallel coordinates visualization with angular histograms. Fifty-five significant wavelet features from the central region using the segmentation (1 s) and the bin size (25) are used to generate the visualization. In the parallel coordinates, numerous skewed angular histograms are generated depending on data distributions. Among them, three distinctively visible asymmetric distributions are observed as (A) skinny distribution, (B) fat distribution, (C) fat distribution with a slim peak.
Figure 8. Analyzing features based on eight conditions (seven neurological disabilities and one healthy) with parallel coordinates visualization with angular histograms. Fifty-five significant wavelet features from the central region using the segmentation (1 s) and the bin size (25) are used to generate the visualization. In the parallel coordinates, numerous skewed angular histograms are generated depending on data distributions. Among them, three distinctively visible asymmetric distributions are observed as (A) skinny distribution, (B) fat distribution, (C) fat distribution with a slim peak.
Applsci 14 00273 g008
Table 1. Descriptive statistics for the data used in this study. (* no age information is found in the original dataset.)
Table 1. Descriptive statistics for the data used in this study. (* no age information is found in the original dataset.)
MaleFemale
# of Subjects (Mean ± SD)# of Subjects (Mean ± SD)
MDD67 (42.11 ± 12.48)66 (43.59 ±13.53)
Dyslexia12 (11.04 ± 2.17)9 (9.68 ± 0.92)
Chronic pain3 (34.95 ± 15.29)9 (49.95 ± 4.40)
OCD10 (36.33 ± 13.94)12 (36.15 ± 14.63)
ADHD8 (29.96 ± 14.14)26 (20.68 ± 13.20)
Parkinson4 (66.46 ± 11.01)13 (65.86 ± 9.04)
Burnout1 (-) *5 (50.02 ± 2.33)
Healthy32 (23.41± 19.36)16 (31.19± 15.48)
Table 2. Five brain regions are defined and analyzed. Each region consists of multiple relevant channels to initiate the region-based analysis.
Table 2. Five brain regions are defined and analyzed. Each region consists of multiple relevant channels to initiate the region-based analysis.
RegionChannel Information
FrontalFP1, FP2, F3,F4, F7, F8, AF3, Fz, Fpz
TemporalT3, T4, T5, T6, T7, T8
CentralC3, C4, CZ, C5, C1, C2
ParietalP3, Pz, P4,
OccipitalO1, O2, Oz
Table 3. F1 scores of using two wavelets and two decomposition levels on the EC- and EO-data.
Table 3. F1 scores of using two wavelets and two decomposition levels on the EC- and EO-data.
RegionlwECEO
SVMRFKNNLRSVMRFKNNLR
CentralSixdb40.8800.9370.9430.8160.8850.9450.9160.757
coif30.9010.9430.9460.7490.8820.9390.9530.753
Sevendb40.8860.9510.9470.7600.9040.9540.9530.706
coif30.8780.9310.9360.8450.8850.9400.9320.846
FrontalSixdb40.9690.9760.9870.8590.9870.9890.9920.866
coif30.9770.9740.9900.7620.9730.9780.9860.730
Sevendb40.9630.9790.9900.7070.9590.9850.9920.695
coif30.9640.9780.9910.8770.9710.9870.9930.857
OccipitalSixdb40.8650.9270.9390.8150.9190.9600.9380.697
coif30.8840.9390.9500.7740.8020.8860.9460.956
Sevendb40.8840.9410.9290.7380.8410.8760.9440.943
coif30.9030.9380.9370.8330.8840.9540.9430.841
ParietalSixdb40.8550.9350.9420.8190.8980.9550.9370.762
coif30.8510.9380.9370.7540.8740.9360.9420.759
Sevendb40.8960.9390.9390.7500.8920.9570.9500.709
coif30.8770.9290.9230.8240.8780.9470.9410.827
TemporalSixdb40.8170.8990.8830.8150.7970.9130.8580.760
coif30.7940.9010.8790.7470.8170.9180.9060.672
Sevendb40.8070.9090.8830.6990.8250.9160.8880.765
coif30.8440.9100.8870.8310.8300.9110.9000.804
l: decomposition level, w: wavelet.
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.

Share and Cite

MDPI and ACS Style

Ji, S.Y.; Jayarathna, S.; Perrotti, A.M.; Kardiasmenos, K.; Jeong, D.H. Identifying Patterns for Neurological Disabilities by Integrating Discrete Wavelet Transform and Visualization. Appl. Sci. 2024, 14, 273. https://doi.org/10.3390/app14010273

AMA Style

Ji SY, Jayarathna S, Perrotti AM, Kardiasmenos K, Jeong DH. Identifying Patterns for Neurological Disabilities by Integrating Discrete Wavelet Transform and Visualization. Applied Sciences. 2024; 14(1):273. https://doi.org/10.3390/app14010273

Chicago/Turabian Style

Ji, Soo Yeon, Sampath Jayarathna, Anne M. Perrotti, Katrina Kardiasmenos, and Dong Hyun Jeong. 2024. "Identifying Patterns for Neurological Disabilities by Integrating Discrete Wavelet Transform and Visualization" Applied Sciences 14, no. 1: 273. https://doi.org/10.3390/app14010273

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop