Construction of machine learning models for recognizing comorbid anxiety in epilepsy patients based on their clinical and quantitative EEG features

Abstract

Background

This study aimed to construct prediction models for the recognizing of anxiety disorders (AD) in patients with epilepsy (PWEs) by combining clinical features with quantitative electroencephalogram (qEEG) features and using machine learning (ML).

Methods

Nineteen clinical features and 20-min resting-state EEG were collected from 71 PWEs comorbid with AD and another 60 PWEs without AD who met the inclusion-exclusion criteria of this study. The EEG were preprocessed and 684 Phase Locking Value (PLV) and 76 Lempel–Ziv Complexity (LZC) features on four bands were extracted. The Fisher score method was used to rank all the derived features. We constructed four models for recognizing AD in PWEs, whether PWEs based on different combinations of features using eXtreme gradient boosting (XGboost) and evaluated these models using the five-fold cross-validation method.

Results

The prediction model constructed by combining the clinical, PLV, and LZC features showed the best performance, with an accuracy of 96.18%, precision of 94.29%, sensitivity of 98.33%, F1-score of 96.06%, and Area Under the Curve (AUC) of 0.96. The Fisher score ranking results displayed that the top ten features were depression, educational attainment, α_P3_LZC, α_T6-Pz_PLV, α_F7_LZC, β_Fp2-O1_PLV, θ_T4-Cz_PLV, θ_F7-Pz_PLV, α_Fp2_LZC, and θ_T4-Pz_PLV.

Conclusions

The model, constructed by combining the clinical and qEEG features PLV and LZC, efficiently identified the presence of AD comorbidity in PWEs and might have the potential to complement the clinical diagnosis. Our findings suggest that LZC features in the α band and PLV features in Fp2-O1 may be potential biomarkers for diagnosing AD in PWEs.