pith:GDCQ5IT2
One-Step Generative Modeling via Wasserstein Gradient Flows
W-Flow achieves one-step ImageNet 256x256 generation at 1.29 FID by training a neural network to compress a Wasserstein gradient flow.
arxiv:2605.11755 v2 · 2026-05-12 · cs.LG · cs.CV · stat.ML
Add to your LaTeX paper
\usepackage{pith}
\pithnumber{GDCQ5IT2DPZZVVSS6XRUJC7UOL}
Prints a linked badge after your title and injects PDF metadata. Compiles on arXiv. Learn more · Embed verified badge
Record completeness
Claims
W-Flow sets a new state of the art for one-step ImageNet 256×256 generation, achieving 1.29 FID, with improved mode coverage and domain transfer. Compared to multi-step diffusion models with similar FID scores, our method yields approximately 100× faster sampling.
The finite-sample training dynamics converge to the continuous-time distributional dynamics under suitable assumptions. The abstract does not specify what those assumptions are or how restrictive they become for high-dimensional image data.
W-Flow achieves state-of-the-art one-step ImageNet 256x256 generation at 1.29 FID by training a static neural network to follow a Wasserstein gradient flow that minimizes Sinkhorn divergence, delivering roughly 100x faster sampling than comparable multi-step models.
Formal links
Receipt and verification
| First computed | 2026-05-28T01:04:42.241787Z |
|---|---|
| Builder | pith-number-builder-2026-05-17-v1 |
| Signature | Pith Ed25519
(pith-v1-2026-05) · public key |
| Schema | pith-number/v1.0 |
Canonical hash
30c50ea27a1bf39ad652f5e3448bf472de9df646ee9eea9bfaadf0cc6be1a80a
Aliases
· · · · ·Agent API
Verify this Pith Number yourself
curl -sH 'Accept: application/ld+json' https://pith.science/pith/GDCQ5IT2DPZZVVSS6XRUJC7UOL \
| jq -c '.canonical_record' \
| python3 -c "import sys,json,hashlib; b=json.dumps(json.loads(sys.stdin.read()), sort_keys=True, separators=(',',':'), ensure_ascii=False).encode(); print(hashlib.sha256(b).hexdigest())"
# expect: 30c50ea27a1bf39ad652f5e3448bf472de9df646ee9eea9bfaadf0cc6be1a80a
Canonical record JSON
{
"metadata": {
"abstract_canon_sha256": "10abfdef1b415dff3ab1aa5ed9c64aca67dd2a3207110c238a4689c3cfa17cac",
"cross_cats_sorted": [
"cs.CV",
"stat.ML"
],
"license": "http://creativecommons.org/licenses/by/4.0/",
"primary_cat": "cs.LG",
"submitted_at": "2026-05-12T08:29:44Z",
"title_canon_sha256": "13d37a7846389cfcca67d31a522784abbd975f8b5b32ab3b1e014d8df87ba55f"
},
"schema_version": "1.0",
"source": {
"id": "2605.11755",
"kind": "arxiv",
"version": 2
}
}