The Score-Difference Flow for Implicit Generative Modeling

The SD flow defines an optimal path between arbitrary probability distributions and also provides a theoretical link between model classes that individually address the three challenges of the generative modeling trilemma—high sample quality, mode coverage, and fast sampling—thereby setting the stage for a unified approach.

July 2023
Transactions on Machine Learning Research (TMLR) 2023
 

 

Authors

Romann M. Weber (DisneyResearch|Studios) 

 

The Score-Difference Flow for Implicit Generative Modeling

Abstract

Implicit generative modeling (IGM) aims to produce samples of synthetic data matching the characteristics of a target data distribution. Recent work (e.g. score-matching networks, diffusion models) has approached the IGM problem from the perspective of pushing synthetic source data toward the target distribution via dynamical perturbations or flows in the ambient space. In this direction, we present the score difference (SD) between arbitrary target and source distributions as a flow that optimally reduces the Kullback-Leibler divergence between them while also solving the Schrödinger bridge problem. We apply the SD flow to convenient proxy distributions, which are aligned if and only if the original distributions are aligned. We demonstrate the formal equivalence of this formulation to denoising diffusion models under certain conditions. We also show that the training of generative adversarial networks includes a hidden data-optimization sub-problem, which induces the SD flow under certain choices of loss function when the discriminator is optimal. As a result, the SD flow provides a theoretical link between model classes that individually address the three challenges of the generative modeling trilemma—high sample quality, mode coverage, and fast sampling—thereby setting the stage for a unified approach.

Copyright Notice