EEG-GNet: GCN-Based EEG Signal Fatigue Detection Network for Edge Devices

Fatigue driving is a common potential traffic safety hazard. To detect and analysis the driver Fatigue degree in real-time, more attention have been increasing paid to application research of EEG fatigue detection device. However, existing methods suffer from complex model structure or large-scale model parameters. In the other words, they requires high computational consumption and large memory footprint, which creates a contradiction between model and portable EEG machine with limited hardware resources. In this work, we take both time and frequency domains of EEG signal into account and design an lightweight and efficient driver fatigue detection network based on GCN (EEG-GNet), which is available edge-end device application. Experiments on 4th Raspberry Pi on a public baseline dataset show that our EEG-GNet outperforms the state-of-the-art methods and takes good tradeoff between detection accuracy and resource consumption. Our EEG-GNet provides a new potential opportunity for future applications in practical portable/wearable (P/W) devices.