← Search

Dana Pe'er

3 accepted papers

2025

Wasserstein Flow Matching: Generative Modeling Over Families of Distributions

ICML 2025poster

Generative modeling typically concerns transporting a single source distribution to a target distribution via simple probability flows. However, in fields like computer graphics and single-cell genomics, samples themselves can be viewed as distributions, where standard flow matching ignores their in…

2024

Wasserstein Wormhole: Scalable Optimal Transport Distance with Transformer

ICML 2024poster

Optimal transport (OT) and the related Wasserstein metric ($W$) are powerful and ubiquitous tools for comparing distributions. However, computing pairwise Wasserstein distances rapidly becomes intractable as cohort size grows. An attractive alternative would be to find an embedding space in which pa…

Cited by 19SourcePDFScholar
2023

Gradient Estimation for Binary Latent Variables via Gradient Variance Clipping

AAAI 2023technical

Gradient estimation is often necessary for fitting generative models with discrete latent variables, in contexts such as reinforcement learning and variational autoencoder (VAE) training. The DisARM estimator achieves state of the art gradient variance for Bernoulli latent variable models in many co…

Cited by 2SourcePDFScholar