Interactive companion to Theorems 3.1 & 3.4 of
Zero-Flow Encoders,
Wang, Wang, Liu & Suzuki, ICML 2026
Theorem 3.1 (zero-flow criterion).
Let \(v_t\) be the rectified flow trained with independent coupling between
\(p_{\mathrm{source}}\) and \(p_{\mathrm{target}}\). Then
\(v_{1/2} \equiv 0\) everywhere if and only if
\(p_{\mathrm{source}} = p_{\mathrm{target}}\).
A rectified flow is trained with independent coupling
between the two distributions you choose; particles are then transported from
\(t = 0\) to \(t = 1\) along the learned velocity field. Pick the same
dataset twice and watch everything freeze blue at the midpoint.
particle speed:
0 (still)
→
→
fast
red arrows: learned velocity field \(v_t\)
loading WebAssembly module...
t = 0.00
Implementation. A velocity-field MLP \(u(x, t)\) is trained with Adam on
$$\min_{u}\;\mathbb{E}\,\bigl\|\,X' - X - u(X_t,\, t)\,\bigr\|^{2},
\qquad X_t = t\,X' + (1 - t)\,X,\quad X \perp X',$$
then \(\mathrm{d}X = u\,\mathrm{d}t\) is integrated from \(t = 0\) to \(1\).
Antisymmetry (Theorem 3.2): with identical distributions the field satisfies
\(v_t = -v_{1-t}\) — particles drift, stop dead at the midpoint, then flow
back out.
Theorem 3.4 (conditional zero-flow condition).
For all pairs \((\xi, \eta)\) such that \(f(\xi) = \eta\), and for all
\(z\), the velocity field \(v_{1/2}(z;\, \eta,\, \xi) = 0\) if and only if
\(p_{X|Y} = p_{X|f(Y)}\).
Data follow the joint model \(Y \sim \mathcal{N}(0,1)\),
\(X = 0.5\,Y + \varepsilon\), \(\varepsilon \sim \mathcal{N}(0,1)\). A conditional
rectified flow \(u_t(x;\, f(Y'),\, Y)\) is trained with independent coupling
between \((X, Y)\) and an independent copy \((X', Y')\); each data point
\((Y_i, X_i)\) is then transported vertically along the learned field. The
midpoint field vanishes if and only if the statistic \(f\) is
sufficient for predicting \(X\). Choose \(f\) below: a sufficient
statistic, e.g. \(\sigma(-2y)\), freezes everything blue at
\(t = \tfrac{1}{2}\); a lossy one, e.g. \(\sin(2y)\), does not.
horizontal: conditioning variable \(y\); vertical: \(x\)
particle speed:
0 (still)
→
→
fast
red arrows: learned field \(v_t(z;\, f(\xi),\, \xi)\)
loading WebAssembly module...
t = 0.00
Implementation. A velocity-field MLP \(u_t(x;\, \eta,\, \xi)\) is
trained with Adam on
$$\min_{u}\;\int_{0}^{1}\mathbb{E}\,\bigl\|\,X' - X - u_t(X_t,\, f(Y'),\, Y)\,\bigr\|^{2}\,\mathrm{d}t,
\qquad X_t = t\,X' + (1 - t)\,X,$$
where \((X', Y')\) is an independent copy of \((X, Y)\). Its optimum
\(v_t(z;\, \eta,\, \xi) = \mathbb{E}\bigl[X' - X \mid X_t = z,\ f(Y') = \eta,\ Y = \xi\bigr]\)
transports \(p_{X|Y=\xi}\) to \(p_{X|f(Y)=\eta}\) (Theorem 3.3); the demo
integrates \(\mathrm{d}X = v\,\mathrm{d}t\) for each data point with the
consistent conditioning pair \(\eta = f(\xi)\).