Approximate Inference with Wasserstein Gradient Flows

Charlie Frogner, Tomaso A. Poggio

Abstract: We present a novel approximate inference method for diffusion processes, based on the Wasserstein gradient flow formulation of the diffusion. In this formulation, the time-dependent density of the ...