arXiv:2603.08339eess.SPcs.AI2026-03

用动力系统特征提升心电图分类,变压器模型表现更优。

Electrocardiogram Classification with Transformers Using Koopman and Wavelet Features

  • 结合科普曼算子与小波变换提取心电信号特征
  • 科普曼特征在四类分类中准确率达98.7%,优于小波特征
  • 适合对时序数据建模感兴趣的科研人员

心电图(ECG)分析对检测心脏异常至关重要,但因生理信号复杂多变,自动化分类仍具挑战。本文研究基于变压器的ECG分类方法,使用科普曼算子和小波变换提取特征。考察两类任务:(1) 二分类(正常 vs. 非正常),(2) 四分类(正常、房颤、室性心律失常、传导阻滞)。采用扩展动态模式分解(EDMD)近似科普曼算子。结果表明,小波特征在二分类中表现更佳;而科普曼特征与变压器结合,在四分类中取得更优性能。简单混合科普曼与小波特征未提升准确率。然而,选择合适的EDMD字典——特别是参数调优的径向基函数字典——带来显著增益,超越仅用小波的基线及混合系统。此外,我们提出基于科普曼的重构分析,提供可解释的动态学习洞察,并与循环神经网络基线对比。总体表明,科普曼特征学习与变压器结合有效,为时序分类中引入动力系统理论指明方向。

原文摘要 · Abstract (English)

Electrocardiogram (ECG) analysis is vital for detecting cardiac abnormalities, yet robust automated classification is challenging due to the complexity and variability of physiological signals. In this work, we investigate transformer-based ECG classification using features derived from the Koopman operator and wavelet transforms. Two tasks are studied: (1) binary classification (Normal vs. Non-normal), and (2) four-class classification (Normal, Atrial Fibrillation, Ventricular Arrhythmia, Block). We use Extended Dynamic Mode Decomposition (EDMD) to approximate the Koopman operator. Our results show that wavelet features excel in binary classification, while Koopman features, when paired with transformers, achieve superior performance in the four-class setting. A simple hybrid of Koopman and wavelet features does not improve accuracy. However, selecting an appropriate EDMD dictionary -- specifically a radial basis function dictionary with tuned parameters -- yields significant gains, surpassing the wavelet-only baseline and the hybrid wavelet-Koopman system. We also present a Koopman-based reconstruction analysis for interpretable insights into the learned dynamics and compare against a recurrent neural network baseline. Overall, our findings demonstrate the effectiveness of Koopman-based feature learning with transformers and highlight promising directions for integrating dynamical systems theory into time-series classification.

心电图分类变压器模型动力系统特征提取

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