Computational and Statistical Guarantees of the \textit{c}-Rectified flow

2026-08-03Machine Learning

Machine Learning
AI summary

The authors study a method called "c-rectified flow," which improves on previous techniques for moving from one data distribution to another in a way that matches optimal transport theory. They show that normal rectified flow can fail under certain conditions, but their c-rectified flow method always converges to the best solution when some assumptions are met. They also provide mathematical guarantees for how fast this convergence happens and develop improved ways to estimate certain functions needed in this process. Their results include optimal rates for estimating optimal transport in different dimensions.

rectified flowoptimal transportvelocity fieldsgradient projectioncovariance matricesconvergence guaranteesquadratic displacement costscore estimationminimax ratesHölder continuity
Authors
Leda Wang, Zhehao Xu, Qiang Liu, Harrison H. Zhou
Abstract
Recently, rectified flow has emerged as a fundamental framework for large-scale image generation, powering state-of-the-art systems such as FLUX.1 and Stable Diffusion 3. Despite its remarkable empirical success, the computational and statistical guarantees of iterative rectified flow have remained largely unexplored. We address this problem by studying \textit{c}-rectified flow, a cost-aware class of rectified flow that projects velocity fields onto a gradient class while preserving endpoint marginals. The ordinary rectified flow can fail to recover the optimal transport coupling: in a Gaussian case study, the iteration converges to the optimal coupling if and only if the source and target covariance matrices commute. In contrast, under suitable compactness and uniform-integrability assumptions, iterative \textit{c}-rectified flow always converges to the optimal transport coupling. We further establish quantitative one-step contraction and exponential convergence guarantees under projection-stability assumptions for both quadratic and strongly convex displacement costs. Finally, under a Hölder ball assumption, we develop new minimax-optimal score estimation rates and show that, when combined with iterative \textit{c}-rectified flow, they yield a rate-optimal estimator of the optimal transport for the dimension \(d \ge 3\) and a nearly parametric rate for \(d=1,2\).