用脑电图和深度学习,100%精准识别多动症患者
ADHDeepNet From Raw EEG to Diagnosis: Improving ADHD Diagnosis through Temporal-Spatial Processing, Adaptive Attention Mechanisms, and Explainability in Raw EEG Signals
- 融合时空特征与自适应注意力,从原始脑电图直接诊断
- 在121人数据集上达到100%敏感度和99.17%准确率
- 通过权重分析和可视化揭示关键脑区与频段,可解释性强
注意缺陷多动障碍(ADHD)是儿童常见脑部疾病,可能持续至成年,影响社交、学业与职业发展。早期诊断对减轻患者及医疗系统负担至关重要,但传统方法耗时费力。本文提出一种新方法,利用深度学习(DL)与脑电图(EEG)信号提升ADHD诊断的精度与效率。我们设计了ADHDeepNet模型,结合全面的时空表征、注意力模块与可解释性技术,优化用于原始EEG信号处理。模型通过特征提取与精炼流程提升诊断性能。在包含121名参与者(61名ADHD,60名健康对照)的数据集上训练并验证,采用嵌套交叉验证确保结果稳健。采用两阶段方法:每轮迭代使用内层2折交叉验证优化超参数;随后应用不同标准差和放大倍数的加性高斯噪声(AGN)进行数据增强。最终模型在分类中实现100%敏感度与99.17%准确率。为解析模型可解释性,我们分析了主要层的权重与激活模式,并使用t-SNE可视化高维数据,辅助理解决策依据。研究证明,深度学习与脑电图结合在提升ADHD诊断准确性与效率方面具有巨大潜力。
原文摘要 · Abstract (English)
Attention Deficit Hyperactivity Disorder (ADHD) is a common brain disorder in children that can persist into adulthood, affecting social, academic, and career life. Early diagnosis is crucial for managing these impacts on patients and the healthcare system but is often labor-intensive and time-consuming. This paper presents a novel method to improve ADHD diagnosis precision and timeliness by leveraging Deep Learning (DL) approaches and electroencephalogram (EEG) signals. We introduce ADHDeepNet, a DL model that utilizes comprehensive temporal-spatial characterization, attention modules, and explainability techniques optimized for EEG signals. ADHDeepNet integrates feature extraction and refinement processes to enhance ADHD diagnosis. The model was trained and validated on a dataset of 121 participants (61 ADHD, 60 Healthy Controls), employing nested cross-validation for robust performance. The proposed two-stage methodology uses a 10-fold cross-subject validation strategy. Initially, each iteration optimizes the model's hyper-parameters with inner 2-fold cross-validation. Then, Additive Gaussian Noise (AGN) with various standard deviations and magnification levels is applied for data augmentation. ADHDeepNet achieved 100% sensitivity and 99.17% accuracy in classifying ADHD/HC subjects. To clarify model explainability and identify key brain regions and frequency bands for ADHD diagnosis, we analyzed the learned weights and activation patterns of the model's primary layers. Additionally, t-distributed Stochastic Neighbor Embedding (t-SNE) visualized high-dimensional data, aiding in interpreting the model's decisions. This study highlights the potential of DL and EEG in enhancing ADHD diagnosis accuracy and efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。