arXiv:2604.23964cs.LGcs.AI2026-04

用扩散增强与任务引导机制,提升脑电图诊断痴呆和认知评分预测的准确率。

Task-guided Spatiotemporal Network with Diffusion Augmentation for EEG-based Dementia Diagnosis and MMSE Prediction

论文配图:Task-guided Spatiotemporal Network with Diffusion Augmentation for EEG-based Dementia Diagnosis and MMSE Prediction
图 1 · 摘自论文原文
  • 设计任务引导的时空网络,分离不同任务特征以减少干扰。
  • 在XY02数据集上实现97.78%痴呆分类准确率,MMSE预测误差降低1.44。
  • 引入扩散数据增强,提升样本多样性,支持跨数据集泛化。

痴呆患者常伴认知障碍,通常通过简易精神状态检查(MMSE)评估。其神经生理异常可通过脑电图(EEG)反映,为联合建模提供依据。然而传统多任务方法存在特征纠缠问题,导致异构目标间相互干扰。为此,本文提出任务引导的时空网络(TGSN)结合扩散增强,用于基于EEG的痴呆诊断与MMSE预测。TGSN融合多频段特征,捕捉互补的频谱信息;引入基于扩散过程的预训练数据增强模块,提升样本多样性;设计门控时空注意力模块,建模EEG的长程空间依赖与时间动态;并设计任务引导查询模块,实现任务特异性特征提取。在XY02数据集上的实验表明,该模型在阿尔茨海默病(AD)/额颞叶痴呆(FTD)分类中达到97.78%准确率,优于最佳基线16.39%;在AD/FTD/血管性认知障碍(VCI)分类中达83.93%,提升8.28%;同时将MMSE预测的均方根误差(RMSE)降至1.93和2.38,分别优于基线1.44和1.43。在DS004504数据集上的验证显示强跨数据集泛化能力。

原文摘要 · Abstract (English)

Patients with dementia typically exhibit cognitive impairment, which is routinely assessed using the Mini-Mental State Examination (MMSE). Concurrently, their underlying neurophysiological abnormalities are reflected in Electroencephalography (EEG), providing a basis for joint modeling. However, traditional multi-task approaches suffer from feature entanglement, which leads to inter-task interference when handling heterogeneous objectives.To address this challenge, we propose a task-guided spatiotemporal network (TGSN) with diffusion augmentation for EEG-based dementia diagnosis and MMSE prediction. Specifically, TGSN integrates a multi-band feature fusion module to capture complementary spectral information from EEG. Meanwhile, a pre-trained data augmentation module utilizing a diffusion process is introduced toincrease sample diversity. To model the complex spatiotemporal patterns of EEG, we propose a gated spatiotemporal attention module that captures long-range spatial dependencies and temporal dynamics. Moreover, we design a task-guided query module to achieve task-specific feature extraction, thereby mitigating task interference. The effectiveness of TGSN is evaluated on the XY02 dataset. Experimental results demonstrate that the proposed network outperforms several state-of-the-art methods, achieving classification accuracies of 97.78\% for Alzheimer's Disease (AD)/Frontotemporal Dementia (FTD) and 83.93\% for AD/FTD/Vascular Cognitive Impairment (VCI), which exceed the best baselines by 16.39\% and 8.28\%, respectively. In parallel, it reduces the RMSE for MMSE prediction to 1.93 and 2.38, achieving significant error reductions of 1.44 and 1.43 compared to the best baselines. Additionally, validation on the DS004504 dataset demonstrates strong cross-dataset generalization...

脑电图痴呆诊断多任务学习扩散模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。