arXiv:2409.01962eess.SPcs.CV2024-09中稿 · publication in IEE…

用注意力膨胀卷积自动分睡眠阶段,准确率超99%。

Attentive Dilated Convolution for Automatic Sleep Staging using Force-directed Layout

  • 通过力导向布局提取脑电时空特征,捕捉关键信息。
  • 在三个数据集上准确率达98.56%~99.66%,参数仅140万。
  • 适合需要高效高精度睡眠分析的临床与研究场景。

睡眠阶段对识别睡眠模式和诊断睡眠障碍至关重要。本文提出一种名为注意力膨胀卷积神经网络(AttDiCNN)的自动化睡眠分期分类器,采用深度学习方法应对数据异质性、计算复杂度高及可靠自动分期的挑战。基于可视图的力导向布局用于从脑电信号中捕获最具代表性信息,以表征时空特征。所提网络包含三个模块:局部空间特征提取网络(LSFE)、时空长程保留网络(S2TLR)和全局平均注意力网络(G2A)。LSFE提取睡眠数据的空间信息,S2TLR用于捕捉长期上下文中的关键信息,G2A通过聚合LSFE与S2TLR的信息降低计算开销。在三个公开可获取的数据集EDFX、HMC和NCH上评估,准确率分别达到98.56%、99.66%和99.08%,同时保持低计算复杂度(1.4 M参数)。该架构在多项性能指标上超越现有方法,证明其在临床环境中作为自动化工具的潜力。

原文摘要 · Abstract (English)

Sleep stages play an important role in identifying sleep patterns and diagnosing sleep disorders. In this study, we present an automated sleep stage classifier called the Attentive Dilated Convolutional Neural Network (AttDiCNN), which uses deep learning methodologies to address challenges related to data heterogeneity, computational complexity, and reliable and automatic sleep staging. We employed a force-directed layout based on the visibility graph to capture the most significant information from the EEG signals, thereby representing the spatial-temporal features. The proposed network consists of three modules: the Localized Spatial Feature Extraction Network (LSFE), Spatio-Temporal-Temporal Long Retention Network (S2TLR), and Global Averaging Attention Network (G2A). The LSFE captures spatial information from sleep data, the S2TLR is designed to extract the most pertinent information in long-term contexts, and the G2A reduces computational overhead by aggregating information from the LSFE and S2TLR. We evaluated the performance of our model on three comprehensive and publicly accessible datasets, achieving state-of-the-art accuracies of 98.56%, 99.66%, and 99.08% for the EDFX, HMC, and NCH datasets, respectively, while maintaining a low computational complexity with 1.4 M parameters. Our proposed architecture surpasses existing methodologies in several performance metrics, thereby proving its potential as an automated tool for clinical settings.

睡眠分期脑电分析深度学习注意力机制

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