arXiv:2501.02949cs.LGeess.SP2025-01被引 9

轻量级神经网络精准识别睡眠阶段,参数仅1万,适合边缘设备部署。

MSA-CNN: A Lightweight Multi-Scale CNN with Attention for Sleep Stage Classification

  • 用多尺度互补池化减少冗余参数,分离时序与空间特征提取
  • 在三个公开数据集上准确率和一致性系数均超越九个主流模型
  • 结构模拟人类专家判读逻辑,代码开源便于复现

机器学习与开放数据推动了自动睡眠分期研究。尽管分类模型众多,但多数未关注模型复杂度,而这对实际应用至关重要。本文提出轻量级多尺度注意力卷积神经网络(MSA-CNN),参数量约10,000。其创新多尺度模块通过互补池化消除冗余滤波器参数,并采用密集卷积;通过分离时间与空间特征提取,结合低成本全局空间卷积进一步压缩模型复杂度。该设计不仅降低计算负担,还模仿人类专家的分期策略。我们在三个公共数据集上评估了MSA-CNN的小型与大型版本,分别对比九个先进基线模型,采用重复交叉验证并重评所有基线。结果表明,大尺寸MSA-CNN在所有数据集上的准确率与Cohen's kappa均优于所有基线,且参数显著减少。最后,我们分析了不同变体及核心模块机制,深入揭示其工作原理。代码与评估流程已公开于https://github.com/sgoerttler/MSA-CNN。

原文摘要 · Abstract (English)

Recent advancements in machine learning-based signal analysis, coupled with open data initiatives, have fuelled efforts in automatic sleep stage classification. Despite the proliferation of classification models, few have prioritised reducing model complexity, which is a crucial factor for practical applications. In this work, we introduce Multi-Scale and Attention Convolutional Neural Network (MSA-CNN), a lightweight architecture featuring as few as ~10,000 parameters. MSA-CNN leverages a novel multi-scale module employing complementary pooling to eliminate redundant filter parameters and dense convolutions. Model complexity is further reduced by separating temporal and spatial feature extraction and using cost-effective global spatial convolutions. This separation of tasks not only reduces model complexity but also mirrors the approach used by human experts in sleep stage scoring. We evaluated both small and large configurations of MSA-CNN against nine state-of-the-art baseline models across three public datasets, treating univariate and multivariate models separately. Our evaluation, based on repeated cross-validation and re-evaluation of all baseline models, demonstrated that the large MSA-CNN outperformed all baseline models on all three datasets in terms of accuracy and Cohen's kappa, despite its significantly reduced parameter count. Lastly, we explored various model variants and conducted an in-depth analysis of the key modules and techniques, providing deeper insights into the underlying mechanisms. The code for our models, baselines, and evaluation procedures is available at https://github.com/sgoerttler/MSA-CNN.

睡眠分期轻量模型注意力机制卷积网络

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