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 条
  • [21] Transfer learning of an ensemble of DNNs for SSVEP BCI spellers without user-specific training
    Guney, Osman Berke
    Ozkan, Huseyin
    JOURNAL OF NEURAL ENGINEERING, 2023, 20 (01)
  • [22] Monkeypox Detection Using CNN with Transfer Learning
    Altun, Murat
    Guruler, Huseyin
    Ozkaraca, Osman
    Khan, Faheem
    Khan, Jawad
    Lee, Youngmoon
    SENSORS, 2023, 23 (04)
  • [23] Exploring Dependence of Subject-Specific Training Strategies for EEG Based Brain-Computer Interfaces
    de Wit, T. Warren
    Menon, Vineetha
    Davis, Thomas
    SOUTHEASTCON 2024, 2024, : 848 - 853
  • [24] MDTL: A Novel and Model-Agnostic Transfer Learning Strategy for Cross-Subject Motor Imagery BCI
    Li, Ang
    Wang, Zhenyu
    Zhao, Xi
    Xu, Tianheng
    Zhou, Ting
    Hu, Honglin
    IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2023, 31 : 1743 - 1753
  • [25] Cross-Subject Transfer Learning for Boosting Recognition Performance in SSVEP-Based BCIs
    Zhang, Yue
    Xie, Sheng Quan
    Shi, Chaoyang
    Li, Jun
    Zhang, Zhi-Qiang
    IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2023, 31 : 1574 - 1583
  • [26] A deep learning method for image-based subject-specific local SAR assessment
    Meliado, E. F.
    Raaijmakers, A. J. E.
    Sbrizzi, A.
    Steens, B. R.
    Maspero, M.
    Savenije, M. H. F.
    Luijten, P. R.
    van den Berg, C. A. T.
    MAGNETIC RESONANCE IN MEDICINE, 2020, 83 (02) : 695 - 711
  • [27] Steps toward subject-specific classification in ECG-based detection of sleep apnea
    Maier, Christoph
    Wenz, Heinrich
    Dickhaus, Hartmut
    PHYSIOLOGICAL MEASUREMENT, 2011, 32 (11) : 1807 - 1819
  • [28] Sharing is caring: An extensive analysis of parameter-based transfer learning for the prediction of building thermal dynamics
    Pinto, Giuseppe
    Messina, Riccardo
    Li, Han
    Hong, Tianzhen
    Piscitelli, Marco Savino
    Capozzoli, Alfonso
    ENERGY AND BUILDINGS, 2022, 276
  • [29] Transfer Learning for SSVEP-based BCI using Riemannian similarities between users
    Kalunga, Emmanuel K.
    Chevallier, Sylvain
    Barthelemy, Quentin
    2018 26TH EUROPEAN SIGNAL PROCESSING CONFERENCE (EUSIPCO), 2018, : 1685 - 1689
  • [30] Deep transfer learning CNN based approach for COVID-19 detection
    Muhammad, Wazir
    Bhutto, Zuhaibuddin
    Shah, Syed Ali Raza
    Shah, Jalal
    Shaikh, Murtaza Hussain
    Hussain, Ayaz
    Thaheem, Imdadullah
    Ali, Shamshad
    INTERNATIONAL JOURNAL OF ADVANCED AND APPLIED SCIENCES, 2022, 9 (04): : 44 - 52