Coordinate-Wise Variance Collapse
- Coordinate-wise variance collapse is a phenomenon where certain coordinates exhibit abnormally low variability, indicating local memorization or direction-specific concentration in high-dimensional spaces.
- Baseline subtraction methods in diffusion models differentiate intrinsic data constraints from overfitting by comparing coordinate-level curvature between conditioned and unconditional settings.
- Similar collapse behaviors are observed in contrastive learning and stochastic optimization, underscoring its broader relevance for variance reduction techniques in machine learning.
Searching arXiv for directly relevant papers and adjacent work on coordinate-wise variance phenomena. Coordinate-wise variance collapse denotes a regime in which variability is suppressed not merely globally but at specific coordinates, so that selected coordinates become nearly fixed while others may remain variable. In the diffusion-model setting, it has been proposed as a geometric characterization of local memorization: memorized regions are those coordinates where local variance collapses abnormally, appearing as elevated diagonal curvature of the conditional log-density (Kim et al., 26 May 2026). More broadly, adjacent literatures use closely related notions—bounded variance, asymptotic variance zero, rank-1 or low-rank concentration, and coordinate-wise variance reduction—without always using the same term. Across these settings, the common structural idea is that variance collapse is a directional or coordinate-local degeneracy rather than a purely scalar reduction in total variance.
1. Conceptual definition and scope
The most explicit use of the term appears in work on diffusion-model memorization, which characterizes local memorization as a coordinate-wise collapse of variance (Kim et al., 26 May 2026). In that formulation, the issue is not merely that a sample has low intrinsic dimensionality, but that specific coordinates—pixels, latent spatial locations, or channels aggregated over spatial sites—have abnormally low variability and are effectively pinned to narrow values. The resulting geometry is localized rather than global: the same total degrees of freedom can be distributed across an entire image or concentrated in a small patch, and only the latter corresponds to local memorization in the sense of verbatim or template-like reproduction (Kim et al., 26 May 2026).
Related papers instantiate the same structural motif with different terminology. In Markov-source asymptotics, a single output coordinate exhibits bounded variance when its asymptotic variance coefficient is zero, yielding what that literature calls bounded variance or quasi-deterministic behavior rather than variance collapse (Kropf, 2015). In contrastive learning, fixed pairwise weights can drive representations to rank-1 solutions in deep linear networks, concentrating variance into a single direction; this is not called coordinate-wise variance collapse, but it is the closest basis-invariant analogue in that setting (Tian, 2022). In stochastic optimization and Langevin sampling, the relevant object is often the coordinate-wise second moment or mean-squared estimator error rather than the variance of a state variable, but the same collapse-versus-persistence distinction appears in the analysis of coordinate-selective updates (Ding et al., 2020, Shestakov et al., 6 Nov 2025).
This suggests a useful distinction between three nearby notions. First, there is coordinate-wise collapse of the underlying sample distribution, where selected coordinates have low conditional variance. Second, there is collapse of representation variance into a low-dimensional subspace, as in rank-1 contrastive solutions. Third, there is collapse of estimator error in coordinate-based optimization or sampling algorithms, where the sum of coordinate-wise squared errors contracts even if the underlying target distribution remains full-dimensional. These are mathematically distinct, but they share the same structural hallmark: variance becomes anisotropic and increasingly concentrated.
2. Geometric formulation in diffusion models
In diffusion models, the core objects are the unconditional and conditional score functions
and their Hessians
The coordinate-wise curvature signal is the diagonal of the negative Hessian,
which is used as a coordinate-level indicator of collapse (Kim et al., 26 May 2026).
The geometric intuition is Gaussian. If is Gaussian with covariance , then
Thus coordinates with small variance correspond to large curvature. The diffusion-model paper uses this as the starting point for a more general argument and, in its synthetic figure caption, writes
for a linear Gaussian model, showing that two distributions with the same intrinsic dimensionality can nevertheless have very different coordinate-wise curvature patterns (Kim et al., 26 May 2026).
The formal bridge from curvature to variance is Proposition 1 of that work. With
the paper proves
Since is typically negative semidefinite near a mode, large values of 0 correspond to lower posterior variance along coordinate 1 (Kim et al., 26 May 2026). This gives an exact covariance–curvature identity in the diffusion setting and turns coordinate-wise curvature into a variance-collapse diagnostic.
The same paper derives the identity via Tweedie’s formula: 2 which yields
3
and then
4
Combined with
5
this recovers the covariance formula above (Kim et al., 26 May 2026). In this framework, coordinate-wise variance collapse is therefore not a heuristic label but a local geometric statement about the Hessian of the log-density.
3. Baseline subtraction and the distinction between memorization and intrinsic constraint
A central complication is that coordinate-wise variance collapse alone is not sufficient evidence of overfitting. The diffusion-model paper emphasizes that some coordinates naturally have low variance because of the true data distribution or the conditioning semantics, such as prompts enforcing a black background (Kim et al., 26 May 2026). High curvature can therefore arise from intrinsic data constraints rather than memorization.
To separate these cases, the paper proposes curvature-difference methods that subtract the curvature of an underfitted baseline. With the unconditional model as baseline, the coordinate-wise metric is
6
With a less-trained checkpoint 7 as baseline, the corresponding metric is
8
The intended interpretation is that intrinsic low-variance directions should already be present in an underfitted model that captures the data manifold, while excess curvature in the final model is more plausibly attributable to overfitting-driven memorization (Kim et al., 26 May 2026).
The per-coordinate quantity is then aggregated spatially in latent space. The paper states that, in practice, it works in the latent image tensor and sums across channels to obtain a spatial map,
9
where 0 is the coordinate-wise curvature-difference tensor (Kim et al., 26 May 2026). This turns the abstract notion of coordinate-wise variance collapse into a localization method for memorized regions.
The paper further derives a score-difference proxy using the Fisher-information identity. Starting from
1
and using
2
it obtains the diagonal relation
3
This motivates the efficient proxy
4
and, for a less-trained baseline,
5
(Kim et al., 26 May 2026). The paper argues that these squared score differences approximate coordinate-wise curvature gaps, especially at late timesteps.
A plausible implication is that coordinate-wise variance collapse becomes operationally meaningful only after controlling for baseline geometry. Without that subtraction, high curvature can reflect either memorization or legitimate concentration induced by the data manifold.
4. Formal analogues in stochastic processes, optimization, and representation learning
Outside diffusion models, the same mathematical structure appears under different names.
In Markov sources, the relevant theorem concerns a single output function 6 on the transitions of a finite, finally connected and finally aperiodic Markov chain. The paper shows that
7
and that bounded variance, equivalently 8, occurs if and only if there exists a constant 9 such that
0
for every directed cycle 1 of the final component (Kropf, 2015). In the strongly connected case, this is equivalent to quasi-deterministic behavior,
2
Although that paper does not use the term coordinate-wise variance collapse, its bounded-variance criterion is exactly a one-coordinate collapse condition. In the multivariate case, the asymptotic covariance matrix 3 is singular if and only if there is a nontrivial cyclewise linear relation
4
for all cycles 5 (Kropf, 2015). This establishes a broader notion of directional collapse: not only coordinates, but arbitrary linear combinations can become asymptotically nonfluctuating.
In contrastive learning, the closest analogue is rank collapse. For fixed pairwise importance 6, the max-player in deep linear contrastive learning solves
7
with 8, so that representation learning reparameterizes PCA on a contrastive covariance matrix (Tian, 2022). Under the stated normalization assumptions and 9, almost all local maxima are globally optimal aligned-rank-1 solutions,
0
with objective value
1
(Tian, 2022). Since the end-to-end map is then rank-1, all representation variance lies in a one-dimensional subspace. In an eigenbasis aligned with that subspace, one coordinate contains all the variance and the remaining coordinates have zero variance. The paper does not call this coordinate-wise variance collapse, but it is a direct low-rank concentration result.
In coordinate-based Langevin sampling, the issue appears as persistence or suppression of estimator variance. Naive random coordinate descent uses the one-coordinate estimator
2
which is unbiased in expectation, yet the paper shows a high-variance obstruction. In a Gaussian counterexample, the per-coordinate conditional second moment satisfies
3
where 4 (Ding et al., 2020). The point is not collapse but the failure of collapse: coordinate-wise randomization induces variance that scales linearly with dimension. The proposed RCAD estimator replaces fresh one-coordinate estimation with a memory-based scheme,
5
6
and satisfies
7
(Ding et al., 2020). The memory mismatch contracts recursively,
8
so the coordinate-wise variance obstruction is controlled indirectly through decay of stale-gradient error (Ding et al., 2020).
A closely related aggregate formulation appears in adaptive variance reduction for coordinate methods. The unified recursion controls
9
and since
0
collapse of the aggregate mean-squared estimator error implies collapse of the sum of coordinate-wise contributions (Shestakov et al., 6 Nov 2025). For JAGUAR, a coordinate-biased method, the paper proves
1
while for SEGA the memory error
2
contracts at rate 3 up to movement terms (Shestakov et al., 6 Nov 2025). Under the PL condition, the Lyapunov
4
decays linearly,
5
forcing the aggregate estimator error to vanish geometrically (Shestakov et al., 6 Nov 2025). This is not collapse of a model coordinate, but of coordinate-wise estimation error.
5. Methodological estimation and algorithmic realization
The diffusion-model formulation is computationally demanding because Hessian diagonals are intractable in full. The paper therefore estimates diagonal curvature differences with the Hutchinson trick: 6 For
7
the diagonal estimator is
8
and analogously for the less-trained baseline (Kim et al., 26 May 2026). Only Hessian-vector products are needed, which makes the approach computationally feasible even though it is heavier than score differences.
The practical localization pipeline in that paper is explicit. It samples a latent state 9 during generation, computes a score difference against either the unconditional or less-trained baseline, forms either a squared score-difference tensor or a Hutchinson-based curvature-difference tensor, sums across channels, and returns the resulting spatial map 0 (Kim et al., 26 May 2026). For evaluation against masks, the maps are resized to 1, globally normalized to 2 independently for each metric over all evaluation samples, and thresholded by sweeping a uniform threshold across 3 (Kim et al., 26 May 2026). For visualization, values are clipped at the 99th percentile; for 4, negative values are clipped to zero; and for 5, a 6 mean filter is reported to suppress sparse outlier pixels and improve IoU (Kim et al., 26 May 2026).
Algorithmically, adjacent literatures realize coordinate-wise collapse through memory refresh or coordinate-specific updates. In RCAD-LMC, only one coordinate of the stored finite-difference table is refreshed per iteration, and the control-variate identity
7
reduces variance by preserving past coordinate information (Ding et al., 2020). In SEGA and JAGUAR, only sampled coordinates are refreshed, while unsampled coordinates remain stale. For JAGUAR,
8
so sampled coordinates collapse immediately to zero estimator error, while the aggregate error contracts at a rate determined by the sampling frequency 9 (Shestakov et al., 6 Nov 2025). This suggests a general methodological pattern: coordinate-wise collapse is often implemented not by direct diagonal variance penalization, but by selectively refreshing or contrasting coordinates against a baseline state.
6. Empirical behavior, interpretations, and limitations
The most direct empirical evidence comes from the diffusion-model memorization study. There, raw curvature
0
often highlights both memorized and intrinsically simple non-memorized regions, confirming that coordinate-wise variance collapse alone produces false positives (Kim et al., 26 May 2026). Difference-based maps
1
are more selective. On Stable Diffusion v1.4, template-verbatim only, Table 1 reports IoU 2 for raw curvature, 3 for 4, and 5 for 6; on SD v1.4, all examples, it reports 7 for the prior attention-based BE baseline and 8 for 9; on SD v2.1, template-verbatim only, it reports 0 for BE, 1 for 2, and 3 for 4 (Kim et al., 26 May 2026). When localization maps are spatially averaged into scalar detection scores, the paper reports AUC 5 for both 6 and 7 on SD v1.4, and AUC 8 and 9, respectively, on SD v2.1 (Kim et al., 26 May 2026). These results support the claim that baseline-subtracted coordinate-wise collapse is informative both locally and globally.
The paper also reports that late denoising steps work much better than early noisy steps, that even 0 Hutchinson sample is competitive, and that both unconditional and less-trained baselines are effective, including SD v1.2 as a relatively more trained baseline (Kim et al., 26 May 2026). A plausible implication is that coordinate-wise variance collapse is easiest to observe when diffusion noise is low enough that model-induced curvature is not hidden by the noise floor.
The optimization and sampling papers present a contrasting empirical message. In RCAD-LMC, naive coordinate randomization is harmful because its high variance destroys the hoped-for computational gain, while RCAD restores favorable total directional-derivative complexity through recursive variance control (Ding et al., 2020). In unified adaptive variance reduction, experiments are consistent with improved convergence of coordinate methods under adaptive stepsizes, but the paper does not report per-coordinate variance, diagonal covariance, or coordinate-wise mean-squared error trajectories (Shestakov et al., 6 Nov 2025). Likewise, the contrastive-learning paper reports competitive or improved performance of new 1-CL losses on CIFAR-10, STL-10, and CIFAR-100, but it does not measure covariance spectra or per-coordinate variances directly (Tian, 2022). These omissions matter: several adjacent literatures give rigorous structural results closely related to coordinate-wise variance collapse, yet still leave the coordinate marginals empirically unobserved.
The main misconceptions addressed by this body of work are therefore twofold. First, low variance in a coordinate is not by itself evidence of pathological memorization, because intrinsic data constraints can induce the same geometry (Kim et al., 26 May 2026). Second, global low-dimensionality does not imply local coordinate-wise collapse; a system may have few degrees of freedom distributed across the entire object rather than concentrated in a specific region (Kim et al., 26 May 2026). Conversely, a singular covariance matrix or rank-1 representation indicates directional collapse, but not necessarily collapse in the original coordinate basis (Tian, 2022, Kropf, 2015).
The principal limitation across the literature is that the strongest results are usually aggregate or basis-invariant rather than explicitly diagonal. Diffusion memorization work is exceptional in making the diagonal curvature itself the object of study (Kim et al., 26 May 2026). Markov-source theory gives exact cycle criteria for bounded-variance coordinates but does not localize collapse spatially (Kropf, 2015). Coordinate-variance-reduction papers control sums of coordinate-wise errors, not individual coordinate recursions in full generality (Ding et al., 2020, Shestakov et al., 6 Nov 2025). This suggests that coordinate-wise variance collapse is presently best understood as a family of related degeneracy phenomena rather than a single standardized object.
A plausible synthesis is that the term now designates a specific diagonalized viewpoint on concentration: variance collapse becomes meaningful when one asks not only whether variability is low, but where it is low, relative to what baseline, and whether the low-variance coordinates reflect data geometry, learned overfitting, or algorithmic error suppression.