Phase lag index-based graph attention networks for detecting driving fatigue

被引:21
作者
Wang, Zhongmin [1 ,2 ]
Zhao, Yupeng [1 ]
He, Yan [1 ,2 ]
Zhang, Jie [1 ,2 ]
机构
[1] Xian Univ Posts & Telecommun, Sch Comp Sci & Technol, Xian 710121, Shaanxi, Peoples R China
[2] Xian Univ Posts & Telecommun, Shaanxi Key Lab Network Data Anal & Intelligent, Xian 710121, Shaanxi, Peoples R China
基金
中国国家自然科学基金;
关键词
EEG;
D O I
10.1063/5.0056139
中图分类号
TH7 [仪器、仪表];
学科分类号
0804 ; 080401 ; 081102 ;
摘要
It is important to understand the changes in the characteristics of the brain network in the state of driving fatigue and to reveal the pattern of functional connectivity between brain regions when fatigue occurs. This paper proposes a method for the detection of driving fatigue based on electroencephalogram (EEG) signals using a phase lag index graph attention network (PLI-GAT). Phase synchronization between EEG signals is a key attribute for establishing communication links among different regions of the brain, and so, the PLI is used to construct a functional brain network reflecting the relationship between EEG signals from different channels. Multi-channel EEG time-frequency features are then modeled as graph data, and the driving fatigue monitoring model is trained using a GAT. Compared with traditional graph neural networks, the GAT applies an aggregation operation to adjacent EEG channel features through the attention mechanism. This enables the adaptive assignment of different neighbor weights, which greatly improves the expressiveness of the graph neural network model. The proposed method is validated on the publicly available SEED-VIG dataset, and the accuracy of fatigue state recognition is found to reach 85.53%. The results show that the functional connectivity among different channels is significantly enhanced in the fatigue state. Published under an exclusive license by AIP Publishing. https://doi.org/10.1063/5.0056139
引用
收藏
页数:9
相关论文
共 31 条
[1]   Semi-Supervised Learning and Graph Neural Networks for Fake News Detection [J].
Benamira, Adrien ;
Devillers, Benjamin ;
Lesot, Etienne ;
Ray, Ayush K. ;
Saadi, Manal ;
Malliaros, Fragkiskos D. .
PROCEEDINGS OF THE 2019 IEEE/ACM INTERNATIONAL CONFERENCE ON ADVANCES IN SOCIAL NETWORKS ANALYSIS AND MINING (ASONAM 2019), 2019, :568-569
[2]  
Bhardwaj I, 2018, 2018 INTERNATIONAL CONFERENCE ON COMPUTING, POWER AND COMMUNICATION TECHNOLOGIES (GUCON), P91, DOI 10.1109/GUCON.2018.8674992
[3]  
Chai RF, 2017, IEEE ENG MED BIO, P1808, DOI 10.1109/EMBC.2017.8037196
[4]   Functional Connectivity Analysis of Mental Fatigue Reveals Different Network Topological Alterations Between Driving and Vigilance Tasks [J].
Dimitrakopoulos, Georgios N. ;
Kakkos, Ioannis ;
Dai, Zhongxiang ;
Wang, Hongtao ;
Sgarbas, Kyriakos ;
Thakor, Nitish ;
Bezerianos, Anastasios ;
Sun, Yu .
IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2018, 26 (04) :740-749
[5]   Driving Mental Fatigue Classification Based on Brain Functional Connectivity [J].
Dimitrakopoulos, Georgios N. ;
Kakkos, Ioannis ;
Vrahatis, Aristidis G. ;
Sgarbas, Kyriakos ;
Li, Junhua ;
Sun, Yu ;
Bezerianos, Anastasios .
ENGINEERING APPLICATIONS OF NEURAL NETWORKS, EANN 2017, 2017, 744 :465-474
[6]   EEG-Based Spatio-Temporal Convolutional Neural Network for Driver Fatigue Evaluation [J].
Gao, Zhongke ;
Wang, Xinmin ;
Yang, Yuxuan ;
Mu, Chaoxu ;
Cai, Qing ;
Dang, Weidong ;
Zuo, Siyang .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2019, 30 (09) :2755-2763
[7]  
Han C., 2018, 2018 11 INT C IM SIG, P1
[8]   Brain Complex Network Characteristic Analysis of Fatigue during Simulated Driving Based on Electroencephalogram Signals [J].
Han, Chunxiao ;
Sun, Xiaozhou ;
Yang, Yaru ;
Che, Yanqiu ;
Qin, Yingmei .
ENTROPY, 2019, 21 (04)
[9]   Graph Attention Networks With Local Structure Awareness for Knowledge Graph Completion [J].
Ji, Kexi ;
Hui, Bei ;
Luo, Guangchun .
IEEE ACCESS, 2020, 8 :224860-224870
[10]  
Kipf T. N., 2016, ARXIV