Bassam El Rawas

Guidance Without Labels: Flow Matching from Coupled Gaussian Mixtures

Flow matching models are typically trained to transport a Gaussian source to the data distribution, with source and target samples paired independently. We replace the source with a fitted Gaussian mixture and couple each data point to a mixture component through an entropic optimal transport problem. We show that the...