One-step generation stops drifting on minibatch proportions: swapping hard marginal constraints for soft penalties cuts CIFAR-10 FID from 8.10 to 4.51
UW-Flow replaces W-Flow's balanced Sinkhorn energy with a debiased unbalanced Sinkhorn divergence, reducing finite-batch velocity error and improving generation on MNIST, CIFAR-10, and ImageNet-100 under matched training budgets.