arXiv:2412.08873cs.LGcs.AI2024-12被引 3

用Transformer建模健康轨迹演变,实现连续动态预测。

Towards modeling evolving longitudinal health trajectories with a transformer-based deep learning model

  • 修改训练目标与因果注意力掩码,实现时间序列轨迹建模。
  • 在多个常见疾病预测任务中表现媲美双向Transformer。
  • 支持连续预测与早期预警,适合健康监测与回顾分析。

健康登记数据包含丰富的个体健康历史信息。本文聚焦于在包含临床编码、操作和药物购买等编码特征的全国性纵向数据集中,理解个体健康轨迹的演变过程。我们提出一种基于Transformer的深度学习模型,通过调整训练目标并应用因果注意力掩码,实现对个体健康轨迹随时间变化的分析。研究重点是预测未来某时间段内多种常见疾病的发病时间,但不同于仅输出单一预测结果,该方法能在整个预测期内每个时间点上提供连续预测,且结果依赖于当前时刻的状态。实验表明,该模型在基本预测性能上与双向Transformer相当,同时具备出色的轨迹建模能力。我们探索了多种利用该模型分析健康轨迹及辅助早期事件检测的方法,假设其可用于持续监测健康状态、干预正在进行的健康变化,并在回顾性分析中发挥价值。

原文摘要 · Abstract (English)

Health registers contain rich information about individuals' health histories. Here our interest lies in understanding how individuals' health trajectories evolve in a nationwide longitudinal dataset with coded features, such as clinical codes, procedures, and drug purchases. We introduce a straightforward approach for training a Transformer-based deep learning model in a way that lets us analyze how individuals' trajectories change over time. This is achieved by modifying the training objective and by applying a causal attention mask. We focus here on a general task of predicting the onset of a range of common diseases in a given future forecast interval. However, instead of providing a single prediction about diagnoses that could occur in this forecast interval, our approach enable the model to provide continuous predictions at every time point up until, and conditioned on, the time of the forecast period. We find that this model performs comparably to other models, including a bi-directional transformer model, in terms of basic prediction performance while at the same time offering promising trajectory modeling properties. We explore a couple of ways to use this model for analyzing health trajectories and aiding in early detection of events that forecast possible later disease onsets. We hypothesize that this method may be helpful in continuous monitoring of peoples' health trajectories and enabling interventions in ongoing health trajectories, as well as being useful in retrospective analyses.

健康轨迹Transformer时间序列预测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。