Using Semi-Supervised Domain Adaptation to Enhance EEG-Based Cross-Task Mental Workload Classification Performance
Mental workload (MWL) assessment is critical for accident prevention and operator safety. However, achieving cross-task generalization of MWL classification models is a significant challenge for real-world applications. Classifiers trained on labeled samples from one task often experience a notable performance drop when directly applied to samples from other tasks, limiting its use cases. To address this issue, we propose a semi-supervised cross-task domain adaptation (SCDA) method using power spectral density (PSD) features for MWL recognition across tasks (MATB-II and n-back). Our results demonstrated that the SCDA method achieved the best cross-task classification performance on our data and COG-BCI public dataset, with accuracies of 90.98% ± 9.36% and 96.61% ± 4.35%, respectively. Furthermore, in the cross-task classification of cross-subject scenarios, SCDA showed the highest average accuracy (75.39% ± 9.56% on our data, 90.98% ± 9.36% on the COG-BCI public dataset). The findings indicate that the semi-supervised transfer learning approach using PSD features is feasible and effective for cross-task MWL assessment.
Authors
- Dong Ming (ORCID: https://orcid.org/0000-0002-8192-2538)
- Tao Wang (ORCID: https://orcid.org/0000-0001-9085-7240)
- Feng He (ORCID: https://orcid.org/0000-0001-8359-2635)
- Yichao Huang (ORCID: https://orcid.org/0000-0003-2320-2919)
- Wenxiao Zhong (ORCID: https://orcid.org/0000-0001-9068-428X)
- Yufeng Ke (ORCID: https://orcid.org/0000-0002-8434-0322)
- Shuang Liu (ORCID: https://orcid.org/0000-0002-4372-8443)
Institutions
- Tianjin University (CN)
- Tianjin Medical University (CN)
Publication Details
- Journal
- IEEE Journal of Biomedical and Health Informatics
- Published
- 2024-08-30
- DOI
- https://doi.org/10.1109/jbhi.2024.3452410
- Citations
- 9
- Primary Topic
- EEG and Brain-Computer Interfaces
- Type
- article
- Field-Weighted Citation Impact
- 3.08
Funders
- National Natural Science Foundation of China