Topological Feature Search Method for Multichannel EEG: Application in ADHD classification

Tianming Cai,Guoying Zhao,Junbin Zang,Chen Zong,Zhidong Zhang,Chenyang Xue
2024-04-10
Abstract:In recent years, the preliminary diagnosis of Attention Deficit Hyperactivity Disorder (ADHD) using electroencephalography (EEG) has garnered attention from researchers. EEG, known for its expediency and efficiency, plays a pivotal role in the diagnosis and treatment of ADHD. However, the non-stationarity of EEG signals and inter-subject variability pose challenges to the diagnostic and classification processes. Topological Data Analysis (TDA) offers a novel perspective for ADHD classification, diverging from traditional time-frequency domain features. Yet, conventional TDA models are restricted to single-channel time series and are susceptible to noise, leading to the loss of topological features in persistence diagrams.This paper presents an enhanced TDA approach applicable to multi-channel EEG in ADHD. Initially, optimal input parameters for multi-channel EEG are determined. Subsequently, each channel's EEG undergoes phase space reconstruction (PSR) followed by the utilization of k-Power Distance to Measure (k-PDTM) for approximating ideal point clouds. Then, multi-dimensional time series are re-embedded, and TDA is applied to obtain topological feature information. Gaussian function-based Multivariate Kernel Density Estimation (MKDE) is employed in the merger persistence diagram to filter out desired topological feature mappings. Finally, persistence image (PI) method is utilized to extract topological features, and the influence of various weighting functions on the results is discussed.The effectiveness of our method is evaluated using the IEEE ADHD dataset. Results demonstrate that the accuracy, sensitivity, and specificity reach 85.60%, 83.61%, and 88.33%, respectively. Compared to traditional TDA methods, our method was effectively improved and outperforms typical nonlinear descriptors. These findings indicate that our method exhibits higher precision and robustness.
Machine Learning,Signal Processing,Applications
What problem does this paper attempt to address?
The main problem that this paper attempts to solve is to improve the accuracy and robustness in the classification of multi - channel EEG signals for Attention - Deficit / Hyperactivity Disorder (ADHD). Specifically, the research aims to accurately identify the hidden topological features in multi - channel EEG time series through an improved Topological Data Analysis (TDA) method, so as to more effectively distinguish between ADHD patients and normal individuals. ### Problem Background 1. **Non - stationarity and Inter - individual Differences**: The non - stationarity of EEG signals and the variability among individuals pose challenges to the diagnosis and classification process. 2. **Limitations of Traditional TDA**: Traditional TDA methods are only applicable to single - channel time series and are easily affected by noise interference, resulting in the loss of topological features in persistence diagrams. ### Solution To solve the above problems, the paper proposes an enhanced TDA method. The specific steps are as follows: 1. **Determination of Optimal Input Parameters**: Determine the best input parameters for multi - channel EEG. 2. **Phase - Space Reconstruction (PSR)**: Perform phase - space reconstruction on the EEG of each channel. 3. **k - Power Distance to Measure (k - PDTM)**: Use k - PDTM to approximate the ideal point cloud. 4. **Multidimensional Time - Series Re - embedding**: Re - embed the multidimensional time series and apply TDA to obtain topological feature information. 5. **Multivariate Kernel Density Estimation Based on Gaussian Function (MKDE)**: Filter out the required topological feature maps in the combined persistence diagrams. 6. **Persistence Image (PI) Method**: Use the PI method to extract topological features and discuss the influence of different weighting functions on the results. ### Main Contributions 1. **Non - linear Topological Analysis**: Conduct non - linear topological analysis on multidimensional time sequences, revealing the topological changes in the EEG of ADHD patients. 2. **Improved TDA Method**: By introducing k - PDTM and MKDE filtering, the robustness and accuracy of the method are ensured. 3. **Persistence Image Method**: Further extract topological features from persistence diagrams and discuss the influence of different weighting functions on the results. 4. **Experimental Verification**: Use the IEEE ADHD data set to verify the proposed method. The average classification accuracy rate reaches 85.60%, which is better than most existing non - linear descriptors. ### Summary Through these improvements, the paper demonstrates the potential of its method in the early screening and diagnosis of ADHD, improving the precision and robustness of classification.