An improved multi-source domain adaptation network for inter-subject mental fatigue detection based on DANN
- PMID: 36797837
- DOI: 10.1515/bmt-2022-0354
An improved multi-source domain adaptation network for inter-subject mental fatigue detection based on DANN
Abstract
Objectives: Electroencephalogram (EEG) is often used to detect mental fatigue because of its real-time characteristic and objective nature. However, because of the individual variability of EEG among different individuals, tedious and time-consuming calibration sessions are needed.
Methods: Therefore, we propose a multi-source domain adaptation network for inter-subject mental fatigue detection named FLDANN, which is short for focal loss based domain-adversarial training of neural network. As for mental state feature extraction, power spectrum density is extracted based on the Welch method from four sub-bands of EEG signals. The features of the source domain and target domain are fed into the FLDANN network. The contributions of FLDANN include: (1) It uses the idea of adversarial to reduce feature differences between the source and target domain. (2) A loss function named focal loss is used to assign weights to source and target domain samples.
Results: The experiment result shows that when the number of the source domains increases, the classification accuracy of domain-adversarial training of neural network (DANN) gradually decreases and finally tends to be stable. The proposed method achieves an accuracy of 84.10% ± 8.75% on the SEED-VIG dataset and 65.42% ± 7.47% on the self-designed dataset. In addition, the proposed method is compared with other domain adaptation methods and the results show that the proposed method outperforms those state-of-the-art methods.
Conclusions: The result proves that the proposed method is able to solve the problem of individual differences across subjects and to solve the problem of low classification performance of multi-source domain transfer learning.
Keywords: EEG; FLDANN; domain adaptation; inter-subject; mental fatigue; multi-source domain.
© 2023 Walter de Gruyter GmbH, Berlin/Boston.
References
-
- Tran, Y, Craig, A, Craig, R, Chai, R, Nguyen, H. The influence of mental fatigue on brain activity: evidence from a systematic review with meta-analyses. Psychophysiology 2020;575:e13554. https://doi.org/10.1111/psyp.13554 . - DOI
-
- Blanco-Díaz, FC, Guerrero-Méndez, CD, Bastos-Filho, T, Jaramillo-Isaza, S, Ruiz-Olaya, AF. Effects of the concentration level, eye fatigue and coffee consumption on the performance of a BCI system based on visual ERP-P300. J Neurosci Methods 2022;382:109722. https://doi.org/10.1016/j.jneumeth.2022.109722 . - DOI
-
- Li, S, Duan, J, Sun, Y, Sheng, X, Zhu, X, Meng, J. Exploring fatigue effects on performance variation of intensive brain–computer interface practice. Front Neurosci 2021;15:773790. https://doi.org/10.3389/fnins.2021.773790 . - DOI
-
- Wang, H, Dragomir, A, Abbasi, NI, Li, J, Thakor, NV, Bezerianos, A. A novel real-time driving fatigue detection system based on wireless dry EEG. Cogn Neurodyn 2018;12:365–76. https://doi.org/10.1007/s11571-018-9481-5 . - DOI
-
- Min, J, Wang, P, Hu, J. Driver fatigue detection through multiple entropy fusion analysis in an EEG-based system. PLoS One 2017;12:e0188756. https://doi.org/10.1371/journal.pone.0188756 . - DOI
MeSH terms
LinkOut - more resources
Full Text Sources
Research Materials