LLM2D
EEGNet:朝着可解释的人工智能在阿尔茨海默病分类中的应用
xEEGNet: Towards Explainable AI in EEG Dementia Classification
作者: Andrea Zanola, Louis Fabrice Tshimanga, Federico Del Pup, Marco Baiesi, Manfredo Atzori
发布日期: 5/1/2025
arXiv ID: oai:arXiv.org:2504.21457v1

摘要

arXiv:2504.21457v1 Announce Type: cross 摘要:本文介绍了xeeegnet,这是一种新颖、紧凑且具有解释性的神经网络,用于分析EEG数据。它完全可解释,并通过主要参数减少来减少过拟合。作为应用场景,我们专注于将阿尔茨海默病和额颞叶痴呆与对照组进行分类。xeeegnet在涉及频谱改变的其他神经学疾病中具有广泛的应用潜力。我们最初使用浅层网络(shallownet),这是一种来自EEGNet家族的简单且流行的模型。对该模型的结构进行了分析,并逐渐进行了修改,以从“黑盒”变为更透明的模型,而不牺牲性能。从临床角度来看,检查了学习到的核和权重以评估医学相关性。包括浅层网络和最终的xeeegnet的模型变体,通过稳健的嵌套的留下N个被试的交叉验证进行了评估,以获得无偏的性能估计。通过嵌入的EEG表示,按类别和组别分组,解释了数据分割间的差异性,并通过成对可分性量化群体差异。通过训练-验证损失相关性和训练速度评估了过拟合。xeeegnet仅使用168个参数,比浅层网络少200倍,但仍保持了解释性,防止了过拟合,实现了可比的中位性能(-1.5%),并且减少了分割间的变异性。这种变异性通过嵌入的EEG表示得到解释:更高的准确性与测试集对照组和阿尔茨海默病病例之间的分离程度增加相关,而训练数据的影响并不显着。xeegnet能够过滤特定的EEG频带、学习频带特异性拓扑结构并使用相关频谱特征,展示了其解释性。尽管大型深度学习模型在性能方面通常被优先考虑,但本研究表明,如 xeegnet 这样的较小架构在EEG病理分类中也可以同样有效。