COT-FM: Cluster-wise Optimal Transport Flow Matching

Chiensheng Chiang, TU, Jiawei, Cheng-Fu Chou, Tsung-Wei Ke

COT-FM: Cluster-wise Optimal Transport Flow Matching: 2 upvotes on Hugging Face Daily Papers, #28 of 32 papers on 2026-03-20. Day-by-day upvote history.

We introduce COT-FM, a general framework that reshapes the probability path in Flow Matching (FM) to achieve faster and more reliable generation. FM models often produce curved trajectories due to random or batchwise couplings, which increase discretization error and reduce sample quality. COT-FM fixes this by clustering target samples and assigning each cluster a dedicated source distribution obtained by reversing pretrained FM models. This divide-and-conquer strategy yields more accurate local transport and significantly straighter vector fields, all without changing the model architecture. As a plug-and-play approach, COT-FM consistently accelerates sampling and improves generation quality across 2D datasets, image generation benchmarks, and robotic manipulation tasks.

Paper page on Hugging Face · arXiv

Data: hysts-bot-data/daily-papers-stats and the Daily Papers API. Open data: tardellirs/paper-pulse-data. Sister project: Model Pulse, the download history of every model on the Hub.