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: 2410.21154

摘要

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