arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.00541cs.LG
收藏

一个损失统治一切:结构化EHR基础模型的标记时间到事件

One Loss to Rule Them All: Marked Time-to-Event for Structured EHR Foundation Models

Department of Computer Science, Columbia University(哥伦比亚大学计算机科学系);Department of Biomedical Informatics, Columbia University(哥伦比亚大学生物医学信息学系);Department of Computational Medicine, UCLA(洛杉矶大学计算医学系);Formation Bio

Zilin Jing, Vincent Jeanselme, Yuta Kobayashi, Simon A. Lee, Chao Pang, Aparajita Kashyap, Yanwei Li, Xinzhuo Jiang, Shalmali Joshi

AI总结:

提出ORA预训练目标,联合建模事件时间和关联测量,相比下一词预测和忽略连续测量的损失,在多个数据集和下游任务上产生更通用的表示,提升回归和时间到事件预测能力。

摘要:

电子健康记录(EHR)中捕获的临床事件是不规则采样的,可能由离散事件和数值测量(如实验室值或治疗剂量)混合组成。EHR的序列性质类似于自然语言,这促使使用下一词预测来训练事件上的EHR基础模型(FM)。然而,这种训练未能捕获EHR的完整结构。必须捕获给定事件发生的时间,但事件值(异常实验室)也会调节其他临床事件的可能性。大多数现有的EHR FM不联合建模这种可能性,无法捕获完整的观察过程,影响下游能力。我们提出ORA,一种标记时间到事件预训练目标,联合建模事件时间和相关测量。在多个数据集、下游任务和模型骨干上,该目标始终比下一词预测和忽略连续测量的预训练损失产生更可泛化的表示。重要的是,所提出的目标在传统分类评估之外带来改进,包括更好的回归和时间到事件预测。除了引入新的FM家族,我们的消融研究提出了更广泛的结论:考虑EHR结构的预训练目标对于扩展下游能力和泛化性至关重要。

英文摘要:

Clinical events captured in Electronic Health Records (EHR) are irregularly sampled and may consist of a mixture of discrete events and numerical measurements, such as laboratory values or treatment dosages. The sequential nature of EHR, analogous to natural language, has motivated the use of next-token prediction to train prior EHR Foundation Models (FMs) over events. However, this training fails to capture the full structure of EHR. When a given event occurs must be captured, but the event value (abnormal lab) also modulates the likelihood of other clinical events. Most existing EHR FMs do not jointly model this likelihood and are unable to capture the full observation process, impacting downstream capabilities. We propose ORA, a marked time-to-event pretraining objective that jointly models event timing and associated measurements. Across multiple datasets, downstream tasks, and model backbones, this objective consistently yields more generalizable representations than next-token prediction and pretraining losses that ignore continuous measurements. Importantly, the proposed objective yields improvements beyond traditional classification evaluation, including better regression and time-to-event prediction. Beyond introducing a new family of FMs, our ablations suggest a broader takeaway: pretraining objectives that account for EHR structure are critical for expanding downstream capabilities and generalizability.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑