Diffeomorphic Measure Matching with Kernels for Generative Modeling

Biraj Pandey,Bamdad Hosseini,Pau Batlle,Houman Owhadi
2024-02-13
Abstract:This article presents a general framework for the transport of probability measures towards minimum divergence generative modeling and sampling using ordinary differential equations (ODEs) and Reproducing Kernel Hilbert Spaces (RKHSs), inspired by ideas from diffeomorphic matching and image registration. A theoretical analysis of the proposed method is presented, giving a priori error bounds in terms of the complexity of the model, the number of samples in the training set, and model misspecification. An extensive suite of numerical experiments further highlights the properties, strengths, and weaknesses of the method and extends its applicability to other tasks, such as conditional simulation and inference.
Machine Learning,Dynamical Systems,Computation
What problem does this paper attempt to address?
This paper presents a probabilistic metric matching framework based on differential equations and Reproducing Kernel Hilbert Space (RKHS) for generative modeling and sampling. It is inspired by differential isomorphism matching in image registration and finds the optimal transport map by solving an optimization problem to minimize the divergence. The paper provides theoretical analysis and gives prior error bounds on model complexity, training sample size, and model mismatch. The characteristics, advantages, and limitations of the method are demonstrated through numerical experiments. The method is applicable to conditional simulation and inference tasks.