English

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

Machine Learning 2026-05-26 v1

Abstract

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.

Keywords

Cite

@article{arxiv.2605.24961,
  title  = {MedMamba: Multi-View State Space Models with Adaptive Graph Learning for Medical Time Series Classification},
  author = {Da Zhang and Bingyu Li and Zhiyuan Zhao and Hongyuan Zhang and Junyu Gao and Xuelong Li},
  journal= {arXiv preprint arXiv:2605.24961},
  year   = {2026}
}

Comments

Accepted to 2026 ICML

R2 v1 2026-07-22T07:30:47.931Z