arXiv:2603.13261cs.AIcs.CV2026-03

3D卷积网络结合时间增广,提升脑电分类准确率

Deep Convolutional Architectures for EEG Classification: A Comparative Study with Temporal Augmentation and Confidence-Based Voting

  • 用3D CNN同时捕捉脑电信号的时空特征
  • 在真实数据上达到0.91的AUC和0.87的平衡准确率
  • 适合做脑机接口的科研人员与工程师参考

脑电图(EEG)分类在脑机接口系统中至关重要,但受信噪比低、神经反应时间变异及数据量少等因素影响,仍具挑战。本文对比了三种深度学习架构对事件相关电位(ERPs)的分类性能。预处理包括带通滤波、空间滤波与归一化。设计并比较三类流程:基于共同空间模式(CSP)的2D CNN、直接输入原始数据的2D CNN,以及联合建模时空特征的3D CNN。为应对ERP潜伏期变化,训练时引入时间偏移增广策略;推理时采用置信度加权的测试时投票机制,提升预测稳定性。在分层五折交叉验证下,尽管CSP能提升2D模型表现,但所提3D CNN在AUC(0.91)和平衡准确率(0.87)上显著优于两个2D变体。结果表明,具备时间感知能力的架构与增广策略对鲁棒的脑电信号分类极为有效。

原文摘要 · Abstract (English)

Electroencephalography (EEG) classification plays a key role in brain-computer interface (BCI) systems, yet it remains challenging due to the low signal-to-noise ratio, temporal variability of neural responses, and limited data availability. In this paper, we present a comparative study of deep learning architectures for classifying event-related potentials (ERPs) in EEG signals. The preprocessing pipeline includes bandpass filtering, spatial filtering, and normalization. We design and compare three main pipelines: a 2D convolutional neural network (CNN) using Common Spatial Pattern (CSP), a second 2D CNN trained directly on raw data for a fair comparison, and a 3D CNN that jointly models spatiotemporal representations. To address ERP latency variations, we introduce a temporal shift augmentation strategy during training. At inference time, we employ a confidence-based test-time voting mechanism to improve prediction stability across shifted trials. An experimental evaluation on a stratified five-fold cross-validation protocol demonstrates that while CSP provides a benefit to the 2D architecture, the proposed 3D CNN significantly outperforms both 2D variants in terms of AUC and balanced accuracy. These findings highlight the effectiveness of temporal-aware architectures and augmentation strategies for robust EEG signal classification.

脑电分类3D卷积时间增广

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