MedSpaformer: A Transferable Transformer with Multi-Granularity Token Sparsification for Medical Time Series Classification

Authors

  • Jiexia Ye The Hong Kong University of Science and Technology (Guangzhou)
  • Weiqi Zhang The Hong Kong University of Science and Technology
  • Ziyue Li Technical University of Munich
  • Jia Li The Hong Kong University of Science and Technology (Guangzhou)
  • Fugee Tsung The Hong Kong University of Science and Technology

DOI:

https://doi.org/10.1609/aaai.v40i33.40001

Abstract

Accurate medical time series (MedTS) classification is essential for effective clinical diagnosis, yet remains challenging due to complex multi-channel temporal dependencies, information redundancy, and label scarcity. While transformer-based models have shown promise in time series analysis, most are designed for forecasting tasks and fail to fully exploit the unique characteristics of MedTS. In this paper, we introduce MedSpaformer, a transformer-based framework tailored for MedTS classification. It incorporates a sparse token-based dual-attention mechanism that enables global context modeling and token sparsification, allowing dynamic feature refinement by focusing on informative tokens while reducing redundancy. This mechanism is integrated into a multi-granularity cross-channel encoding scheme to capture intra- and inter-granularity temporal dependencies and inter-channel correlations, enabling progressive refinement of task-relevant patterns in medical signals. The sparsification design allows our model to flexibly accommodate inputs with variable lengths and channel dimensions. We also introduce an adaptive label encoder to extract label semantics and address cross-dataset label space misalignment. Together, these components enhance the model’s transferability across heterogeneous medical datasets, which helps alleviate the challenge of label scarcity. Our model outperforms 13 baselines across 7 medical datasets under supervised learning. It also excels in few-shot learning and demonstrates zero-shot capability in both in-domain and cross-domain diagnostics. These results highlight MedSpaformer's robustness and its potential as a unified solution for MedTS classification across diverse settings.

Published

2026-03-14

How to Cite

Ye, J., Zhang, W., Li, Z., Li, J., & Tsung, F. (2026). MedSpaformer: A Transferable Transformer with Multi-Granularity Token Sparsification for Medical Time Series Classification. Proceedings of the AAAI Conference on Artificial Intelligence, 40(33), 27791–27799. https://doi.org/10.1609/aaai.v40i33.40001

Issue

Section

AAAI Technical Track on Machine Learning X