arXiv:2606.21973cs.LGcs.AI2026-06中稿 · IJCAI

用单令牌压缩重建法提升生理信号自监督学习效果

SPOTR: Spatio-temporal Pooling One-Token Reconstruction for Universal Physiological Signal Self-supervised Learning

论文配图:SPOTR: Spatio-temporal Pooling One-Token Reconstruction for Universal Physiological Signal Self-supervised Learning
图 1 · 摘自论文原文
  • 将生理信号压缩为单令牌,仅凭该表示重构原始信号
  • 在4类生理数据上线性探测平均AUC提升超17%
  • 计算与内存开销大幅降低,适合临床实时应用

脑电、心电和脉搏波等生理信号广泛用于临床监测。现有自监督学习方法虽能利用无标签数据,但在异构数据集上仍表现不佳,常扭曲临床有意义结构或依赖时间与跨通道冗余。这导致其在线性探测(轻量适配)下性能有限,而此设置更贴近真实医疗场景。此外,多数基于Transformer的模型对时空标记序列进行展平编码,计算与内存开销高,且多局限于单一模态。为此,我们提出SPOTR(时空池化单令牌重构),一种压缩-重建预训练框架,引入单令牌全局瓶颈。该方法将每条波形压缩为单令牌表示,并仅基于此表示重构信号。同时,设计高效时空压缩模块以降低计算与内存成本。在涵盖EEG、iEEG、ECG和PPG的20个数据集上预训练后,SPOTR在所有数据集上的线性探测表现均优于最强基线,平均AUC分别提升18.49%、21.71%、17.86%和4.64%。相比代表性通用时间序列基础模型,其平均延迟降低78%,峰值GPU内存降低52%。代码见:https://github.com/5GYYYYY/SPOTR。

原文摘要 · Abstract (English)

Physiological signals such as EEG, ECG, and PPG are widely used in clinical monitoring. Recent self-supervised learning (SSL) methods offer an attractive way to leverage unlabeled recordings, yet they still fall short in practice. In particular, current SSL methods struggle across heterogeneous datasets, often distorting clinically meaningful structures or learning shortcuts from temporal and cross-channel redundancy. Consequently, existing SSL methods often deliver limited performance under linear probing, a lightweight adaptation setting that better matches real-world medical scenarios. Moreover, most Transformer-based SSL models encode a flattened spatiotemporal token sequence, incurring high computation and memory cost, and are typically developed within a single modality. To address these limitations, we present SPOTR (Spatio-temporal Pooling One-Token Reconstruction), a compress-reconstruct pretraining framework that introduces a single-token global bottleneck for physiological signals. SPOTR compresses each waveform into a single-token representation and reconstructs the signal conditioned only on this representation. Meanwhile, SPOTR introduces an efficient spatio-temporal compaction module to reduce computation and memory cost. Pretrained on 20 datasets spanning EEG, iEEG, ECG, and PPG, SPOTR consistently outperforms the strongest baseline under linear probing, improving average AUC by 18.49%, 21.71%, 17.86%, and 4.64%, respectively. Compared with a representative general-purpose time-series foundation model, SPOTR achieves around 78% lower latency and 52% lower peak GPU memory on average. The code can be found at https://github.com/5GYYYYY/SPOTR.

自监督学习生理信号时空建模轻量化

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