arXiv:2504.21457cs.LGcs.AI2025-04被引 12

小模型也能解释:EEG痴呆分类新模型,性能不输大模型

xEEGNet: Towards Explainable AI in EEG Dementia Classification

  • 设计轻量可解释网络,参数减少200倍,避免过拟合
  • 仅168参数实现与大模型相当的分类准确率(中位差-1.5%)
  • 适合临床研究者用,能看懂哪些脑波特征在起作用

本文提出xEEGNet,一种新型、紧凑且可解释的脑电图(EEG)分析神经网络。该模型通过大幅减少参数(仅168个,为ShallowNet的1/200),在保持性能的同时提升透明度,有效缓解过拟合。以阿尔茨海默病和额颞叶痴呆与健康对照的分类为例,采用嵌套留人交叉验证评估模型。结果显示,xEEGNet在中位性能上仅比基准低1.5%,且跨数据集的变异更小。通过嵌入表示分析发现,分类准确率与测试集控制组和阿尔茨海默病组的分离度正相关,训练数据影响不大。模型能筛选特定脑电频段,学习频段特异性拓扑图,利用关键谱特征,具备临床可解释性。研究表明,小型模型在癫痫等神经疾病分类中同样高效。

原文摘要 · Abstract (English)

This work presents xEEGNet, a novel, compact, and explainable neural network for EEG data analysis. It is fully interpretable and reduces overfitting through major parameter reduction. As an applicative use case, we focused on classifying common dementia conditions, Alzheimer's and frontotemporal dementia, versus controls. xEEGNet is broadly applicable to other neurological conditions involving spectral alterations. We initially used ShallowNet, a simple and popular model from the EEGNet-family. Its structure was analyzed and gradually modified to move from a "black box" to a more transparent model, without compromising performance. The learned kernels and weights were examined from a clinical standpoint to assess medical relevance. Model variants, including ShallowNet and the final xEEGNet, were evaluated using robust Nested-Leave-N-Subjects-Out cross-validation for unbiased performance estimates. Variability across data splits was explained using embedded EEG representations, grouped by class and set, with pairwise separability to quantify group distinction. Overfitting was assessed through training-validation loss correlation and training speed. xEEGNet uses only 168 parameters, 200 times fewer than ShallowNet, yet retains interpretability, resists overfitting, achieves comparable median performance (-1.5%), and reduces variability across splits. This variability is explained by embedded EEG representations: higher accuracy correlates with greater separation between test set controls and Alzheimer's cases, without significant influence from training data. xEEGNet's ability to filter specific EEG bands, learn band-specific topographies, and use relevant spectral features demonstrates its interpretability. While large deep learning models are often prioritized for performance, this study shows smaller architectures like xEEGNet can be equally effective in EEG pathology classification.

EEG可解释性痴呆小模型

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