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.
@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