LLM2D
轨迹流匹配及其在临床时间序列建模中的应用
Trajectory Flow Matching with Applications to Clinical Time Series Modeling
作者: Xi Zhang, Yuan Pu, Yuki Kawamura, Andrew Loza, Yoshua Bengio, Dennis L. Shung, Alexander Tong
发布日期: 2/5/2025
arXiv ID: oai:arXiv.org:2410.21154v2

摘要

arXiv:2410.21154v2 通知类型: 替换-交叉 摘要:建模随机且不规则采样的时间序列是一个在广泛应用中发现的具有挑战性的问题,特别是在医学领域。神经随机微分方程(Neural SDEs)是解决这个问题的一种有吸引力的建模技术,它使用神经网络参数化随机微分方程(SDE)的漂移和扩散项。然而,当前用于训练Neural SDEs的算法需要通过SDE动力学进行反向传播,这极大地限制了其可扩展性和稳定性。为了解决这个问题,我们提出了一种轨迹流匹配(TFM)的方法,以无模拟的方式训练Neural SDEs,避免了通过动力学进行反向传播。TFM利用生成模型中的流匹配技术来建模时间序列。在这项工作中,我们首先建立了TFM学习时间序列数据所需的条件。接下来,我们提出了一种重新参数化技巧,以提高训练稳定性。最后,我们将TFM适应临床时间序列设置,在绝对性能和不确定性预测方面展示了在三个临床时间序列数据集上的改进表现。