Abstract
Modeling stochastic and irregularly sampled time series is a challengingproblem found in a wide range of applications, especially in medicine. Neuralstochastic differential equations (Neural SDEs) are an attractive modelingtechnique for this problem, which parameterize the drift and diffusion terms ofan SDE with neural networks. However, current algorithms for training NeuralSDEs require backpropagation through the SDE dynamics, greatly limiting theirscalability and stability. To address this, we propose Trajectory Flow Matching(TFM), which trains a Neural SDE in a simulation-free manner, bypassingbackpropagation through the dynamics. TFM leverages the flow matching techniquefrom generative modeling to model time series. In this work we first establishnecessary conditions for TFM to learn time series data. Next, we present areparameterization trick which improves training stability. Finally, we adaptTFM to the clinical time series setting, demonstrating improved performance onthree clinical time series datasets both in terms of absolute performance anduncertainty prediction.