计算机科学
脑电图
人工智能
模式识别(心理学)
联营
卷积神经网络
图形
树(集合论)
数据挖掘
机器学习
理论计算机科学
数学
心理学
精神科
数学分析
作者
Quan Xue,Yi Niu,Huiyu Zhou,Xiaojie Li,Weikuan Jia,Yuanjie Zheng
标识
DOI:10.1142/s0129065723500090
摘要
Accurate identification of driver’s drowsiness state through Electroencephalogram (EEG) signals can effectively reduce traffic accidents, but EEG signals are usually stored in various clients in the form of small samples. This study attempts to construct an efficient and accurate privacy-preserving drowsiness monitoring system, and proposes a fusion model based on tree Federated Learning (FL) and Convolutional Neural Network (CNN), which can not only identify and explain the driver’s drowsiness state, but also integrate the information of different clients under the premise of privacy protection. Each client uses CNN with the Global Average Pooling (GAP) layer and shares model parameters. The tree FL transforms communication relationships into a graph structure, and model parameters are transmitted in parallel along connected branches of the graph. Moreover, the Class Activation Mapping (CAM) is used to find distinctive EEG features for representing specific classes. On EEG data of 11 subjects, it is found that this method has higher average accuracy, F1-score and AUC than the traditional classification method, reaching 73.56%, 73.26% and 78.23%, respectively. Compared with the traditional FL algorithm, this method better protects the driver’s privacy and improves communication efficiency.
科研通智能强力驱动
Strongly Powered by AbleSci AI