acceptodds
Under review as a conference paper at ICLR 2027

Exact and Inexact Marginal Penalties: When Does the Wasserstein Autoencoder Recover the Wasserstein Distance?

Abstract

Primal optimal transport (OT) minimizes a transport cost over couplings with prescribed marginals. Parameterizing a coupling by a conditional distribution enforces one marginal by construction and leaves the other as an explicit constraint. The Wasserstein autoencoder relaxes this remaining constraint into a divergence penalty. We characterize when the resulting penalized objective recovers the OT cost at a finite coefficient . On bounded domains and under an active constraint, -divergences with differentiable at , as well as squared maximum mean discrepancy, strictly underestimate the cost for every finite coefficient. For satisfying a generalized Pinsker inequality, the optimal-value gap is as . In contrast, the square roots of such -divergences and the 1-Wasserstein distance recover the OT value and global minimizers above explicit finite thresholds, and for square roots of the Kullback-Leibler and chi-square divergences exactness extends to unbounded priors. The dichotomy follows from comparing reconstruction descent with penalty growth in the Wasserstein space of joint distributions. Uniform exactness over a decoder class further yields statistical guarantees for learning the pushforward model. Simulations with computable ground truth illustrate the dichotomy, while CelebA experiments compare latent matching and local image redundancy across penalty classes.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.