LLM2D
TFT-multi:重症监护病房中生命体征轨迹的同步预测
TFT-multi: simultaneous forecasting of vital sign trajectories in the ICU
发布日期: 9/25/2024
arXiv ID: oai:arXiv.org:2409.15586v1

摘要

医疗数据中的轨迹预测一直是精准医疗和临床整合中计算方法研究的重要领域。近年来,生成式 AI 模型在捕捉时间序列数据中的短期和长期依赖关系方面展现出了可喜的成果。尽管这些模型也已应用于医疗保健领域,但大多数模型一次只预测一个值,这在临床环境中是不现实的,因为在临床环境中,往往会同时采集多个指标。在本研究中,我们扩展了时间融合转换器 (TFT) 的框架,这是一种多时域时间序列预测工具,并提出了 TFT-multi,一个能够同时预测多个生命体征轨迹的端到端框架。我们将 TFT-multi 应用于预测重症监护病房记录的 5 种生命体征:血压、脉搏、血氧饱和度、体温以及呼吸频率。我们假设通过联合预测这些通常相互关联的指标,我们可以做出更准确的预测,特别是在缺失率较高的变量中。我们使用公共 MIMIC 数据集和一个独立的机构数据集验证了我们的模型,并证明这种方法优于最先进的单变量预测工具,包括原始 TFT 和 Prophet,以及用于多变量预测的向量回归建模。此外,我们通过将我们的管道应用于预测对实际和假设的升压剂给药的反应的血压变化,进行了案例分析研究。