1: \begin{abstract}

2: Time series imputation is important for numerous real-world applications.

3: To overcome the limitations of diffusion model-based imputation methods, e.g., slow convergence in inference, we propose a novel method for time series imputation in this work, called Conditional Lagrangian Wasserstein Flow.

4: The proposed method leverages the (conditional) optimal transport theory to learn the probability flow in a simulation-free manner, in which the initial noise, missing data, and observations are treated as the source distribution, target distribution, and conditional information, respectively.

5: According to the principle of least action in Lagrangian mechanics, we learn the velocity by minimizing the corresponding kinetic energy.

6: Moreover, to incorporate more prior information into the model, we parameterize the derivative of a task-specific potential function via a variational autoencoder, and combine it with the base estimator to formulate a Rao-Blackwellized sampler.

7: The propose model allows us to take less intermediate steps to produce high-quality samples for inference compared to existing diffusion methods.  

8: Finally, the experimental results on the real-word datasets show that the proposed method achieves competitive performance on time series imputation compared to the state-of-the-art methods. 

9: \end{abstract}

10: