Bridge Graphical Models

From latent-path design to causal sampler representations: the broader research program, the precise Markovization layer studied in the paper, and what the experiments actually establish.
Author: Tiantian ZHANGPost-rebuttal research exposition
BGM is the framework; the Markovization gap studies one compression interface inside it.
The research story

1. From latent codes to latent paths

The broad idea behind Bridge Graphical Models is not merely that a velocity predictor has nonzero error. It is that continuous-time generative models contain a latent path graphical model, and a sampler must turn that endpoint-informed latent object into a causal computation.

VAE

Choose a latent code and a decoder. The central question is whether the code preserves the information needed to reconstruct data while remaining compatible with a prior.

Diffusion / flow

The latent object becomes a trajectory. A Gaussian corruption, straight interpolation, Brownian bridge, or field line specifies how information is distributed along the path.

BGM

Treat the endpoint coupling, bridge law, target, information state, clock, and dynamics representation as distinct design choices.

\[ \text{latent code}\;\longrightarrow\;\text{latent path}\;\longrightarrow\;\text{causal sampler representation}. \]
Hierarchy after the rebuttal. BGM is the broader research program. The present paper studies one universal interface inside it: the compression of an endpoint-conditioned bridge target into the information available to a non-anticipative sampler.

This hierarchy is important. The path-space VAE interpretation, probability-current gauge, and augmented field geometry belong to the ambient BGM program. The Markovization gap is the first sharply measurable obstruction inside that program; it does not by itself complete the whole unification.

Claim audit

2. What the earlier wording undersold—and oversold

It undersold the idea

Calling the contribution only an “\(L_2\) regression error” hides the path-space ontology: a non-causal teacher bridge is being compressed into a causal information state.

It oversold “velocity”

For a deterministic path, \(U_t=\partial_tX_t\) is a velocity. For a stochastic bridge, drift, current velocity, control drift, score target, and flow target are different objects and define different gaps.

It oversold prediction

A lower image proxy does not universally imply better FID, KID, or sample quality. It is evidence about absolute fixed-budget Markov-regression difficulty.

It hid the observation map

The sampler need not observe only \((X_t,t)\). It may receive memory, a latent coordinate, history, or an augmented field state. The gap is relative to that information interface.

We formulate these models as endpoint-conditioned latent path processes and study the compression from non-causal bridge information to causal sampler representations. The Markovization gap quantifies the target-, observation-, and clock-relative irreducible squared-loss uncertainty of this projection.

Here “non-causal” does not mean a physical violation of causality. It means that the training target may depend on a terminal endpoint or bridge randomness that is unavailable to the sampler at generation time. “Causal sampler representation” means any state constructed from information that the sampler is permitted to use.

Formal object

3. The Markovization gap after the rebuttal

Let \(L\sim\mathscr Q\) denote the full bridge record: endpoints, bridge randomness, latent or augmented coordinates, and any relevant history. The training objective selects a local target \(U_t=u_t(L)\). The sampler receives only

\[ S_t=[\mathcal O_U(L)]_t=\mathcal O_{U,t}(L_{\le t}). \]

For a training-time law \(\mu\), define

\[ \boxed{ \mathfrak G_\mu(\mathscr Q,U,\mathcal O_U) = \int \mathbb E_{\mathscr Q}\!\left[ \operatorname{Tr}\operatorname{Var}(U_t\mid S_t,t) \right]\,\mu(dt) }. \]
ArgumentWhat it fixesWhy it matters
\(\mathscr Q\)Endpoint coupling and bridge/path lawChanges which hidden paths can pass through the same visible state.
\(U\)The exact target selected by the objectiveDrift-, current-, score-, and flow-target gaps need not agree.
\(\mathcal O_U\)Sampler-visible informationThe same path law can have zero gap in a full state and positive gap after projection.
\(\mu\)Training-time sampling law / clockTime reparameterization changes the weighting and scale of the regression problem.

For squared loss, the Bayes decoder is

\[ v^\star(s,t)=\mathbb E_{\mathscr Q}[U_t\mid S_t=s,t], \]

and the classical Hilbert-space projection identity gives

\[ \int\mathbb E\|U_t-v(S_t,t)\|_2^2\,\mu(dt) = \mathfrak G_\mu(\mathscr Q,U,\mathcal O_U) + \int\mathbb E\|v^\star(S_t,t)-v(S_t,t)\|_2^2\,\mu(dt). \]
Novelty boundary. The \(L_2\) projection theorem is classical. The contribution is to identify the correct bridge-to-sampler projection problem, expose its target/information/clock dependence, and turn the residual into a generative-design diagnostic.

If a richer state \(S'_t\) contains at least as much information as \(S_t\), then conditioning reduces variance:

\[ \sigma(S_t)\subseteq\sigma(S'_t) \quad\Longrightarrow\quad \mathfrak G_\mu(\mathscr Q,U,\mathcal O'_U) \le \mathfrak G_\mu(\mathscr Q,U,\mathcal O_U). \]

Memory or augmented latent state can therefore reduce, and sometimes eliminate, the gap. It is an information-relative bottleneck, not an impossibility result for every stateful generator.

Exact population experiment

4. Why a 3D Gaussian experiment is still an \(L_2\) regression problem

The words “3D Gaussian” and “\(L_2\) regression” describe different aspects of the same experiment:

3D describes the random variables

The endpoint, state, and target live in \(\mathbb R^3\). The coupling specifies their joint Gaussian law.

\(L_2\) describes the loss

The decoder predicts a vector \(U_t\in\mathbb R^3\), and the loss sums squared coordinate errors.

\[ \|U_t-f(S_t,t)\|_2^2 =\sum_{j=1}^3\bigl(U_{t,j}-f_j(S_t,t)\bigr)^2, \qquad \inf_f\mathbb E\|U_t-f(S_t,t)\|_2^2 =\mathbb E\operatorname{Tr}\operatorname{Var}(U_t\mid S_t,t). \]

The coupling changes the distribution under which this regression is evaluated. The observation map changes how much of that distribution the decoder can see. Neither changes the fact that the measured floor is squared \(L_2\) error.

The decisive observation control

The same straight Monge bridge keeps endpoint cost and total target variance fixed. Only the sampler-visible state changes.

Visible stateEndpoint costVar\((U)\)Exact gapOracle MSEMLP MSE
Full 3D1.2221.2220.0000.0000.000004
2D projection1.2221.2220.2270.2260.226
1D projection1.2221.2220.5750.5770.577

The value \(4\times10^{-6}\) is not a new nonzero gap; it is numerical/optimization residue in a case whose analytic floor is exactly zero. The substantive result is the transition \(0\to0.227\to0.575\) caused solely by information removal.

Across 11 coupling, bridge, and observation configurations, exact gap versus MLP test MSE has Spearman and Pearson correlation \(1.0000\) to four decimals; learned error is within about 2% of the analytic floor.

Endpoint-conditioned paths crossing at a visible state
Crossing paths illustrate the mechanism: different hidden bridge records can produce the same visible state but different local targets. Projection forces the decoder to average them.
Not yet a thermodynamic phase transition. The full-to-projected experiment is a finite-dimensional sufficient-information threshold. A genuine phase-transition theory would require a representation-capacity parameter and an asymptotic nonanalytic limit.
First-class design variable

5. Coupling changes the regression problem before training

The exact Gaussian experiment fixes the endpoint marginals and gradually degrades their cross-covariance. This turns the original two-point OT-versus-independent comparison into a graded intervention.

CouplingEndpoint costStraight gapBrownian-target gap
Monge1.2220.0003.882
Noisy OT 0.31.4060.1824.072
Noisy OT 0.61.9880.7494.666
Noisy OT 1.03.6222.2706.272
Independent10.5007.48411.969

For the straight target, the exact floor follows \(0\to0.182\to0.749\to2.270\to7.484\). The same endpoint marginals therefore induce very different Markov-decoder problems depending on how endpoints are paired.

Fashion-MNIST coupling continuum

One time-conditioned decoder architecture, optimizer, training budget, and three seeds are held fixed. Starting from minibatch OT, progressively larger fractions of the assignment are scrambled before reaching independence; Brownian OT changes bridge target at the original OT endpoint cost.

DesignValidation sum-MSECross-fitted proxyVar\((U)\)Endpoint cost
Straight OT140.62 ± 0.55877.81985.491247.94
Scramble 25%152.61 ± 0.48891.661002.681265.58
Scramble 50%163.64 ± 0.52906.831021.461282.99
Scramble 75%173.95 ± 0.92920.431039.381300.70
Independent183.70 ± 0.60933.711058.821318.34
Brownian OT250.23 ± 0.511791.571866.941247.94

The proxy and validation MSE have Spearman correlation \(1.000\) over all six designs and over the five straight couplings alone. Brownian and straight OT share endpoint cost, yet their validation errors differ sharply, so endpoint cost alone cannot rank bridge targets.

Qualification. Target variance is also strongly correlated with image-space error. The image experiment therefore supports prediction of absolute regression difficulty; the exact fixed-variance observation control supplies the clean separation from scale.
From population object to screening score

6. What the practical proxy does—and does not estimate

The population conditional variance is generally unavailable on images. The primary practical score is cross-fitted conditional-mean error:

\[ \widehat{\mathfrak G}_{\mathrm{CF}} =\frac{1}{N_{\mathrm{te}}} \sum_{i=1}^{N_{\mathrm{te}}} \left\|U_i-\widehat{\mathbb E}[U_i\mid S_i,t_i]\right\|_2^2. \]

This is a finite-sample ranking proxy. It contains conditional-mean estimation error and is not an unbiased estimator, a certified lower bound, or a number that should be compared across unrelated feature pipelines.

Calibration against known exact gaps

A 32D Gaussian family with known population gaps varies \(N\in\{2000,5000,10000\}\), \(k\in\{16,32,64\}\), PCA dimension \(\{8,16,32\}\), seven bridge/coupling configurations, and three seeds. At \(N=10000\), the median exact-versus-proxy Spearman correlation is \(1.000\).

Image estimatorMean rank stabilityMinimumMain role
Cross-fitted \(k\)-NN MSE0.9930.983Primary proxy
Local variance, varying \(k\)0.9590.900Robustness only
Local variance, varying PCA0.7630.617Representation-sensitive

Cross-fitted error is preferred because it directly asks the operational question—how well can a predictor trained on other samples recover the target at a held-out state? Local variance additionally mixes conditional uncertainty with variation of the conditional mean across a finite neighborhood.

Stress test and negative controls

7. Broader bridge families: strong signal, imperfect oracle

The CIFAR-10 experiment holds architecture and optimization budget fixed while varying deterministic schedules, a VP-cosine path, Brownian targets at three noise levels, independent coupling, and a VE corruption control. The seven principal OT-coupled rows have identical endpoint cost \(3738.17\).

Bridge designValidation sum-MSECross-fitted proxyVar\((U)\)
Linear464.56 ± 1.603445.493725.90
Brownian \(\epsilon=0.15\)483.82 ± 5.953605.083882.34
Brownian \(\epsilon=0.35\)566.83 ± 3.384315.794578.83
Power-2 schedule582.57 ± 6.544350.204743.42
Smoothstep632.11 ± 8.314627.974943.76
VP cosine675.80 ± 8.074347.114640.20
Brownian \(\epsilon=0.60\)728.28 ± 11.765999.546228.52

Across these seven designs, proxy versus validation MSE has Spearman \(0.893\) and Pearson \(0.897\). Within the Brownian-noise family, both increase monotonically. The fixed endpoint cost rejects transport cost as a complete explanation.

Informative failure. VP cosine has a lower proxy than smoothstep but a higher validation error. Closely related clocks can therefore be reordered by optimization, target weighting, approximation, or numerical conditioning. Total target variance is comparably predictive in this sweep, and normalized \(\eta\) is not reliable.

CIFAR-100: lower regression difficulty does not guarantee better samples

CouplingProxyTrain lossFIDKIDIS
Independent115.610.162547.160.03505.97
Minibatch OT98.860.141848.600.03785.79

Minibatch OT lowers the proxy and training loss, but it does not improve FID, KID, or IS in this pilot. This is a direct boundary case for the claim. The retained pilot summary has no seed-level uncertainty, so it is not presented as a powered benchmark.

The larger BGM program

8. What lies outside the Markovization layer

The original research vision remains broader than the present theorem. A useful stack is

\[ (\Gamma,\mathsf R,\mathcal M,a) \longrightarrow L\sim\mathscr Q \longrightarrow U_t \xrightarrow{\mathcal O_U} S_t \xrightarrow{\text{projection}} v^\star \longrightarrow P_\theta. \]

Path-space VAE

Coupling and bridge can be read as an inference path; a causal path law acts as decoder. A complete joint ELBO theory is a broader layer, not proved by the gap identity.

Probability-current gauge

ODE and SDE representations may realize the same marginal current under standard Fokker–Planck relations. This organizes dynamics representation but is distinct from the information projection.

Field geometry

Poisson or electrostatic constructions live in augmented state spaces. Their branch or field coordinate can change \(\mathcal O_U\) and therefore the gap.

Representation threshold

A future rate–distortion theory could ask how much memory or latent capacity is required before the gap vanishes. The present paper fixes the information state.

Observed and augmented field-line states
Field-line example from the original exposition. The conceptual point is now stated observation-relatively: projected data-space state may lose branch identity, whereas augmented state can retain it.
BGM is the framework; Markovization is the interface studied here. Keeping this distinction explicit preserves the depth of the original program without pretending that one projection theorem proves the entire path-space VAE, gauge, and field-geometry theory.
Cost of diagnosis

9. Computational complexity

For \(N_{\mathrm{tr}}\) reference samples, \(N_{\mathrm{te}}\) held-out samples, feature dimension \(r\), and \(M\) sampled times, brute-force cross-fitted \(k\)-NN computes every train–test distance at every time:

\[ O\!\left(MN_{\mathrm{tr}}N_{\mathrm{te}}r\right)\quad\text{time}. \]

Storing all distances would require \(O(N_{\mathrm{tr}}N_{\mathrm{te}})\) memory. Query chunks of size \(b\) reduce working distance memory to \(O(bN_{\mathrm{tr}})\). Projecting \(N\) original \(d\)-dimensional features into \(r\) dimensions costs \(O(Ndr)\) after the PCA basis is fitted.

Coupling construction is separate. Independent pairing is \(O(Nd)\). Exact Hungarian matching within batches of size \(B\) costs \(O(B^3)\) per batch, hence \(O(NB^2)\) over \(N/B\) batches, with \(O(B^2)\) memory. A learned coupling or feature encoder adds its own training and inference cost.

Completed blockProxy / analysisDownstream neural work
Exact 3D analytic grid3.1 s—
Exact 3D neural verification—366.7 s
Gaussian proxy calibration14.6 s—
Fashion-MNIST14.1 s3429.5 s (18 runs)
CIFAR-10 bridge sweep49.2 s11925.3 s (27 runs)
Evidence boundary

10. What the current evidence supports

ClaimStatusEvidence or qualification
Exact gap is the irreducible squared-loss decoder floor.SupportedAnalytic covariance, Gaussian oracle, and MLP agree across 11 configurations.
Coupling, bridge target, and visible information change the floor.SupportedNoisy-coupling continuum, straight/Brownian control, and full/2D/1D ablation.
Cross-fitted proxy preserves exact rankings.QualifiedPerfect median rank calibration at the largest \(N\) in the tested Gaussian family; not unbiased.
Proxy ranks absolute image-space decoder difficulty.SupportedFashion Spearman \(1.000\); CIFAR-10 comparable bridges \(0.893\); CIFAR-100 training-loss direction.
Image proxy adds information beyond target variance.Not establishedTarget variance remains comparably predictive on Fashion-MNIST and CIFAR-10.
Normalized \(\eta\) universally ranks designs.Not establishedEstimator- and representation-sensitive.
Lower proxy universally improves FID/KID/IS.Not establishedCIFAR-100 is a direct counterexample in the current pilot.
The full path-space VAE/current-gauge/field theory is completed.Open programThe present paper rigorously studies the bridge-to-sampler projection layer.
The target- and observation-relative Markovization gap is the exact irreducible regression floor in solvable bridge models. Cross-fitted PCA/\(k\)-NN error is an inexpensive, empirically rank-calibrated proxy for absolute fixed-budget Markov-decoder difficulty across the tested coupling and bridge designs. It is not a certified population estimate, a scale-free score, or a universal predictor of final generative quality.
Practical use

11. Design rules after the rebuttal

Define the target first

Do not call every stochastic local object “velocity.” State whether \(U_t\) is a drift, current, score-derived target, or deterministic derivative.

Define the information state

Write \(S_t=[\mathcal O_U(L)]_t\), including any memory, latent coordinate, or augmented field variable available to the sampler.

Report simple baselines

Compare the proxy with endpoint cost and total target variance; do not interpret raw reduction as scale-free information gain.

Prefer cross-fitting

Use held-out conditional-mean error as the primary image proxy and local variance only as a sensitivity diagnostic.

Audit the clock

The time law \(\mu\) and schedule determine objective weighting. Closely related parameterizations need not have the same optimization behavior.

Screen, do not canonize

Use the proxy to eliminate structurally difficult designs before training—not as a replacement for full downstream evaluation.

References and neighboring frameworks

This page is a research-oriented exposition of the BGM program and the current Markovization-gap paper. It deliberately distinguishes exact population claims, calibrated proxy claims, downstream evidence, and open extensions.