Subject-specific CNN model with parameter-based transfer learning for SSVEP detection

被引:0
|
作者
Ji, Zhouyu [1 ]
Xu, Tao [2 ]
Chen, Chuangquan [1 ]
Yin, Haojun [1 ]
Wan, Feng [3 ,4 ]
Wang, Hongtao [1 ]
机构
[1] Wuyi Univ, Sch Elect & Informat Engn, Jiangmen, Peoples R China
[2] Shantou Univ, Dept Biomed Engn, Shantou, Peoples R China
[3] Univ Macau, Fac Sci & Technol, Dept Elect & Comp Engn, Macau, Peoples R China
[4] Univ Macau, Inst Collaborat Innovat, Ctr Cognit & Brain Sci, Macau, Peoples R China
关键词
Brain-computer interface; Deep learning; Electroencephalogram (EEG); Transfer learning; Steady-state visual evoked potential (SSVEP); BRAIN-COMPUTER INTERFACE; FREQUENCY RECOGNITION; NEURAL-NETWORK; CLASSIFICATION;
D O I
10.1016/j.bspc.2024.107404
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Steady-state visual evoked potentials (SSVEP)-based brain-computer interfaces (BCIs) leverage machine learning methods to enhance performance. However, these methods require a sufficiently long time window to achieve high accuracy and information transfer rate (ITR), which restricts their applications in real-world scenarios, particularly for user-specific decoding. To address this issue, we propose a parameter-based transfer learning CNN (PTL-CNN) approach for the SSVEP-BCI system, which can automatically fuse and extract both inter- and intra-subject features in EEG signals. Specifically, we first introduce a shallow CNN architecture and adopt a short time-window to train a pretrained model on a dataset comprising numerous subjects, aiming to explore the universal features across subjects. Subsequently, anew user is utilized to fine-tune the model, calibrating it to this specific user. Experimental results demonstrate that PTL-CNN achieves remarkable performance and significantly outperforms the compared algorithms under short time windows. For instance, in a time window of 0.4 s, PTL-CNN achieves an average accuracy of 80.60% with an average ITR of 247.77 bits/min on the Benchmark dataset, and an average accuracy of 66.91% with an average ITR of 185.90 bits/min on the Beta dataset. This performance is significantly better than that of Ensemble-TRCA (Benchmark: 71.21%, 209.12 bits/min; Beta: 53.04%, 135.53 bits/min). In summary, our proposed PTL-CNN achieves the highest average accuracy with the fastest average ITR and is of implications for the development of real-time BCI applications, as well as inspiration for other application paradigms.
引用
收藏
页数:9
相关论文
共 50 条
  • [31] TumorDet: A Breast Tumor Detection Model Based on Transfer Learning and ShuffleNet
    Zhang, Tao
    Pan, Leying
    Yang, Qiang
    Yang, Guoping
    Han, Nan
    Qiao, Shaojie
    CURRENT BIOINFORMATICS, 2024, 19 (02) : 119 - 128
  • [32] Diagnosis of cervical cancer using CNN deep learning model with transfer learning approaches
    Sharma, Arpit Kumar
    Nandal, Amita
    Dhaka, Arvind
    Alhudhaif, Adi
    Polat, Kemal
    Sharma, Arvind
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2025, 105
  • [33] A Fusion Model for Cross-Subject Stress Level Detection Based on Transfer Learning
    Mozafari, Mohsen
    Goubran, Rafik
    Green, James R.
    2021 IEEE SENSORS APPLICATIONS SYMPOSIUM (SAS 2021), 2021,
  • [34] Comparison of subject-independent and subject-specific EEG-based BCI using LDA and SVM classifiers
    Eliana M. dos Santos
    Rodrigo San-Martin
    Francisco J. Fraga
    Medical & Biological Engineering & Computing, 2023, 61 : 835 - 845
  • [35] Subject-specific feature identification of arousal and valence based on EEG
    Polo, Edoardo Maria
    Farabbi, Andrea
    Milekic, Maja
    Steyde, Giulio
    Signorini, Maria Gabriella
    Figueiredo, Patricia
    Mainardi, Luca
    Barbieri, Riccardo
    2024 IEEE INTERNATIONAL SYMPOSIUM ON MEDICAL MEASUREMENTS AND APPLICATIONS, MEMEA 2024, 2024,
  • [36] Deep transfer learning CNN based for classification quality of organic vegetables
    Promboonruang, Suksun
    Boonrod, Thummarat
    INTERNATIONAL JOURNAL OF ADVANCED AND APPLIED SCIENCES, 2023, 10 (12): : 203 - 210
  • [37] A Canonical Correlation Analysis-Based Transfer Learning Framework for Enhancing the Performance of SSVEP-Based BCIs
    Wei, Qingguo
    Zhang, Yixin
    Wang, Yijun
    Gao, Xiaorong
    IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2023, 31 : 2809 - 2821
  • [38] A hybrid CNN with transfer learning for skin cancer disease detection
    Shukla, Man Mohan
    Tripathi, B. K.
    Dwivedi, Tanay
    Tripathi, Ashish
    Chaurasia, Brijesh Kumar
    MEDICAL & BIOLOGICAL ENGINEERING & COMPUTING, 2024, 62 (10) : 3057 - 3071
  • [39] A CNN Based Transfer Learning Method for High Impedance Fault Detection
    Zhang, Yongjie
    Wang, Xiaojun
    Luo, Yiping
    Xu, Yin
    He, Jinghan
    Wu, Guohong
    2020 IEEE POWER & ENERGY SOCIETY GENERAL MEETING (PESGM), 2020,
  • [40] Multitype Damage Detection of Container Using CNN Based on Transfer Learning
    Wang, Zixin
    Gao, Jing
    Zeng, Qingcheng
    Sun, Yuhui
    MATHEMATICAL PROBLEMS IN ENGINEERING, 2021, 2021