<p>To address challenges such as the strong non-stationarity and inter-subject distribution shifts of EEG data, as well as the limitations of conventional DANN-based methods in feature representation and multi-source domain adaptation, a Multi-branch Domain Adversarial Neural Network with Multi-scale Channel Attention (MBCA-DANN) is proposed. To enhance feature richness, a multi-scale channel attention Module (MSCA) is designed, which provides multi-scale features and adaptively adjusts the feature channel weights, improving the feature capture ability of the network. A multi-branch architecture is constructed by combining auxiliary Maximum Mean Discrepancy (MMD), domain discriminators, and label discriminators, ensuring optimal matching between the source and target domains. Furthermore, a multi-source domain method with dynamic weight allocation is introduced, enhancing classification performance and robustness. Experimental results demonstrate that the classification accuracy for single-source domain transfer on the MII and MIII datasets is 71.89% and 71.82%, respectively, while the multi-source domain transfer classification accuracy improves to 79.83% and 82.87%. The model achieves a classification accuracy of 98.69% on the fatigue detection dataset, outperforming all currently known state-of-the-art algorithms, validating its strong generalization ability and providing an effective solution for multi-source cross-subject EEG classification.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Multi-branch Domain Adversarial Neural Network with dynamic weight allocation for multi-source EEG classification

  • Yunyuan Gao,
  • Yuetao Ma,
  • Yici Liu,
  • Ganggang Yin,
  • Yanhua Qin

摘要

To address challenges such as the strong non-stationarity and inter-subject distribution shifts of EEG data, as well as the limitations of conventional DANN-based methods in feature representation and multi-source domain adaptation, a Multi-branch Domain Adversarial Neural Network with Multi-scale Channel Attention (MBCA-DANN) is proposed. To enhance feature richness, a multi-scale channel attention Module (MSCA) is designed, which provides multi-scale features and adaptively adjusts the feature channel weights, improving the feature capture ability of the network. A multi-branch architecture is constructed by combining auxiliary Maximum Mean Discrepancy (MMD), domain discriminators, and label discriminators, ensuring optimal matching between the source and target domains. Furthermore, a multi-source domain method with dynamic weight allocation is introduced, enhancing classification performance and robustness. Experimental results demonstrate that the classification accuracy for single-source domain transfer on the MII and MIII datasets is 71.89% and 71.82%, respectively, while the multi-source domain transfer classification accuracy improves to 79.83% and 82.87%. The model achieves a classification accuracy of 98.69% on the fatigue detection dataset, outperforming all currently known state-of-the-art algorithms, validating its strong generalization ability and providing an effective solution for multi-source cross-subject EEG classification.