Exact Fit Attention in Node-Holistic Graph Convolutional Network for Improved EEG-Based Driver Fatigue Detection

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Xu, Meiyan, Chen, Qingqing, Chen, Duo, Ding, Yi, Wang, Jingyuan, Gu, Peipei, Pan, Yijie, Huang, Deshuang, Zhang, Xun, Guo, Jiayang
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909466326204416
author Xu, Meiyan
Chen, Qingqing
Chen, Duo
Ding, Yi
Wang, Jingyuan
Gu, Peipei
Pan, Yijie
Huang, Deshuang
Zhang, Xun
Guo, Jiayang
author_facet Xu, Meiyan
Chen, Qingqing
Chen, Duo
Ding, Yi
Wang, Jingyuan
Gu, Peipei
Pan, Yijie
Huang, Deshuang
Zhang, Xun
Guo, Jiayang
contents EEG-based fatigue monitoring can effectively reduce the incidence of related traffic accidents. In the past decade, with the advancement of deep learning, convolutional neural networks (CNN) have been increasingly used for EEG signal processing. However, due to the data's non-Euclidean characteristics, existing CNNs may lose important spatial information from EEG, specifically channel correlation. Thus, we propose the node-holistic graph convolutional network (NHGNet), a model that uses graphic convolution to dynamically learn each channel's features. With exact fit attention optimization, the network captures inter-channel correlations through a trainable adjacency matrix. The interpretability is enhanced by revealing critical areas of brain activity and their interrelations in various mental states. In validations on two public datasets, NHGNet outperforms the SOTAs. Specifically, in the intra-subject, NHGNet improved detection accuracy by at least 2.34% and 3.42%, and in the inter-subjects, it improved by at least 2.09% and 15.06%. Visualization research on the model revealed that the central parietal area plays an important role in detecting fatigue levels, whereas the frontal and temporal lobes are essential for maintaining vigilance.
format Preprint
id arxiv_https___arxiv_org_abs_2501_15062
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Exact Fit Attention in Node-Holistic Graph Convolutional Network for Improved EEG-Based Driver Fatigue Detection
Xu, Meiyan
Chen, Qingqing
Chen, Duo
Ding, Yi
Wang, Jingyuan
Gu, Peipei
Pan, Yijie
Huang, Deshuang
Zhang, Xun
Guo, Jiayang
Machine Learning
EEG-based fatigue monitoring can effectively reduce the incidence of related traffic accidents. In the past decade, with the advancement of deep learning, convolutional neural networks (CNN) have been increasingly used for EEG signal processing. However, due to the data's non-Euclidean characteristics, existing CNNs may lose important spatial information from EEG, specifically channel correlation. Thus, we propose the node-holistic graph convolutional network (NHGNet), a model that uses graphic convolution to dynamically learn each channel's features. With exact fit attention optimization, the network captures inter-channel correlations through a trainable adjacency matrix. The interpretability is enhanced by revealing critical areas of brain activity and their interrelations in various mental states. In validations on two public datasets, NHGNet outperforms the SOTAs. Specifically, in the intra-subject, NHGNet improved detection accuracy by at least 2.34% and 3.42%, and in the inter-subjects, it improved by at least 2.09% and 15.06%. Visualization research on the model revealed that the central parietal area plays an important role in detecting fatigue levels, whereas the frontal and temporal lobes are essential for maintaining vigilance.
title Exact Fit Attention in Node-Holistic Graph Convolutional Network for Improved EEG-Based Driver Fatigue Detection
topic Machine Learning
url https://arxiv.org/abs/2501.15062