acceptodds
Under review as a conference paper at ICLR 2027

Bridge-Preserving Gradient Calibration for Replay in Neural Schrödinger Bridges

Abstract

Replay reduces simulation costs in neural Schrödinger bridge training, but stale endpoint pairs can distort gradient updates. Matching intermediate-state distributions alone does not resolve this mismatch and may alter conditional bridge laws. We propose Bridge-Preserving Gradient Calibration (BPGC), which reweights intact replay endpoint pairs through a bounded convex gradient-matching objective and generates independent training bridges. We establish conditional bridge preservation and a finite calibration certificate, and derive update-error bounds that expose sampling noise and unmeasured gradient components. The analysis distinguishes projected calibration from full-gradient accuracy and identifies limitations from replay support and parameter drift. We specify an evaluation protocol that separates gradient fidelity from downstream quality and compares calibration with direct fresh-data use under matched computational budgets.

Then back it, or bet against it.

Related papers

Open the market on this paper to see 7 more related papers.