When Do Local Score Models Extrapolate Across Size? A Diagnostic Theory and Benchmark
arXiv:2606.09705v1 [cs.LG] 8 Jun 2026
Wenjie Xi Department of Physics and HK Institute of Quantum Science & Technology The University of Hong Kong, Pokfulam Road, Hong Kong, China [email protected] Abstract Scientific generative modeling often requires size transfer, where models trained on small systems are evaluated on larger ones. While translation-invariant architectures enable this evaluation, we show that architectural locality alone does not guarantee stable size extrapolation. Instead, stable extrapolation is governed by the quasi-locality of the Gaussian-smoothed score. Through Tweedie’s formula, far-away perturbations can influence local score components via posterior covariance, meaning a local model succeeds only if its receptive field covers the smoothed score’s response range. We formalize this mechanism, proving a size-uniform comparison theorem for local marginals under reverse diffusion. We also introduce Finite-Depth Local Flow (FDLF), a white-box diagnostic benchmark with exact scores, densities, and controllable response ranges. Empirically, we validate the interplay between spatial mixing, smoothed-score quasi-locality, and model receptive fields. Under spatial mixing, the smoothed score remains quasilocal relative to the receptive field, enabling stable extrapolation. Conversely, when spatial mixing weakens, the score’s locality rapidly degrades, causing size transfer to fail.
1
Introduction
Many scientific generative modeling problems require size transfer: a model is trained on small, accessible systems but is expected to operate on substantially larger ones. While translation-invariant architectures like local CNNs or graph neural networks make this evaluation possible at the implementation level, a fundamental question remains: when is the learned local score rule actually the correct physical rule after the domain grows? Our answer is that stable size extrapolation is governed by the quasi-locality of the smoothed score. Denoising score matching learns Gaussian-smoothed scores Hyvarinen (2005); Vincent (2011); Song et al. (2021), not merely the clean score. Even when the clean density has finite-range structure, smoothing rewrites the score as a posterior expectation through Tweedie’s formula Robbins (1992); Efron (2011); far-away perturbations can influence a local score component through posterior covariance. When the noisy posterior is spatially mixing Dobrushin (1968); Georgii (2011), this influence decays and a finite receptive field can capture the target. When the smoothed score has substantial response beyond the receptive field, stable extrapolation should not be expected from architectural locality alone. This insight leads to a testable diagnostic: a local score model should stably transfer across sizes when its receptive field covers the effective response range of the target smoothed score, and it becomes unreliable when it does not. This diagnostic is stronger than merely checking whether an architecture can be evaluated on larger systems; it asks whether the learned rule has captured the spatial response that actually determines the local reverse-diffusion dynamics. However, diagnosing this mechanism in practical scientific applications is highly challenging, as exact scores are typically unavailable and the data-generating rules of realistic systems are black-box. To address these challenges, we make three main contributions. First, we formalize local score extrapolation and identify posterior covariance as the static object governing the smoothed-score response. Second, we prove a sizeuniform local-marginal comparison theorem. Under the assumption of dynamic quasi-locality of the exact reverse process, we show that spatially weighted on-rollout score error controls fixed-patch marginals uniformly in system size. Third, we introduce Finite-Depth Local Flow (FDLF), a white-box diagnostic benchmark with exact scores, exact densities, and controllable response ranges. This allows us to systematically evaluate positive controls, receptive-fieldlimited cases, and controlled failures under a single protocol.
1
Empirically, our proposed diagnostic is predictive. On 2D continuous FDLF teachers, local scores extrapolate stably, with medium-range teachers showing clear improvement as the CNN receptive field expands, while long-range responses remain a controlled failure. This behavior persists in a more challenging mixed discrete-continuous setting with exact local validity constraints. Finally, a critical Ising stress test demonstrates the physical mechanism behind failure: as the posterior spatial mixing weakens near criticality, the smoothed-score response becomes long-ranged, and fixed-radius local extrapolation becomes unreliable. Relation to prior work. Learning across discretization and size. Neural operators learn maps between function spaces and are designed for discretization or resolution transfer in PDE settings (Li et al., 2021; Kovachki et al., 2023; Lu et al., 2021). This is related in spirit to size transfer, but the target is usually a deterministic solution operator rather than a probabilistic score field whose locality changes under Gaussian smoothing. A second line learns size-extensive or thermodynamic-limit predictors by decomposing energies or observables into local contributions, including neuralnetwork interatomic potentials and extensive neural networks for large physical systems (Behler and Parrinello, 2007; Mills et al., 2019; Schutt et al., 2017; Batzner et al., 2022). These works exploit locality to make energy or force prediction scalable. Our focus is different: even when a clean model is local, the denoising score at positive noise is controlled by the noisy posterior response, so size extrapolation depends on a learned score-response range, not only on extensivity or local energy decomposition. Local generative models, score theory, and statistical-mechanics limits. Equivariant and local generative models provide powerful inductive biases for particles and molecular geometries, including equivariant graph networks and diffusion models (Satorras et al., 2021; Hoogeboom et al., 2022; Xu et al., 2022; Jing et al., 2022). These methods address symmetry and geometry; our diagnostic asks when a finite local rule remains valid as system size grows. Existing score-based theory gives convergence guarantees from accurate score estimates (Lee et al., 2022; Chen et al., 2023), whereas our theorem is local and size-uniform, emphasizing spatially weighted score error and response tails. The physical side is closest to finite-size scaling (Fisher and Barber, 1972; Privman, 1990), correlation decay and spatial mixing for Gibbs measures and sampling (Dobrushin, 1968; Georgii, 2011; Weitz, 2006), and local weak convergence of rooted finite graphs (Benjamini and Schramm, 2001; Aldous and Steele, 2004). We borrow the local-limit viewpoint but study a different object: the Gaussian-smoothed score learned by diffusion training and its measurable response range under finite receptive fields.
2
Local Score Extrapolation
Let ΛL be a finite spatial domain and let x = (xi )i∈ΛL with xi ∈ Rd . For each system size L, let pL be a density on (Rd )|ΛL | . In the size-extrapolation setting, a score model is trained on samples from pLtrain and then evaluated without retraining on larger systems pLtest , where Ltest > Ltrain . The modeling assumption is that {pL }L is generated by a shared size-uniform mechanism; only the domain size changes. This assumption rules out a trivial obstruction. If the distribution itself changes qualitatively with L, no fixed local rule should be expected to extrapolate. Our focus is the more subtle case in which the underlying rule is shared across sizes, yet the learned local score may still be wrong because the score component at a site depends on a spatial context larger than the model can see. We therefore evaluate extrapolation locally: a model succeeds if fixed-size patches, bulk score components, and local response probes remain stable as L increases. For score-based diffusion learning (Sohl-Dickstein et al., 2015; Song and Ermon, 2019; Ho et al., 2020; Song et al., 2021), let X0 ∼ pL and Xσ = X0 + σε, ε ∼ N (0, I). (1) The smoothed density is Z pL,σ (x) =
pL (x0 ) N (x; x0 , σ 2 I) dx0 ,
with target score sL,σ (x) = ∇x log pL,σ (x). Denoising score matching (Vincent, 2011) trains sθ by ε 2 . LDSM (θ) = Eσ,X0 ,ε sθ (X0 + σε, σ) + σ 2 At population optimum, this objective recovers the smoothed score sL,σ (Vincent, 2011; Song et al., 2021).
2
(2)
(3)
A local score model represents a size-independent site rule, sθ (x, σ)i = Fθ (xBR (i) , σ),
(4)
where BR (i) = {j : dist(i, j) ≤ R}. The same R and θ are used for all system sizes. Thus, the relevant extrapolation question becomes: is the true score component at site i effectively determined by information inside BR (i)? To make this question operational, define the response tail of a differentiable score field by TL,σ (i; R, x) =
X dist(i,j)>R
∂sL,σ (x)i ∂xj
.
(5)
op
If TL,σ (i; R, x) is small uniformly over relevant x, then variables outside the receptive field have little first-order influence on the local score. If it is large, no radius-R score model can represent the true response without additional global information. The experiments use finite-difference versions of this quantity as observable diagnostics.
2.1
Clean and Smoothed Locality
For clean densities, strict locality follows from a local log-density decomposition. Suppose X log pL (x) = ϕj (xBr (j) )
(6)
j∈ΛL
with interaction radius r independent of L. Then ∇xi log pL (x) only receives contributions from terms whose neighborhoods contain i, and each such term depends on variables at distance at most 2r from i. Hence the clean score is finite-range local. This elementary statement is useful but insufficient for diffusion training, which targets smoothed scores. For σ > 0, Tweedie’s formula (Robbins, 1992; Efron, 2011) gives sL,σ (x) = σ −2 E[X0 | Xσ = x] − x , (7) where the posterior over clean configurations is
x πL,σ (dx0 ) ∝ pL (x0 ) exp
∥x − x0 ∥22 − 2σ 2
dx0 .
(8)
Consequently, differentiating the posterior mean gives the covariance, or linear-response, identity (Georgii, 2011): for i ̸= j, ∂sL,σ (x)i x = σ −4 CovπL,σ (X0,i , X0,j ). (9) ∂xj Thus the spatial response of the smoothed score is governed by posterior correlations. If the noisy posterior has uniform correlation decay, as in classical spatial-mixing regimes for Gibbs measures (Dobrushin, 1968; Georgii, 2011), the score Jacobian is quasi-local. If spatial mixing fails, the smoothed score can have substantial long-range response even when the clean prior has short-range interactions. Equation (9) is the conceptual hinge of the paper. It says that the effective receptive field required by a denoising score model is a property of the noisy posterior, not only of the clean interaction graph. Increasing CNN depth should help when it reduces the response tail in Eq. (5); it should not solve cases where the teacher still has large response beyond all tested radii. This is exactly the prediction tested by the receptive-field sweep in Section 4.1.
2.2
Local-Marginal Target and Size-Uniform Control
The theory supports a deliberately local target. In an Ld -site system, global total variation, KL, or Wasserstein errors can scale with volume even when every fixed patch is generated accurately. Size extrapolation for local scientific structure is therefore better phrased as a thermodynamic-limit question: for every fixed patch size m, do the generated marginals on all |A| ≤ m patches remain accurate uniformly over the ambient size L? Let PL,A be the target terminal b L,A be the learned terminal marginal, and measure local distribution error by the boundedmarginal on A ⊆ ΛL , let Q Lipschitz distance dBL , i.e., the largest discrepancy over bounded 1-Lipschitz test functions on the patch. 3
To connect this target to score learning, write the exact and learned reverse processes as √ √ bt = bbL (X bt ) dt + 2 dWt . 2 dWt , dX dXt = bL t t (Xt ) dt +
(10)
L The drift error bbL t − bt is the score error up to the usual reverse-diffusion coefficients. For a patch observable ϕ(xA ), define the exact backward semigroup L Pt,T ϕ(x) = E[ϕ(XT ) | Xt = x],
the expected terminal value of ϕ under the exact reverse process initialized at state x and time t. Although ϕ observes L only patch A, the function Pt,T ϕ may depend on sites outside A because errors can propagate backward through the reverse dynamics. The dynamic quasi-locality condition says this propagated sensitivity has a summable spatial envelope: X L L L ∥∇i Pt,T ϕ∥ ≤ Cloc (m, T ) Lip(ϕ) ωA,t (i), sup ωA,t (i) < ∞. (11) L,A,t
i
L Here Lip(ϕ) is the Lipschitz constant of the patch test function, ωA,t (i) is an influence weight measuring how much
a drift error at site i can affect patch A, and Cloc (m, T ) is independent of the ambient size L. Thus Eq. (11) is a finite-speed-of-influence condition for local marginals: the exact reverse dynamics may spread information, but the total influence budget around a fixed patch remains uniformly bounded. This turns global score error into a spatially weighted on-rollout error around the target patch: Z T X L L ηm = sup EνbtL ωA,t (i)∥bbL (12) t,i − bt,i ∥ dt, L,|A|≤m
0
i
where νbtL is the law of the learned process at time t. In words, ηm averages the site-wise drift/score error along learned rollouts, but weights each site by how much it can influence the target patch. Score errors far outside the response cone are therefore irrelevant to this local metric, while errors inside the cone are counted uniformly over system size. The appendix proves the following conditional comparison theorem. Theorem 1 (Informal size-uniform local-marginal control). Fix m < ∞. Under well-posedness, initialization consistency, and dynamic quasi-locality of the exact reverse process, the learned terminal local marginals satisfy b L,A , PL,A ) ≤ δ0,m + Cloc (m, T )ηm . sup sup dBL (Q
(13)
L A⊆ΛL |A|≤m
Here δ0,m is the initial local mismatch, Cloc (m, T ) is independent of L, and ηm is defined in Eq. (12). L The proof idea is the bridge to the experiments. Apply the backward Kolmogorov equation to Pt,T ϕ and Itô’s formula along the learned process. The terminal patch error equals an initial mismatch plus an integral of the drift L L ϕ. Dynamic quasi-locality weights this gradient by ωA,t , so only score errors inside the response error against ∇i Pt,T cone of the patch matter. Thus the theorem says: if the exact reverse dynamics are quasi-local and the learned score is accurate in the spatial region that can influence the patch, then local marginals are controlled uniformly over L. The response-tail diagnostics in the experiments test whether this sufficient mechanism is plausible for a given teacher and receptive field.
3
FDLF: A Testable Diagnostic Benchmark
Realistic scientific datasets are essential for applications, but they are poor instruments for isolating size extrapolation mechanisms: exact scores are unavailable, the data-generating rule is unknown, and the score-response range cannot be independently controlled. We therefore introduce Finite-Depth Local Flow (FDLF) as a testable diagnostic benchmark. Its purpose is not to maximize realism, but to make the locality mechanism observable. FDLF is built around three requirements that standard datasets usually cannot satisfy simultaneously. First, the teacher must expose exact clean scores, so score error can be measured directly rather than inferred from samples. Second, the same teacher rule must be reusable at every L, so size extrapolation is not confounded with a change in data distribution. Third, the teacher’s response range must be tunable, so we can create positive controls, intermediate 4
Need for diagnosis
FDLF provides
Role in this paper
Target score access Size-uniform mechanism Controlled locality Response inspection
exact density and clean score same local map for every L tunable depth/range/smoothing exact or finite-difference probes
direct score extrapolation error isolates domain-size transfer positive and negative controls tests the receptive-field hypothesis
Table 1: Why FDLF is useful as a diagnostic benchmark. The benchmark is designed to expose the mechanism behind local size extrapolation, not to serve as a black-box application dataset. cases, and controlled failures under one protocol. These requirements make FDLF a diagnostic instrument rather than a realism benchmark. Following the normalizing-flow change-of-variables framework (Rezende and Mohamed, 2015; Papamakarios et al., 2021), let u = (ui )i∈ΛL be a latent field with tractable density qL , and let TL be a size-uniform finite-depth local invertible map. The target field is x = TL (u),
pL = (TL )# qL .
(14)
Its density and clean score are pL (x) = qL (TL−1 x) det DTL−1 (x) ,
sL,0 (x) = ∇x log pL (x).
(15)
All concrete FDLF teachers used here employ coupling-style invertible constructions (Dinh et al., 2017) whose inverse and log-Jacobian remain local or controlled-range. By varying the depth, coupling range, and smoothing kernel, we obtain positive controls with short response ranges, medium-range teachers whose responses are captured only by larger CNNs, and long-range stress teachers. The short- and medium-range teachers reuse the same finiterange rule across L. The long-range stress family uses the same fixed parameters but includes an exact mean-mode component whose strength is γ(L) = 0.005L3/2 , making it a deliberately non-quasi-local negative control. This benchmark design also clarifies what is being claimed. We do not claim that FDLF itself is a realistic scientific data model. Instead, we claim that any proposed mechanism for local size extrapolation should pass controlled tests of this kind: it should succeed when the teacher response is inside the receptive field, improve when the receptive field is enlarged to cover an intermediate response range, and fail or degrade when the teacher response remains outside the field. The following sections instantiate exactly these cases.
4
Controlled FDLF Experiments
4.1
2D Continuous Diagnostic
We first study periodic 2D continuous lattices, the cleanest setting in which the receptive-field hypothesis can be tested. The fixed-architecture size sweep trains only at Ltrain = 16; the receptive-field sweep trains at Ltrain = 32, so even the largest tested CNN radius Reff = 14 remains below half the training lattice width and does not become a wrapped global model. All score models in this section are pure CNNs with finite receptive fields; increasing the residual-block count increases the effective radius while preserving the same local convolutional inductive bias. A single 3 × 3 convolution expands the square lattice support by one site in Chebyshev distance. In our CNN, a model with B residual blocks has one input projection, two 3 × 3 convolutions per residual block, and one output projection, so its effective radius is Reff = 2B + 2. The primary diagnostic is the low-noise clean-score relative RMSE RMSEscore (L) =
Ex∼pL ∥sθ (x, σmin ) − sL,0 (x)∥22 Ex∼pL ∥sL,0 (x)∥22
1/2 .
(16)
Although DSM targets sL,σ , this low-noise clean-score metric is a controlled diagnostic of extrapolation trends. We also perturb local and distant regions and compare finite-difference score responses, ∆sL,0 = sL,0 (x + δ) − sL,0 (x),
∆sθ = sθ (x + δ, σmin ) − sθ (x, σmin ).
5
(17)