Lan Zhen, Li Zixing, Yan Chao, Xiang Xiaojia, Tang Dengqing, Wu Min, Chen Zhenghua
College of Intelligence Science and Technology, National University of Defense Technology, Changsha, 410073, China; Institute for Infocomm Research (I2R), Agency for Science, Technology and Research (A*STAR), 138632, Singapore.
College of Intelligence Science and Technology, National University of Defense Technology, Changsha, 410073, China.
Neural Netw. 2025 May;185:107133. doi: 10.1016/j.neunet.2025.107133. Epub 2025 Jan 8.
Accurate decoding of electroencephalogram (EEG) signals in the shortest possible time is essential for the realization of a high-performance brain-computer interface (BCI) system based on the steady-state visual evoked potential (SSVEP). However, the degradation of decoding performance of short-length EEG signals is often unavoidable due to the reduced information, which hinders the development of BCI systems in real-world applications. In this paper, we propose a relaxed matching knowledge distillation (RMKD) method to transfer both feature-level and logit-level knowledge in a relaxed manner to improve the decoding performance of short-length EEG signals. Specifically, the long-length EEG signals and short-length EEG signals are decoded into the frequency representation by the teacher and student models, respectively. At the feature-level, the frequency-masked generation distillation is designed to improve the representation ability of student features by forcing the randomly masked student features to generate full teacher features. At the logit-level, the non-target class knowledge distillation and the inter-class relation distillation are combined to mitigate loss conflicts by imitating the distribution of non-target classes and preserve the inter-class relation in the prediction vectors of the teacher and student models. We conduct comprehensive experiments on two public SSVEP datasets in the subject-independent scenario with six different signal lengths. The extensive experimental results demonstrate that the proposed RMKD method has significantly improved the decoding performance of short-length EEG signals in SSVEP-based BCI systems.
在尽可能短的时间内准确解码脑电图(EEG)信号对于实现基于稳态视觉诱发电位(SSVEP)的高性能脑机接口(BCI)系统至关重要。然而,由于信息减少,短长度EEG信号的解码性能下降往往不可避免,这阻碍了BCI系统在实际应用中的发展。在本文中,我们提出了一种宽松匹配知识蒸馏(RMKD)方法,以宽松的方式传递特征级和逻辑级知识,从而提高短长度EEG信号的解码性能。具体而言,长长度EEG信号和短长度EEG信号分别由教师模型和学生模型解码为频率表示。在特征级,频率掩码生成蒸馏旨在通过迫使随机掩码的学生特征生成完整的教师特征来提高学生特征的表示能力。在逻辑级,非目标类知识蒸馏和类间关系蒸馏相结合,通过模仿非目标类的分布来减轻损失冲突,并在教师模型和学生模型的预测向量中保留类间关系。我们在两个公共SSVEP数据集上进行了独立于受试者的场景下的综合实验,信号长度有六种不同情况。大量实验结果表明,所提出的RMKD方法显著提高了基于SSVEP的BCI系统中短长度EEG信号的解码性能。