Computational Optimal Transport I: For Uniform Marginals 1/n Some Permutation Matrix Solves the Kantorovich Problem, so the Relaxation of Optimal Assignment Is TightTextbook
Motivation
An assignment problem pairs each of sources with exactly one of destinations while minimizing a specified cost. A permutation records those pairings. The number of permutations grows as , so a direct search quickly becomes impractical. In discrete optimal transport, the same data can be placed in a linear optimization problem over nonnegative matrices: a matrix entry records how much mass moves from one source to one destination. That formulation permits a source's mass to be split, which gives it many more feasible solutions than the assignment problem. The question for this mission is whether the added flexibility can improve the optimum when every source and destination carries the same mass. The setting and answer are given in Peyré and Cuturi, Computational Optimal Transport, §2.2–2.3.
This capstone is the first in a series based on the textbook. It links a finite combinatorial optimization problem to the matrix model used throughout later chapters. The general transport model is needed again when marginals are unequal, when cost matrices come from distances, and when entropy is added to the objective. The uniform matching case isolates a useful boundary: relaxing an assignment to allow fractional flows expands the feasible set, yet it does not change the best objective value.
Setting
Fix an integer and a real cost matrix . The entry is the cost of sending a unit of mass from source to destination . No sign or metric assumption is imposed on . A permutation assigns source to destination , with every destination used once. The assignment objective of equation (2.2) is
The factor matters: it treats each source as having mass , so the assignment and transport objectives use the same total mass.
A histogram is a vector of masses. The uniform histogram has at each of the indices. A coupling of histograms and is a nonnegative matrix with row sums and column sums . The set of such matrices is
Its linear cost is , and denotes the minimum cost over , as in equations (2.10)–(2.11). For a permutation , the scaled permutation matrix has entry in column of row and zero elsewhere. Consequently is a coupling of with itself, and its transport cost equals . A general member of may distribute a row's mass across several columns.
Formalization targets
Relaxation bound
Because every scaled permutation matrix is feasible for the transport problem, the unrestricted minimum cannot exceed the assignment minimum:
This is the weaker claim in the milestone list. The symmetry of couplings under transposition and the identity supply reusable statements about the model. The list also states the book's scaled form of Birkhoff's extreme point characterization.
Proposition 2.1: tightness for uniform marginals
The goal is the stronger statement: there is an assignment that also minimizes over the full coupling polytope,
This is Proposition 2.1, pp. 372–373. It quantifies over every real cost matrix of the given size. The equality of optimal values follows from the stated common optimizer; the proposition does not claim that every optimal transport plan is a permutation matrix.
Significance
The result gives an exact linear relaxation for uniform assignment. One can optimize over a convex set of matrices without paying an optimal value penalty for allowing fractional mass. The conclusion has a precise limit: it guarantees at least one integral, scaled permutation optimizer, while other optimal couplings can coexist when the cost matrix has ties. When source and destination weights are unequal, the assignment formulation may not even be feasible; the broader coupling model then carries information that a permutation cannot represent.
For formalization, the result fixes a finite transport interface that later missions can compare with their own conventions: nonnegative matrix entries, prescribed row and column sums, and a cost that is linear in the matrix. Mathlib already contains Birkhoff's theorem for doubly stochastic matrices with row sums equal to . This mission's formal target uses the probability normalization , so the scaling is part of the mathematical statement. The textbook result is known; the goal here is a machine checked Lean proof of this precise version and its cited supporting claims. The listed Lean declarations are currently open statements with sorry placeholders.
Difficulty
The evident inequality points in the easy direction: enlarging a feasible set can only reduce a minimum. It gives no reason why a minimizer over the larger set should happen to be a permutation matrix. A generic feasible coupling can split mass between several destinations, and even a unique assignment need not be the only point under consideration in the relaxed problem. The central issue is the geometry of the uniform coupling polytope and its relation to permutation matrices. Its normalization differs from the usual doubly stochastic convention, so a proof using a standard library theorem must retain the factor at every point where a matrix or objective is converted.
Formalization scope
The Lean index types are Fin n and Fin m; they have and elements but are numbered from zero. Histograms are real valued functions on those types, and couplings are real matrices constrained to be nonnegative with exact finite row and column sums. The pairing is a finite double sum. The goal assumes so that the displayed is meaningful and the assignment problem has an index to assign. There is no additional positivity requirement on . The minimum values are represented by real infima only in statements where the uniform coupling polytope and permutation set are nonempty and their finite dimensional objectives are bounded below.
An optimal coupling in the goal is compared with all matrices in , including fractional ones. Restricting that comparison to permutation matrices would erase the proposition's content. The definitions of coupling feasibility, scaled permutation matrices, and the assignment cost are reusable beyond this chapter. Contributions that establish the scaling relation to Mathlib's doubly stochastic matrices, the extreme point characterization in this normalization, or the goal from these components fit the mission. The optional strict inclusion claim on p. 372 is excluded from the milestones because its printed formulation fails at ; it would require a separate statement. The page also prints row and column sums of as despite defining entries ; the Lean statement uses .
Selected references
- G. Peyré and M. Cuturi, Computational Optimal Transport, Foundations and Trends in Machine Learning 11(5–6):355–607, 2019, DOI: 10.1561/2200000073.
- D. Bertsimas and J. N. Tsitsiklis, Introduction to Linear Optimization, Athena Scientific, 1997, Theorem 2.7, publisher record. Cited in the proof of Proposition 2.1.