arXiv:2509.24118cs.LG2025-09被引 3

融合Mamba与Transformer,高效建模电子病历中的长期时序数据。

HyMaTE: A Hybrid Mamba and Transformer Model for EHR Representation Learning

  • 结合Mamba线性复杂度与Transformer注意力机制,处理长序列病历。
  • 在多个临床数据集上实现更优的预测性能,提升表征丰富性。
  • 模型可解释性强,适合医疗场景下的实际部署与分析。

电子健康记录(EHR)是现代医疗的核心数据,其长期、多变量、稀疏及缺失值特性给传统深度学习带来挑战。尽管基于Transformer的模型在建模EHR方面表现优异,但其二次计算复杂度和有限上下文长度限制了效率与应用。相比之下,状态空间模型(SSMs)如Mamba具备线性时间复杂度,更适合处理长序列,但通常聚焦于序列级信息混合,忽视通道级数据特征。为此,我们提出HyMaTE(一种用于EHR表示学习的混合Mamba与Transformer模型),融合两者优势,构建适用于纵向数据的统一表征。在多个临床数据集上的预测任务验证了该模型能捕捉更有效、更丰富且更细致的EHR表示。同时,自注意力机制带来的可解释性进一步证明其作为可扩展、通用化医疗解决方案的有效性。代码已开源:https://github.com/healthylaife/HyMaTE。

原文摘要 · Abstract (English)

Electronic health Records (EHRs) have become a cornerstone in modern-day healthcare. They are a crucial part for analyzing the progression of patient health; however, their complexity, characterized by long, multivariate sequences, sparsity, and missing values poses significant challenges in traditional deep learning modeling. While Transformer-based models have demonstrated success in modeling EHR data and predicting clinical outcomes, their quadratic computational complexity and limited context length hinder their efficiency and practical applications. On the other hand, State Space Models (SSMs) like Mamba present a promising alternative offering linear-time sequence modeling and improved efficiency for handling long sequences, but focus mostly on mixing sequence-level information rather than channel-level data. To overcome these challenges, we propose HyMaTE (A Hybrid Mamba and Transformer Model for EHR Representation Learning), a novel hybrid model tailored for representing longitudinal data, combining the strengths of SSMs with advanced attention mechanisms. By testing the model on predictive tasks on multiple clinical datasets, we demonstrate HyMaTE's ability to capture an effective, richer, and more nuanced unified representation of EHR data. Additionally, the interpretability of the outcomes achieved by self-attention illustrates the effectiveness of our model as a scalable and generalizable solution for real-world healthcare applications. Codes are available at: https://github.com/healthylaife/HyMaTE.

电子病历混合模型长序列建模医疗AI

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