Classification of sleep apnea using EMD-based features and PSO-trained neural networks

Abstract

In this study, we propose a method for detecting obstructive sleep apnea (OSA) based on the features extracted from empirical mode decomposition (EMD) and the neural networks trained by particle swarm optimization (PSO) in the classification phase. After extracting the features from the intrinsic mode functions (IMF) of each heart rate variability (HRV) signal of each segment, these features were applied to the input of popular classifiers such as multi-layer perceptron neural networks (MLPNN), Naïve Bayes, linear discriminant analysis (LDA), k-nearest neighborhood (KNN), and support vector machines (SVM) were applied. The results show that the MLPNN learned with back propagation (BP) algorithm has a diagnostic accuracy of less than 90%, and this may be due to being derivative based property of the BP algorithm, which causes trapping in the local minima. For Improving MLPNN's performance, we used the PSO algorithm instead of the BP method in training part. Therefore, the MLPNN's accuracy improved from 89.36 to 97.66% after the application of the PSO algorithm. The proposed method has also reached to 97.78 and 97.96% in sensitivity and specificity, respectively. So, it can be concluded that the proposed method achieves better or comparable results when compared with the previous works in this field.

Keywords: EMD; MLPNN; PSO; diagnosis; feature extraction; obstructive sleep apnea.