STCGRU: A hybrid model based on CNN and BiGRU for mild cognitive impairment diagnosis
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE(2024)
摘要
Background and Objective: Early diagnosis of mild cognitive impairment (MCI) is one of the essential measures to prevent its further development into Alzheimer's disease (AD). In this paper, we propose a hybrid deep learning model for early diagnosis of MCI, called spatio-temporal convolutional gated recurrent unit network (STCGRU). Methods: The STCGRU comprises three bespoke convolutional neural network (CNN) modules and a bidirectional gated recurrent unit (BiGRU) module, which can effectively extract the spatial and temporal features of EEG and obtain excellent diagnostic results. We use a publicly available EEG dataset that has not undergone pre-processing to verify the robustness and accuracy of the model. Ablation experiments on STCGRU are conducted to showcase the individual performance improvement of each module. Results: Compared with other state-of-the-art approaches using the same publicly available EEG dataset, the results show that STCGRU is more suitable for early diagnosis of MCI. After 10-fold cross-validation, the average classification accuracy of the hybrid model reached 99.95 %, while the average kappa value reached 0.9989. Conclusions: The experimental results show that the hybrid model proposed in this paper can directly extract compelling spatio-temporal features from the raw EEG data for classification. The STCGRU allows for accurate diagnosis of patients with MCI and has a high practical value.
更多查看译文
关键词
Electroencephalography (EEG),Mild cognitive impairment (MCI),Hybrid deep learning model,Convolutional neural network (CNN),Bi-directional gated recurrent unit (BIGRU)
AI 理解论文
溯源树
样例
![](https://originalfileserver.aminer.cn/sys/aminer/pubs/mrt_preview.jpeg)
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要