LLM2D
TFT-multi:重症监护室生命体征轨迹的同步预测
TFT-multi: simultaneous forecasting of vital sign trajectories in the ICU
作者: Rosemary Y. He, Jeffrey N. Chiang
发布日期: 9/25/2024
arXiv ID: oai:arXiv.org:2409.15586v2

摘要

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