Calibration-Free Driver Drowsiness Classification With Prototype-Based Multi-Domain Mixup

被引:0
|
作者
Kim, Dong-Young [1 ]
Han, Dong-Kyun [2 ]
Jeong, Ji-Hoon [3 ]
Lee, Seong-Whan [1 ]
机构
[1] Korea Univ, Dept Artificial Intelligence, Seoul 02841, South Korea
[2] Korea Univ, Dept Brain & Cognit Engn, Seoul 02841, South Korea
[3] Chungbuk Natl Univ, Sch Comp Sci, Cheongju 28644, South Korea
基金
新加坡国家研究基金会;
关键词
Electroencephalography; Vehicles; Feature extraction; Calibration; Brain modeling; Monitoring; Training; Sleep; Safety; Computer vision; Driver drowsiness state classification; calibration-free; electroencephalogram; domain generalization; prototype-based multi-domain mixup; EEG;
D O I
10.1109/TITS.2024.3522308
中图分类号
TU [建筑科学];
学科分类号
0813 ;
摘要
Drowsy driving is one of the greatest threats to road safety, which increases the importance of intelligent systems that can monitor driver drowsiness. Electroencephalogram (EEG)-based monitoring systems have gained attention because EEG is known to directly measure brain activities that reflect the mental state of the driver. However, calibration is necessary before using the system because EEG signals vary between and within subjects. Therefore, generalized EEG-based drowsiness estimation has become challenging. In this paper, we propose an EEG-based driver drowsiness classification framework without the need for calibration, which can be generalized to unseen subjects. We augment the features of unseen domains (i.e., subjects) with a Dirichlet mixup between prototypes of source domains to complement other domain knowledge. The parameter alpha vector of the Dirichlet distribution adjusts the intensity of the mixup, allowing for diverse enhancement. Furthermore, we utilize an auxiliary batch normalization module for augmented samples to avoid inaccurate estimation by the difference in distribution. The experiments were carried out using two EEG datasets, each measured using different drowsiness indicators, the Karolinska sleepiness scale, and reaction time. In leave-one-subject-out cross-validation, the proposed framework achieved outstanding performance in both datasets, an F1-score of 62.69% and 70.33% and an area under the receiver operating characteristic curve (AUROC) of 71.73% and 73.80%, respectively. The experimental results demonstrate the potential for practical applications of brain-computer interfaces without calibration.
引用
收藏
页码:2955 / 2966
页数:12
相关论文
共 50 条
  • [1] Calibration-Free Driver Drowsiness Classification based on Manifold-Level Augmentation
    Kim, Dong-Young
    Han, Dong-Kyun
    Shin, Hye-Bin
    2023 11TH INTERNATIONAL WINTER CONFERENCE ON BRAIN-COMPUTER INTERFACE, BCI, 2023,
  • [2] ENHANCING FEDERATED DOMAIN ADAPTATION WITH MULTI-DOMAIN PROTOTYPE-BASED FEDERATED FINE-TUNING
    Zhang, Jingyuan
    Duan, Yiyang
    Niu, Shuaicheng
    Cao, Yang
    Lim, Wei Yang Bryan
    arXiv,
  • [3] MIXUP REGULARIZED ADVERSARIAL NETWORKS FOR MULTI-DOMAIN TEXT CLASSIFICATION
    Wu, Yuan
    Inkpen, Diana
    El-Roby, Ahmed
    2021 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP 2021), 2021, : 7733 - 7737
  • [4] Prototype-Based Domain Description for One-Class Classification
    Angiulli, Fabrizio
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2012, 34 (06) : 1131 - 1144
  • [5] A multi-granularity locally optimal prototype-based approach for classification
    Gu, Xiaowei
    Li, Miqing
    INFORMATION SCIENCES, 2021, 569 : 157 - 183
  • [6] Classification of coal/rock based on Hyperspectral LiDAR calibration-free signals
    He, Zixin
    Shao, Hui
    Guo, Hang
    Chen, Jie
    Hongwai yu Jiguang Gongcheng/Infrared and Laser Engineering, 2021, 50 (10):
  • [7] Prototype-Based Self-Adaptive Distribution Calibration for Few-Shot Image Classification
    Du, Wei
    Hu, Xiaoping
    Wei, Xin
    Zuo, Ke
    ELECTRONICS, 2023, 12 (01)
  • [8] Semi-supervised Domain Adaptation via Prototype-based Multi-level Learning
    Huang, Xinyang
    Zhu, Chuang
    Chen, Wenkai
    PROCEEDINGS OF THE THIRTY-SECOND INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, IJCAI 2023, 2023, : 884 - 892
  • [9] The Classification of Multi-Domain Samples Based on the Cooperation of Multiple Models
    Song, Qingzeng
    Xu, Junting
    Ma, Lei
    Yang, Ping
    Jin, Guanghao
    COMPLEXITY, 2022, 2022
  • [10] Heart Sounds Classification Based on Xgboost and Multi-domain Features
    Li, Ting
    Chen, Xingrong
    Zhou, Lin
    Zhu, Hanxia
    Sun, Fuming
    BASIC & CLINICAL PHARMACOLOGY & TOXICOLOGY, 2020, 127 : 9 - 10