Official implementation of the paper "Score-Regularized Joint Sampling with Importance Weights for Flow Matching."
Illustration of importance-weighted non-IID sampling. Under IID sampling, both samples are likely drawn from the same dominant mode. In contrast, diversity velocity encourages samples to diverge along their trajectories, leading to coverage of multiple modes. To correct the resulting sampling bias, importance weights are required. Intuitively, w_1 > 1 > w_2, since non-IID sampling draws the second sample from a minor mode.
Code coming soon!