A bidirectional interaction-based hybrid network architecture for EEG cognitive recognition
Yue Zhao1, Hong Zeng2, Haohao Zheng1
1School of Computer Science and Technology, Hangzhou Dianzi University, Hangzhou 310018, China.
Background And Objective:
Extracting cognitive representation and computational representation information simultaneously from electroencephalography (EEG) data and constructing corresponding information interaction models can effectively improve the recognition capability of brain cognitive status. However, due to the huge gap in the interaction between the two types of information, existing studies have yet to consider the advantages of the interaction of both.
Methods:
This paper introduces a novel architecture named the bidirectional interaction-based hybrid network (BIHN) for EEG cognitive recognition. BIHN consists of two networks: a cognitive-based network named CogN (e.g., graph convolution network, GCN; capsule network, CapsNet) and a computing-based network named ComN (e.g., EEGNet). CogN is responsible for extracting cognitive representation features from EEG data, while ComN is responsible for extracting computational representation features. Additionally, a bidirectional distillation-based coadaptation (BDC) algorithm is proposed to facilitate information interaction between CogN and ComN to realize the coadaptation of the two networks through bidirectional closed-loop feedback.
Results:
Cross-subject cognitive recognition experiments were performed on the Fatigue-Awake EEG dataset (FAAD, 2-class classification) and SEED dataset (3-class classification), and hybrid network pairs of GCN + EEGNet and CapsNet + EEGNet were verified. The proposed method achieved average accuracies of 78.76% (GCN + EEGNet) and 77.58% (CapsNet + EEGNet) on FAAD and 55.38% (GCN + EEGNet) and 55.10% (CapsNet + EEGNet) on SEED, outperforming the hybrid networks without the bidirectional interaction strategy.
Conclusions:
Experimental results show that BIHN can achieve superior performance on two EEG datasets and enhance the ability of both CogN and ComN in EEG processing as well as cognitive recognition. We also validated its effectiveness with different hybrid network pairs. The proposed method could greatly promote the development of brain-computer collaborative intelligence.
