Jianwei Liu, Enzeng Dong, Jigang Tong, Sen Yang, Shengzhi Du
{"title":"基于CNN-Transformer模型的脑电信号分类","authors":"Jianwei Liu, Enzeng Dong, Jigang Tong, Sen Yang, Shengzhi Du","doi":"10.1109/ICMA57826.2023.10215899","DOIUrl":null,"url":null,"abstract":"Brain-computer interfaces (BCI) based on EEG have attracted extensive research and attention worldwide, while motor imagery (MI), mental arithmetic (MA), and P300 event-related potentials are a few of the more commonly used paradigms.Vision Transformer(ViT) is a new Transformer model that has superior global processing power compared to Convolutional Neural Networks (CNN) and Recurrent Neural Networks (RNN).In this study, we propose a hybrid CNN-Transformer based model that uses CNN to convolve EEG signals in time and space, followed by ViT for global processing, and finally optimizes the model using 10-run $\\times 10$-fold cross-validation and validates it on a publicly available dataset of 29 subjects. Final accuracies of 87.23% and 90.79% were achieved on the MI and MA tasks, respectively. Compared to other literature, the model achieved higher classification accuracies.","PeriodicalId":151364,"journal":{"name":"2023 IEEE International Conference on Mechatronics and Automation (ICMA)","volume":"75 1","pages":"0"},"PeriodicalIF":0.0000,"publicationDate":"2023-08-06","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":"{\"title\":\"Classification of EEG signals based on CNN-Transformer model\",\"authors\":\"Jianwei Liu, Enzeng Dong, Jigang Tong, Sen Yang, Shengzhi Du\",\"doi\":\"10.1109/ICMA57826.2023.10215899\",\"DOIUrl\":null,\"url\":null,\"abstract\":\"Brain-computer interfaces (BCI) based on EEG have attracted extensive research and attention worldwide, while motor imagery (MI), mental arithmetic (MA), and P300 event-related potentials are a few of the more commonly used paradigms.Vision Transformer(ViT) is a new Transformer model that has superior global processing power compared to Convolutional Neural Networks (CNN) and Recurrent Neural Networks (RNN).In this study, we propose a hybrid CNN-Transformer based model that uses CNN to convolve EEG signals in time and space, followed by ViT for global processing, and finally optimizes the model using 10-run $\\\\times 10$-fold cross-validation and validates it on a publicly available dataset of 29 subjects. Final accuracies of 87.23% and 90.79% were achieved on the MI and MA tasks, respectively. Compared to other literature, the model achieved higher classification accuracies.\",\"PeriodicalId\":151364,\"journal\":{\"name\":\"2023 IEEE International Conference on Mechatronics and Automation (ICMA)\",\"volume\":\"75 1\",\"pages\":\"0\"},\"PeriodicalIF\":0.0000,\"publicationDate\":\"2023-08-06\",\"publicationTypes\":\"Journal Article\",\"fieldsOfStudy\":null,\"isOpenAccess\":false,\"openAccessPdf\":\"\",\"citationCount\":\"0\",\"resultStr\":null,\"platform\":\"Semanticscholar\",\"paperid\":null,\"PeriodicalName\":\"2023 IEEE International Conference on Mechatronics and Automation (ICMA)\",\"FirstCategoryId\":\"1085\",\"ListUrlMain\":\"https://doi.org/10.1109/ICMA57826.2023.10215899\",\"RegionNum\":0,\"RegionCategory\":null,\"ArticlePicture\":[],\"TitleCN\":null,\"AbstractTextCN\":null,\"PMCID\":null,\"EPubDate\":\"\",\"PubModel\":\"\",\"JCR\":\"\",\"JCRName\":\"\",\"Score\":null,\"Total\":0}","platform":"Semanticscholar","paperid":null,"PeriodicalName":"2023 IEEE International Conference on Mechatronics and Automation (ICMA)","FirstCategoryId":"1085","ListUrlMain":"https://doi.org/10.1109/ICMA57826.2023.10215899","RegionNum":0,"RegionCategory":null,"ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"","JCRName":"","Score":null,"Total":0}
Classification of EEG signals based on CNN-Transformer model
Brain-computer interfaces (BCI) based on EEG have attracted extensive research and attention worldwide, while motor imagery (MI), mental arithmetic (MA), and P300 event-related potentials are a few of the more commonly used paradigms.Vision Transformer(ViT) is a new Transformer model that has superior global processing power compared to Convolutional Neural Networks (CNN) and Recurrent Neural Networks (RNN).In this study, we propose a hybrid CNN-Transformer based model that uses CNN to convolve EEG signals in time and space, followed by ViT for global processing, and finally optimizes the model using 10-run $\times 10$-fold cross-validation and validates it on a publicly available dataset of 29 subjects. Final accuracies of 87.23% and 90.79% were achieved on the MI and MA tasks, respectively. Compared to other literature, the model achieved higher classification accuracies.