arXiv:2502.05494cs.LGcs.AI2025-02被引 6

无需心拍分割,用多尺度掩码自编码器高效检测心电图异常。

Multi-scale Masked Autoencoder for Electrocardiogram Anomaly Detection

  • 通过多尺度掩码与位置嵌入,捕捉心电图的局部和全局特征。
  • 在保持顶尖性能的同时,推理计算量降低至原有1/78。
  • 适合临床部署,无需复杂预处理,轻量高效。

心电图(ECG)分析是诊断心血管疾病的基础工具,但其信号的复杂性和变异性使得异常检测仍具挑战。本文提出多尺度掩码自编码器(MMAE-ECG),一种端到端框架,可有效捕捉ECG数据中的全局与局部依赖关系。不同于依赖心拍分割或R波检测的现有方法,MMAE-ECG无需此类预处理,更适用于临床应用。该方法将ECG信号划分为非重叠段,并为每段分配可学习的位置嵌入;结合新颖的多尺度掩码策略与多尺度注意力机制,辅以独立位置嵌入,使轻量Transformer编码器能同时建模局部与全局信息。被掩码段通过单层Transformer块重建,并在推理阶段采用聚合策略优化输出。实验表明,该方法性能媲美当前最优,而推理浮点运算量(FLOPs)仅约为原方法的1/78。消融实验验证了各组件的有效性,凸显多尺度掩码自编码器在异常检测中的潜力。

原文摘要 · Abstract (English)

Electrocardiogram (ECG) analysis is a fundamental tool for diagnosing cardiovascular conditions, yet anomaly detection in ECG signals remains challenging due to their inherent complexity and variability. We propose Multi-scale Masked Autoencoder for ECG anomaly detection (MMAE-ECG), a novel end-to-end framework that effectively captures both global and local dependencies in ECG data. Unlike state-of-the-art methods that rely on heartbeat segmentation or R-peak detection, MMAE-ECG eliminates the need for such pre-processing steps, enhancing its suitability for clinical deployment. MMAE-ECG partitions ECG signals into non-overlapping segments, with each segment assigned learnable positional embeddings. A novel multi-scale masking strategy and multi-scale attention mechanism, along with distinct positional embeddings, enable a lightweight Transformer encoder to effectively capture both local and global dependencies. The masked segments are then reconstructed using a single-layer Transformer block, with an aggregation strategy employed during inference to refine the outputs. Experimental results demonstrate that our method achieves performance comparable to state-of-the-art approaches while significantly reducing computational complexity-approximately 1/78 of the floating-point operations (FLOPs) required for inference. Ablation studies further validate the effectiveness of each component, highlighting the potential of multi-scale masked autoencoders for anomaly detection.

心电图异常检测自编码器Transformer

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