REVIEW 3 major objections 3 minor 30 references
Sampling in High-Dimensions using Stochastic Interpolants and Forward-Backward Stochastic Differential Equations
T0 review · 3 major / 3 minor · reviewed 2026-08-09 · deepseek-v4-flash
Pith's one-line read This paper designs diffusion processes that carry Gaussian samples to a target distribution in finite time by solving a Hamilton–Jacobi–Bellman PDE through forward-backward stochastic differential equations.
desk verdict The theory is mostly sound and the FBSDE training idea is new, but the Eq. (19) loss is not a consistent finite-time surrogate for the HJB and the paper's own high-dimensional benchmarks fail exactly there. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The load-bearing object is the stochastic interpolant $x_t=g(t)x_*+r(t)z$, a time-indexed random variable whose density bridges the Gaussian prior to the target. The core identity is the representation of the conditional expectation as a gradient: $E[x_*\mid x_t=x]=\frac{r^2(t)}{\beta(t)g(t)}\nabla u(t,x/\beta(t))$, where $u(t,x)=\log[\rho(t,\beta(t)x)/\psi(t,\beta(t)x)]$; this converts the sampling problem into the problem of finding $\nabla u$. The function $u$ is characterized as the solution of the HJB equation (14), and the machinery that solves this PDE is its FBSDE formulation (16), whose processes satisfy $Y_t=u(t,X_t)$ and $Z_t=\sigma(t)\nabla u(t,X_t)$. Training uses the local FBSDE loss (19), which enforces the backward equation at randomly sampled times along a detached ODE trajectory (17); detachment keeps the training domain from depending on the network's gradient path and avoids Neural SDE back-propagation.
What would settle it
Train the sampler on a target with known normalization and exact score (for example, a non-isotropic Gaussian or the $d=100$ spin-glass model with its exact free energy), and monitor the terminal residual $\|\nabla u_\theta(T,X_T)-\nabla\varphi(X_T)\|$ and the Wasserstein distance between $S_T$ and $\pi$ as $\lambda$ and the training steps grow; if neither approaches zero, or if the $\log Z$ estimate shows a bias that persists when $\lambda$ is increased, the finite-time exactness claim is falsified.
Extended reading notes
Core claim
On its own terms, the paper claims that exact finite-time transport is a consequence of the stochastic-interpolant construction. Given a target density $\pi$ on $\mathbb{R}^d$, choose interpolant functions $g,r$ with $g/r$ non-decreasing and define $x_t=g(t)x_*+r(t)z$ with $x_*\sim\nu$ and $z\sim N(0,I_d)$; the terminal condition $\rho(T,\cdot)=\pi$ fixes $\nu$ implicitly. Lemma 3 shows that the tilting function $u(t,x)=\log[\rho(t,\beta(t)x)/\psi(t,\beta(t)x)]$ satisfies the HJB equation (14) with terminal condition (15), and Lemma 4 shows that the FBSDE (16) has solutions $Y_t=u(t,X_t)$ and $Z_t=\sigma(t)\nabla u(t,X_t)$. Since $E[x_*\mid x_t=x]=r^2(t)\nabla u(t,x/\beta(t))/(\beta(t)g(t))$, learning $\nabla u$ on the right domain gives both drift $b$ and score $s$ in (12), and then the SDE (11) has exactly the interpolant density $\rho(t,\cdot)$ at every time, so $S_T\sim\pi$. The empirical version replaces the PDE solution with a neural-network approximation trained through the local FBSDE loss (19), and the paper reports that this yields accurate samples and $\log Z$ estimates on distributions that conventional samplers struggle with.
Load-bearing premise
The central guarantee rests on the assumption that minimizing the local FBSDE loss actually converges to the true solution $\sigma\nabla u$ of the HJB equation, even though the terminal condition is imposed only through a finite penalty and the region where the equation is enforced is defined by the very network being trained.
Editorial extensions
If this is right
- If the central claim is correct, a trained network turns the SDE (11) into a finite-time sampler: Gaussian draws at $t=0$ become target draws at $t=T$ with no annealing, rejection, or burning-in.
- Because the paper estimates moments and $\log Z$ directly from the samples, the claim implies that the generated samples are good enough for downstream estimators without importance reweighting; the Itô-formula identity (35) gives normalization constants from the same trained $\nabla u$.
- The detached training ODE (17) means path gradients are not back-propagated, and the paper reports per-step training times between 98 and 105 ms for targets of dimension 2 through 100.
- The half-interpolant sampler generalizes the Föllmer process and reduces to the Path Integral Sampler when $g(t)=f(t)$, $r(t)=\sqrt{f(t)}$, while the full-interpolant two-step version extends finite-time transport to interpolants with $r(T)=0$.
Reading between the lines
- A sharper test of the exactness claim would be to measure, across many random disorder matrices, the variance of the sampler's free-energy estimate for the spin-glass model near the predicted $\beta=1$ phase transition; a bias that does not vanish with $\lambda$ would locate the effect of the soft terminal constraint.
- The local FBSDE loss may transfer to other HJB problems where the evaluation paths differ from the controlled paths, since the loss enforces the PDE residual pointwise in time rather than over the trajectory of a single SDE.
- Replacing the soft terminal penalty with a hard terminal parameterization, exploiting the fact that the network is initialized so that $u_\theta(T,\cdot)=\varphi$, would directly probe the hyperparameter sensitivity reported in Table 6 and could yield a sampler with fewer tunable constants.
- If exact finite-time transport holds, the sampler should also work as a proposal for annealed importance sampling when reweighting is desired, since the paper itself notes that importance weights are currently unavailable.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes a class of finite-time diffusion samplers for unnormalized densities, built on stochastic interpolants and forward-backward stochastic differential equations (FBSDEs). The authors define half and full interpolants, derive Hamilton-Jacobi-Bellman (HJB) equations for the associated log-density functions, and propose neural-network training based on a local FBSDE residual together with a terminal gradient penalty. After training, sampling is performed by simulating the SDE (or ODE) whose drift and score are obtained from the learned gradient. The theoretical portion (Lemmas 1-4 and the corresponding appendix proofs) is largely coherent: the HJB/FBSDE relation and the Gaussian-to-target transport identity are derived consistently. The empirical portion compares the resulting Full Interpolant Sampler against Langevin Monte Carlo and the Generalized Bridge Sampler on Gaussian mixtures, funnels, double-well potentials, and a spin-glass model, with log-normalizing-constant and Wasserstein metrics. The central empirical claim, however, is not established, because the training objective in Eq. (19) has a scaling defect and the training trajectories are generated by the very network being trained.
Significance. If the proposed training procedure were sound, the paper would offer a useful finite-time alternative to Neural-SDE-based samplers such as PIS, with the practical advantage that no gradients through stochastic paths are required. The mathematical derivation of the HJB/FBSDE connection and the finite-time exact-transport statement in the ideal (exactly solved PDE) case are valid and constitute a genuine theoretical contribution. I checked the core identities in Lemmas 1-4 and the Itô-based relation between the FBSDE (16) and the HJB equation (14), and they are consistent. Nevertheless, the load-bearing step is the claim that minimizers of the empirical loss (19) solve the HJB equation on the support of the sampling SDE. That step is neither proved nor convincingly supported by the numerical evidence, and the loss-scaling issue in §3.1.1 casts serious doubt on the current formulation as a consistent PDE-solving method.
major comments (3)
- The FBSDE term in the loss is not a consistent surrogate for the HJB residual. In Eq. (18), the leading O(√δ) Brownian terms in Y^δ and u_θ(τ+δ, X^δ) cancel for any u_θ, because both terms use the same Z = σ∇u_θ: expanding u_θ at (τ+δ, X+(μ+σZ)δ+σ√δ w) gives the same √δ Z^T w as Y^δ. Consequently the pathwise residual is O(δ), its square is O(δ²), and with λ = 2000/δ the FBSDE contribution is O(δ) per sampled time, about 10⁻² at δ = 5×10⁻⁶, while the terminal gradient penalty in (19) is O(1) and is evaluated at a single detached point X_T. Unless the number of FBSDE checks is made to grow as 1/δ — which is not stated and would be computationally burdensome — the terminal penalty dominates and the loss does not enforce the PDE (14). No consistency or convergence theorem is provided for minimizers of (19).
- The domain on which the FBSDE residual is trained is defined by the current network and is explicitly detached from the computational graph in Section F.2. The ODE (17) uses ∇u_θ to generate the trajectories X_t, so at any stage of training the loss is minimized over paths that the network itself produces. At convergence to the true ∇u these trajectories would coincide with the sampling ODE (10), but that is exactly the conclusion to be established; nothing guarantees that a small loss on these self-generated trajectories transfers to the support of the sampling SDE (11). This circularity is load-bearing, because the finite-time transport claim requires ∇u to be accurate precisely on the domain sampled by (11), not merely on trajectories of the detached ODE.
- The empirical evidence does not support the abstract's claim that the method can effectively draw samples from distributions that conventional methods struggle to handle. On a 10-component Gaussian mixture at d = 10, FIS reports log Z = −2.43 ± 0.02 against the true value 0 and a 2-Wasserstein distance of 243.5 ± 5, whereas GBS gives −0.02 ± 0.03 and 6.21 ± 0.04; at d = 200 the FIS log Z estimate is −183.56 ± 90.24. Table 6 also shows that on a 2D GMM the log Z estimate varies from −0.25 to −0.74 when δ or λ is changed by a factor of 10. These results are consistent with the loss-scaling issue in Major Comment 1 and do not demonstrate that the trained sampler transports the Gaussian prior to the target in finite time.
minor comments (3)
- There are several typos: “non-deceasing” in Definition 1, “score funciton” in §3.1, and “paramterized” in §3.1.1. These should be corrected.
- The reference to “Bismuth” [Bismut, 1973] is misspelled; it should be Bismut.
- The log Z estimator (35) is derived by Itô's formula for the exact solution u. In practice ∇u is replaced by the trained ∇u_θ, and the resulting bias is not analyzed. The paper should state explicitly that the estimator is valid only to the extent that the learned gradient approximates the true gradient on the relevant SDE paths.
Circularity Check
No significant circularity: the HJB/FBSDE construction is self-contained, with the target density entering only through the terminal condition and the log-normalizer estimate derived by Itô's formula.
full rationale
The derivation chain is not circular. Section 3.1 defines u from the interpolant density (Eq. 13), proves by direct differentiation (Lemma 9) that u satisfies the HJB PDE (14), and imposes the terminal condition (15) equal to the target log-density. The terminal condition is an input of the construction, not a derived prediction; if the PDE is solved exactly, the Fokker-Planck argument (Lemma 1, Eq. 9) ensures S_T has law π. The FBSDE representation (16) and the identification Y_t = u(t,X_t), Z_t = σ∇u (Lemma 10) rely on the standard Pardoux-Peng theorem, which is cited as external mathematical support. The log-normalizer estimator (35) is an Itô identity, not a fitted parameter. No load-bearing self-citation appears: the stochastic-interpolant framework is cited to Albergo et al. and the FBSDE connection to Pardoux-Peng. The self-referential training loop (Eq. 17 uses the current network to define the training distribution, and the process is detached from the computational graph as stated in Section F.2) and the soft terminal penalty with λ = 2000/δ are convergence and consistency risks, not circular reductions, because the loss is a PDE residual rather than a restatement of the target π. The paper's own limitations (Section 5: no importance-weight correction; Table 6: strong hyperparameter sensitivity; Table 5: poor MoG log Z estimates) are correctness concerns that do not make the derivation circular.
Assumptions & free parameters
free parameters (5)
- interpolant schedules g(t) and r(t) =
e.g., sin(pi*t/2), cos(pi*t/2) for full interpolants; sqrt(t) variants for half interpolants
- beta(t) =
r(t)/g(t)
- lambda (FBSDE penalty weight) =
2000/delta = 4e8
- delta (FBSDE discretization step) =
5e-6
- epsilon(t) (sampling diffusion coefficient) =
1.0
assumptions (5)
- standard math Existence and uniqueness of the FBSDE (16) with quadratic generator (1/2)||Z||^2 and coupled forward drift mu + sigma Z
- domain assumption The HJB PDE (14) has a smooth classical solution u for the terminal condition phi = log(pi(beta(T)x)/psi(T,beta(T)x))
- standard math Tweedie's formula E[z|x_t=x] = -r(t)s(t,x) for the Gaussian z
- ad hoc to paper The ODE (17) discretization error does not affect training accuracy
- domain assumption Density pi is positive everywhere so log pi and its gradient are finite
Cite this review
Pith. "Pith review of Sampling in High-Dimensions using Stochastic Interpolants and Forward-Backward Stochastic Differential Equations." pith.science (2026). https://pith.science/paper/7DQL37R4
@misc{pith2026250200355,
author = {Pith},
title = {Pith review of: Sampling in High-Dimensions using Stochastic Interpolants and Forward-Backward Stochastic Differential Equations},
year = {2026},
howpublished = {\url{https://pith.science/paper/7DQL37R4}},
note = {Machine review of arXiv:2502.00355}
}
read the original abstract
We present a class of diffusion-based algorithms to draw samples from high-dimensional probability distributions given their unnormalized densities. Ideally, our methods can transport samples from a Gaussian distribution to a specified target distribution in finite time. Our approach relies on the stochastic interpolants framework to define a time-indexed collection of probability densities that bridge a Gaussian distribution to the target distribution. Subsequently, we derive a diffusion process that obeys the aforementioned probability density at each time instant. Obtaining such a diffusion process involves solving certain Hamilton-Jacobi-Bellman PDEs. We solve these PDEs using the theory of forward-backward stochastic differential equations (FBSDE) together with machine learning-based methods. Through numerical experiments, we demonstrate that our algorithm can effectively draw samples from distributions that conventional methods struggle to handle.
Figures
Figures from the paper (6 more)
Reference graph
Works this paper leans on
-
[1]
write newline
" write newline "" before.all 'output.state := FUNCTION fin.entry add.period write newline FUNCTION new.block output.state before.all = 'skip after.block 'output.state := if FUNCTION new.sentence output.state after.block = 'skip output.state before.all = 'skip after.sentence 'output.state := if if FUNCTION not #0 #1 if FUNCTION and 'skip pop #0 if FUNCTIO...
-
[2]
Albergo, M. S., Boffi, N. M., and Vanden-Eijnden, E. (2023). Stochastic Interpolants : A Unifying Framework for Flows and Diffusions . arXiv:2303.08797 [cond-mat]
arXiv 2023
-
[3]
B., Dembo, A., and Guionnet, A
Arous, G. B., Dembo, A., and Guionnet, A. (2001). Aging of spherical spin glasses. Probability Theory and Related Fields , 120(1):1--67
work page 2001
-
[4]
Barra, A., Genovese, G., Guerra, F., and Tantari, D. (2014). About a solvable mean field model of a Gaussian spin glass. Journal of Physics A: Mathematical and Theoretical , 47(15):155002
work page 2014
-
[5]
Berner, J., Richter, L., and Ullrich, K. (2023). An optimal control perspective on diffusion-based generative modeling. Transactions on Machine Learning Research
work page 2023
-
[6]
Bismut, J.-M. (1973). Conjugate convex functions in optimal stochastic control. Journal of Mathematical Analysis and Applications , 44(2):384--404
work page 1973
-
[7]
Blessing, D., Jia, X., Esslinger, J., Vargas, F., and Neumann, G. (2024). Beyond ELBOs : A Large - Scale Evaluation of Variational Methods for Sampling . arXiv:2406.07423 [cs, stat]
arXiv 2024
-
[8]
Chen, T., Liu, G.-H., and Theodorou, E. (2021). Likelihood Training of Schrödinger Bridge using Forward - Backward SDEs Theory . In International Conference on Learning Representations
work page 2021
Show all 30 references
-
[9]
Del Moral, P., Doucet, A., and Jasra, A. (2006). Sequential Monte Carlo samplers. Journal of the Royal Statistical Society: Series B (Statistical Methodology) , 68(3):411--436. \_eprint: https://onlinelibrary.wiley.com/doi/pdf/10.1111/j.1467-9868.2006.00553.x
2006
-
[10]
E, W., Han, J., and Jentzen, A. (2022). Algorithms for Solving High Dimensional PDEs : From Nonlinear Monte Carlo to Machine Learning . Nonlinearity , 35(1):278--310
2022
-
[11]
Föllmer, H. (1986). Time reversal on Wiener space. In Albeverio, S. A., Blanchard, P., and Streit, L., editors, Stochastic Processes — Mathematics and Physics , pages 119--129, Berlin, Heidelberg. Springer
1986
-
[12]
Grenioux, L., Noble, M., Gabrié, M., and Durmus, A. O. (2024). Stochastic Localization via Iterative Posterior Sampling
2024
-
[13]
Han, J., Jentzen, A., and E, W. (2018). Solving high-dimensional partial differential equations using deep learning. Proceedings of the National Academy of Sciences , 115(34):8505--8510. arXiv:1707.02568
2018 arXiv
-
[14]
and Gimpel, K
Hendrycks, D. and Gimpel, K. (2023). Gaussian Error Linear Units ( GELUs ). arXiv:1606.08415 [cs]
2023 arXiv
-
[15]
Ho, J., Jain, A., and Abbeel, P. (2020). Denoising Diffusion Probabilistic Models . In Advances in Neural Information Processing Systems , volume 33, pages 6840--6851. Curran Associates, Inc
2020
-
[16]
Huang, X., Dong, H., Hao, Y., Ma, Y.-A., and Zhang, T. (2023). Reverse Diffusion Monte Carlo
2023
-
[17]
Kingma, D. P. and Ba, J. (2017). Adam: A Method for Stochastic Optimization . arXiv:1412.6980 [cs]
2017 arXiv
-
[18]
M., Thouless, D
Kosterlitz, J. M., Thouless, D. J., and Jones, R. C. (1976). Spherical Model of a Spin - Glass . Physical Review Letters , 36(20):1217--1220. Publisher: American Physical Society
1976
-
[19]
Neal, R. M. (2001). Annealed importance sampling. Statistics and Computing , 11(2):125--139
2001
-
[20]
Neal, R. M. (2003). Slice sampling. The Annals of Statistics , 31(3):705--767. Publisher: Institute of Mathematical Statistics
2003
-
[21]
Pardoux, E. (1998). Backward Stochastic Differential Equations and Viscosity Solutions of Systems of Semilinear Parabolic and Elliptic PDEs of Second Order . In Decreusefond, L., Øksendal, B., Gjerde, J., and Üstünel, A. S., editors, Stochastic Analysis and Related Topics VI ,...
1998
-
[22]
and Peng, S
Pardoux, E. and Peng, S. G. (1990). Adapted solution of a backward stochastic differential equation. Systems & Control Letters , 14(1):55--61
1990
-
[23]
and Tang, S
Pardoux, E. and Tang, S. (1999). Forward-backward stochastic differential equations and quasilinear parabolic PDEs . Probability Theory and Related Fields , 114(2):123--150
1999
-
[24]
Raissi, M. (2018). Forward- Backward Stochastic Neural Networks : Deep Learning of High -dimensional Partial Differential Equations . arXiv:1804.07010 [cs, math, stat]
2018 arXiv
-
[25]
Richter, L., Berner, J., and Liu, G.-H. (2023). Improved sampling via learned diffusions. arXiv:2307.01198 [cs, math, stat]
2023 arXiv
-
[26]
and Ermon, S
Song, Y. and Ermon, S. (2019). Generative Modeling by Estimating Gradients of the Data Distribution . In Advances in Neural Information Processing Systems , volume 32. Curran Associates, Inc
2019
-
[27]
P., Kumar, A., Ermon, S., and Poole, B
Song, Y., Sohl-Dickstein, J., Kingma, D. P., Kumar, A., Ermon, S., and Poole, B. (2021). Score- Based Generative Modeling through Stochastic Differential Equations . arXiv:2011.13456 [cs, stat]
2021 arXiv
-
[28]
Tancik, M., Srinivasan, P., Mildenhall, B., Fridovich-Keil, S., Raghavan, N., Singhal, U., Ramamoorthi, R., Barron, J., and Ng, R. (2020). Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains . In Advances in Neural Information Processing Sys...
2020
-
[29]
S., and Doucet, A
Vargas, F., Grathwohl, W. S., and Doucet, A. (2022). Denoising Diffusion Samplers . In International Conference on Learning Representations
2022
-
[30]
and Chen, Y
Zhang, Q. and Chen, Y. (2022). Path Integral Sampler : a stochastic control approach for sampling. arXiv:2111.15141 [cs]
2022 arXiv
Reviewed August 9, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.