English

TAMER: A Test-Time Adaptive MoE-Driven Framework for EHR Representation Learning

Machine Learning 2025-03-19 v2

Abstract

We propose TAMER, a Test-time Adaptive MoE-driven framework for Electronic Health Record (EHR) Representation learning. TAMER introduces a framework where a Mixture-of-Experts (MoE) architecture is co-designed with Test-Time Adaptation (TTA) to jointly mitigate the intertwined challenges of patient heterogeneity and distribution shifts in EHR modeling. The MoE focuses on latent patient subgroups through domain-aware expert specialization, while TTA enables real-time adaptation to evolving health status distributions when new patient samples are introduced. Extensive experiments across four real-world EHR datasets demonstrate that TAMER consistently improves predictive performance for both mortality and readmission risk tasks when combined with diverse EHR modeling backbones. TAMER offers a promising approach for dynamic and personalized EHR-based predictions in practical clinical settings.

Keywords

Cite

@article{arxiv.2501.05661,
  title  = {TAMER: A Test-Time Adaptive MoE-Driven Framework for EHR Representation Learning},
  author = {Yinghao Zhu and Xiaochen Zheng and Ahmed Allam and Michael Krauthammer},
  journal= {arXiv preprint arXiv:2501.05661},
  year   = {2025}
}

Comments

8 pages, 3 figures, 7 tables. Code is available at: https://github.com/yhzhu99/TAMER

R2 v1 2026-06-28T21:02:08.620Z