arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.06813eess.SPcs.LG

Improving EEG Decoding via Clustering-based Multi-task Feature Learning

Lehigh University(理海大学);Inception Institute of Artificial Intelligence(起源人工智能研究院);South China University of Technology(华南理工大学);Stanford University(斯坦福大学);West China Hospital of Sichuan University(四川大学华西医院);Guangdong University of Technology(广东工业大学);Tokyo University of Agriculture and Technology(东京农工大学);Nicolaus Copernicus University(尼古拉·哥白尼大学)

Yu Zhang, Tao Zhou, Wei Wu, Hua Xie, Hongru Zhu, Guoxu Zhou, Andrzej Cichocki

英文摘要:

Accurate electroencephalogram (EEG) pattern decoding for specific mental tasks is one of the key steps for the development of brain-computer interface (BCI), which is quite challenging due to the considerably low signal-to-noise ratio of EEG collected at the brain scalp. Machine learning provides a promising technique to optimize EEG patterns toward better decoding accuracy. However, existing algorithms do not effectively explore the underlying data structure capturing the true EEG sample distribution, and hence can only yield a suboptimal decoding accuracy. To uncover the intrinsic distribution structure of EEG data, we propose a clustering-based multi-task feature learning algorithm for improved EEG pattern decoding. Specifically, we perform affinity propagation-based clustering to explore the subclasses (i.e., clusters) in each of the original classes, and then assign each subclass a unique label based on a one-versus-all encoding strategy. With the encoded label matrix, we devise a novel multi-task learning algorithm by exploiting the subclass relationship to jointly optimize the EEG pattern features from the uncovered subclasses. We then train a linear support vector machine with the optimized features for EEG pattern decoding. Extensive experimental studies are conducted on three EEG datasets to validate the effectiveness of our algorithm in comparison with other state-of-the-art approaches. The improved experimental results demonstrate the outstanding superiority of our algorithm, suggesting its prominent performance for EEG pattern decoding in BCI applications.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑