MATH · IN · MODELS
structures / Manifolds / Score-Jacobian pullback Riemannian metric

Score-Jacobian pullback Riemannian metric

TOOLmetric-modelscore-jacobian-metricadvancedhow it's classified →

A genuine Riemannian metric tensor G = JᵀJ built directly from the Jacobian of a real diffusion model's own score function, with no linearity assumption and no ambient/latent metric pulled back through an assumed linear structure — used to compute length-minimizing geodesics between real data points for interpolation. Distinct from a Fisher-Rao information-geometric metric (diffusion-spacetime-information-geometry) despite both living on a diffusion model's own manifold.

Replicationcomputed from the corpus — never hand-assigned
1 paper1 architecture class1 domain2 model families · one study
Filled = two or more values reported by papers that share no author — replication. Outlined = two or more values, but all from a single study — breadth, not replication. Grey = a single value. Derived from paper authorship and each model's architecture class, domain and family; it updates itself when a paper is added.

Definition

For a diffusion model’s score function sθ(xt,t)xtlogp(xt)s_\theta(x_t,t)\approx\nabla_{x_t}\log p(x_t), the Jacobian Jxt=sθ/xtJ_{x_t}=\partial s_\theta/\partial x_t defines a Riemannian metric tensor at each point directly in data/noise space:

Gxt=JxtJxtG_{x_t} = J_{x_t}^\top J_{x_t}

This metric lives natively where the model itself operates — no assumption that some separate linear latent space carries the “true” geometry, and no metric pulled back from an ambient activation space through an assumed-linear encoder. Length-minimizing geodesics under GxtG_{x_t} between two real data points (computed via DDIM inversion to locate the corresponding noise-space endpoints) define a genuinely curved interpolation path, distinct from a straight line (Lerp) or a constant-norm spherical arc (Slerp) between the same two endpoints.

Relative to diffusion-spacetime-information-geometry

Diffusion Spacetime Information Geometry is a different metric construction on a diffusion model’s own manifold: it treats the model’s family of denoising distributions {p(x0xt,t)}\{p(x_0\mid x_t,t)\} as an exponential-family statistical manifold and derives a Fisher-Rao metric from that distributional structure, indexed by the joint “spacetime” coordinate (xt,t)(x_t,t). This node’s metric is instead built directly from the score function’s own Jacobian at a fixed diffusion step, with no distributional/exponential-family argument involved — same broad theme (a metric intrinsic to the trained denoiser, not pulled back from an assumed ambient/linear structure), different mathematical object and different derivation. The two are linked via related: rather than merged, since neither reduces to the other.

Relative to pullback-of-assumed-linear-latent approaches

Park et al. (2023, already in this map under Linear Direction) pull back an assumed linear latent-space metric to define local geometry for diffusion-model latent-space edits. This node’s metric instead requires no such assumption: it is derived directly from the real model’s own score-Jacobian in data/noise space, with the paper below explicitly contrasting the two approaches.

Key evidence

Saito & Matsubara (2025) construct Gxt=JxtJxtG_{x_t}=J_{x_t}^\top J_{x_t} from the Jacobian of a real, pretrained Stable Diffusion v1-5 model’s own score function, and use it to compute length-minimizing geodesics between real image pairs (via DDIM inversion). Geodesic-guided interpolation improves CLIP-IQA “Reality” (0.716 vs. 0.389-0.704 for Lerp/Slerp/NAO/NoiseDiffusion baselines) and “Noisiness” (0.818) while matching the best reconstruction error (MSE, LPIPS) among all baselines tested. A secondary experiment repeats the same score-Jacobian metric construction and geodesic computation on a small custom diffusion model trained from scratch on MNIST, confirming the method is not specific to one model’s architecture or training scale. See a-riemannian-metric-derived-directly-from-stable-diffusions-own-score-function-jacobian-not-pulled-back-from-any-assumed-linear-latent-and-geodesic-guided-interpolation-outperforms-lerp-slerp-baselines.

Key papers

  • Saito, S. & Matsubara, T. (2025). Image Interpolation with Score-based Riemannian Metrics of Diffusion Models. arXiv:2504.20288 — origin of the score-Jacobian pullback metric and its use for geodesic-guided image interpolation.

Found in (1 observation · 2 families)

Stable Diffusion

Image Interpolation with Score-based Riemannian Metrics of Diffusion Models (2025)measured

A Riemannian metric derived directly from Stable Diffusion's own score-function Jacobian -- not pulled back from any assumed linear latent -- and geodesic-guided interpolation outperforms Lerp/Slerp baselines

Details

Saito & Matsubara (2025) construct a genuine Riemannian metric tensor $G_{x_t}=J_{x_t}^\top J_{x_t}$ directly from the Jacobian of a real, pretrained diffusion model's own score function, explicitly contrasting this with the existing Park et al. (2023) approach (already in this map, structure `linear-direction`) of pulling back an assumed linear latent-space metric -- here the metric lives natively in data/noise space with no linearity assumption. Using this metric to compute length-minimizing geodesics between real image pairs (via DDIM inversion) in real Stable Diffusion (v1-5), geodesic-guided interpolation improves CLIP-IQA "Reality" (0.716 vs. 0.389-0.704 for Lerp/Slerp/NAO/NoiseDiffusion baselines) and "Noisiness" (0.818) while matching the best reconstruction error (MSE, LPIPS) among all baselines. A secondary experiment repeats the same score-Jacobian metric construction and geodesic computation on a small custom diffusion model trained from scratch on MNIST, confirming the method is not specific to one model's architecture or training scale. Both criteria clear: (a) a genuinely derived, quantified Riemannian metric on real model activations; (b) using that metric to guide generation (geodesic interpolation) with a measured, baseline-beating image-quality effect.

models: Stable Diffusion v1.5 · method: Riemannian pullback-metric curvature analysis, Geometric analysis

Saito & Matsubara (2025) Custom MNIST Diffusion Model

Image Interpolation with Score-based Riemannian Metrics of Diffusion Models (2025)measured

A Riemannian metric derived directly from Stable Diffusion's own score-function Jacobian -- not pulled back from any assumed linear latent -- and geodesic-guided interpolation outperforms Lerp/Slerp baselines

Details

Saito & Matsubara (2025) construct a genuine Riemannian metric tensor $G_{x_t}=J_{x_t}^\top J_{x_t}$ directly from the Jacobian of a real, pretrained diffusion model's own score function, explicitly contrasting this with the existing Park et al. (2023) approach (already in this map, structure `linear-direction`) of pulling back an assumed linear latent-space metric -- here the metric lives natively in data/noise space with no linearity assumption. Using this metric to compute length-minimizing geodesics between real image pairs (via DDIM inversion) in real Stable Diffusion (v1-5), geodesic-guided interpolation improves CLIP-IQA "Reality" (0.716 vs. 0.389-0.704 for Lerp/Slerp/NAO/NoiseDiffusion baselines) and "Noisiness" (0.818) while matching the best reconstruction error (MSE, LPIPS) among all baselines. A secondary experiment repeats the same score-Jacobian metric construction and geodesic computation on a small custom diffusion model trained from scratch on MNIST, confirming the method is not specific to one model's architecture or training scale. Both criteria clear: (a) a genuinely derived, quantified Riemannian metric on real model activations; (b) using that metric to guide generation (geodesic interpolation) with a measured, baseline-beating image-quality effect.

models: Custom diffusion model (MNIST, T=1000 steps) · method: Riemannian pullback-metric curvature analysis, Geometric analysis