Bayesian Nonnegative CP Decomposition-based Feature Extraction Algorithm for Drowsiness Detection

作者：Qian Dong, Bei Wang, Yun Qing, Tao Zhang, Yu Zhang, Xing Wang, Masatoshi Nakamura · 发表于：IEEE Transactions on Neural Systems and Rehabilitation Engineering · 年份：2016 · DOI：10.1109/tnsre.2016.2618902 · 被引用次数：16 · 研究领域：Sleep and Work-Related Fatigue、EEG and Brain-Computer Interfaces、Blind Source Separation Techniques

Daytime short nap involves physiological processes, such as alertness, drowsiness and sleep. The study of the relationship between drowsiness and nap based on physiological signals is a great way to have a better understanding of the periodical rhymes of physiological states. A model of Bayesian nonnegative CP decomposition (BNCPD) was proposed to extract common multiway features from the group-level electroencephalogram (EEG) signals. As an extension of the nonnegative CP decomposition, the BNCPD model involves prior distributions of factor matrices, while the underlying CP rank could be determined automatically based on a Bayesian nonparametric approach. In terms of computational speed, variational inference was applied to approximate the posterior distributions of unknowns. Extensive simulations on the synthetic data illustrated the capability of our model to recover the true CP rank. As a real-world application, the performance of drowsiness detection during daytime short nap by using the BNCPD-based features was compared with that of other traditional feature extraction methods. Experimental results indicated that the BNCPD model outperformed other methods for feature extraction in terms of two evaluation metrics, as well as different parameter settings. Our approach is likely to be a useful tool for automatic CP rank determination and offering a plausible multiway physiological information of individual states.