DRIFT:直接-递归干预条件ICU生理轨迹预测

DRIFT: Direct-Recursive Intervention-Conditioned Forecasting of ICU Physiological Trajectories

精选理由

这篇论文提出了DRIFT,一种ICU生理轨迹预测新方法,在MIMIC-IV和eICU-CRD上比TFT-action的MAP误差低0.67%,尤其在治疗序列变化时表现更准。

AI 摘要

DRIFT是一个混合框架,结合直接预测与递归动作条件校正。在MIMIC-IV(6046次住院)和eICU-CRD(8345次住院)上评估,8/24/48小时平均动脉压(MAP)平均绝对误差相比动作条件TFT(TFT-action)降低0.673%。在处理序列变化的窗口内,DRIFT在8和24小时MAP误差低于TFT-action。序列变化导致DRIFT的MAP误差增加0.21-0.26 mmHg,预测差异主要出现在路径分叉后。

原文 · arXiv cs.LG

DRIFT: Direct-Recursive Intervention-Conditioned Forecasting of ICU Physiological Trajectories

Many time-series forecasts depend not only on prior observations but also on actions specified during the forecast period. In intensive care units (ICUs), future vital signs and laboratory values are influenced by treatments such as vasopressors. However, models that predict the full future sequence all at once make little use of these treatments, whereas autoregressive models can accumulate errors. We introduce DRIFT, a hybrid framework in which a direct model produces the primary forecast and a recursive, action-conditioned model contributes constrained corrections. We evaluate DRIFT on 6,046 admissions from MIMIC-IV and 8,345 admissions from eICU-CRD. Averaged across the 8-, 24-, and 48-hour forecast endpoints, DRIFT reduces mean absolute error for mean arterial pressure (MAP) by 0.673% relative to an action-conditioned Temporal Fusion Transformer (TFT-action) on MIMIC-IV and achieves the lowest corresponding error among the compared models on eICU-CRD. Although the overall accuracy improvement is modest, a MIMIC-IV audit restricted to windows in which the supplied treatment sequence was altered showed that DRIFT achieved lower observed-target MAP error than TFT-action at 8 and 24 hours. Treatment-sequence alteration increased DRIFT's MAP error by 0.21-0.26 mmHg more than it increased TFT-action's error, with prediction changes occurring primarily after the supplied paths diverged. In a separate robustness experiment, the MAP advantage persisted under three shared checkpoint-selection rules emphasizing overall endpoint error, MAP error, or both equally.