arXiv:2605.24961cs.LG2026-05中稿 · ICML

MedMamba用状态空间模型提升医疗时间序列分类,自动捕捉通道关系和非平稳性。

MedMamba: Multi-View State Space Models with Adaptive Graph Learning for Medical Time Series Classification

论文配图:MedMamba: Multi-View State Space Models with Adaptive Graph Learning for Medical Time Series Classification
图 1 · 摘自论文原文
  • 多尺度卷积+三分支状态空间编码,融合原始、差分与频域信息。
  • 在5个真实数据集上达最新性能,计算复杂度线性增长。
  • 自学习有向图结构,无需预设通道关系,适合临床时序分析。

医疗时间序列对连续监测和及时临床决策至关重要。尽管已有进展,现有方法仍难以同时建模局部-全局动态并应对基线漂移等非平稳性问题,且常忽略潜在通道交互。为此,我们提出MedMamba,一种端到端架构,融合状态空间模型与领域先验知识。首先,采用多尺度卷积嵌入捕获判别性局部形态。其次,为缓解非平稳性,引入三分支差分状态空间编码器,处理原始信号、时间差分及频域视图,融合后突出有效模式并抑制漂移。此外,设计空间图Mamba模块,学习有向依赖结构,并通过稀疏性与无环性正则化,避免预定义图的依赖。在五个真实数据集上的大量实验表明,MedMamba达到当前最优性能,同时保持线性计算复杂度;消融实验验证各组件贡献。代码已开源于https://github.com/zhangda1018/MedMamba。

原文摘要 · Abstract (English)

Medical time series are central to healthcare, enabling continuous monitoring and supporting timely clinical decisions. Despite recent progress, existing methods struggle to jointly model local-global dynamics and handle nonstationarities like baseline drift, while often failing to capture latent channel interactions. To address these challenges, we propose MedMamba, an end-to-end architecture that integrates state space models with domain-specific inductive biases. Specifically, MedMamba first employs multi-scale convolutional embeddings to capture discriminative local morphology. Second, to mitigate nonstationarity, we introduce a tri-branch differential state space encoder that processes raw, temporal-difference, and frequency-domain views, fusing them to emphasize informative patterns while suppressing drift. Furthermore, to uncover latent channel correlations, we design a spatial graph Mamba module that learns a directed dependency structure regularized toward sparsity and acyclicity, which obviates the need for predefined graphs. Extensive experiments on five real-world datasets demonstrate that MedMamba achieves state-of-the-art performance while maintaining linear computational complexity, and ablation studies validate each component's contribution.Code is available at https://github.com/zhangda1018/MedMamba.

医疗时序状态空间图学习分类

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