SDSN: Scaling Doubly Stochastic Normalization
- SDSN is a family of methods that transforms nonnegative matrices into doubly stochastic matrices, ensuring balanced row and column sums for robust mass conservation.
- It is applied across optimal transport, manifold learning, graph neural networks, and clustering, offering scalable solutions that preserve key spectral properties.
- Design trade-offs in SDSN, including entropy regularization and iteration limits, are crucial to preventing spectral collapse while maintaining informative geometric detail.
Scaling Doubly Stochastic Normalization (SDSN) denotes a family of procedures that transform a nonnegative matrix into a doubly stochastic matrix, typically enforcing
or the corresponding balanced-marginal variant used in transport. Across recent work, SDSN appears in matrix scaling, optimal transport, manifold learning, graph neural networks, subspace clustering, visualization, Transformer attention, and graph matching. The common attraction is that doubly stochastic structure provides balanced mass conservation, numerical regularity, and a convex feasible set; the central caveat is that some implementations, especially high-entropy Sinkhorn projections, can induce spectral collapse and loss of expressive detail (Nathanson, 2019, Liu, 5 Jan 2026).
1. Classical matrix scaling and the doubly stochastic constraint
A standard formalization uses the set
which is the nonnegative row- and column-stochastic feasible set used throughout the recent SDSN literature (Liu, 5 Jan 2026). In the classical matrix-scaling setting, one starts from a strictly positive matrix and alternates row scaling and column scaling. Nathanson summarizes the corresponding operators and , where and are diagonal matrices built from inverse row sums and inverse column sums, respectively; the alternating process converges to a unique doubly stochastic matrix , the Sinkhorn limit, with uniqueness up to the trivial scalar ambiguity . In the symmetric case one may take , so the limit has the form 0 (Nathanson, 2019).
This convergence statement is complemented by support conditions in later summaries. For nonnegative matrices with total support or full indecomposability, Sinkhorn–Knopp converges geometrically to the unique doubly stochastic scaling; sparse implementations preserve zeros exactly and can therefore be used on 1-NN or graph-derived matrices without densifying them (Lu et al., 2016). Nathanson also emphasizes that explicit Sinkhorn limits are rare in closed form: even among symmetric 2 two-value families, some cases reduce to quadratic formulas, others to cubic radicals, and at least one family leads to an octic polynomial (Nathanson, 2019). This establishes SDSN as both an algorithmic procedure and an algebraic object.
Lu et al. describe an alternative “fast” normalization for asymmetric or bipartite affinities. Starting from an 3 nonnegative matrix 4, one first row-normalizes it to 5, then forms
6
The resulting 7 is symmetric and doubly stochastic, and no further iteration is required (Lu et al., 2016).
2. Variational formulations: KL projection, entropic transport, Gaussian kernels, and Frobenius regularization
A major modern interpretation of SDSN is variational. In Sinkhorn-based architectures, the projection of a nonnegative kernel 8 onto 9 is written as
0
or, equivalently, as an entropic optimal transport problem
1
The practical solver is the Sinkhorn iteration, alternating row normalization and column normalization on 2 until convergence (Liu, 5 Jan 2026). In this formulation, doubly stochasticity is not merely a constraint; it is the output of an entropy-regularized projection.
In manifold learning, SDSN is imposed on Gaussian affinity matrices. For noisy observations 3, one defines
4
and seeks positive diagonal scalings 5 so that 6 is doubly stochastic, or, in the symmetric form, scalars 7 such that 8 and 9 (Landa et al., 2022). The corresponding population object is a positive continuous function 0 solving an integral equation on the manifold. The paper derives concentration of finite-sample scaling factors around this population form under high-dimensional sub-Gaussian noise, and then uses the scaled matrix for robust density estimation, noise inference, clean-distance estimation, and graph Laplacian construction (Landa et al., 2022).
Landa and Cheng extend the same Gaussian-kernel scaling to manifolds with boundary. Their continuous scaling factor 1 satisfies, to second order in 2, a quadratic equation with coefficients depending on the boundary distance 3, boundary mean curvature 4, and explicit Gaussian moments 5. They further derive a first-order boundary correction in which 6, with 7 an explicit strictly decreasing convex function of 8 (Kohli et al., 2024).
A distinct variational route appears in graph matching. In FRAM, SDSN solves exactly
9
equivalently
0
where 1 is the doubly stochastic set. This Frobenius-Regularized Linear Assignment (FRA) formulation is introduced specifically to remove the scale sensitivity of the ordinary Euclidean projection onto 2 (Shen et al., 26 Jul 2025). This suggests that SDSN is best understood as a constraint class instantiated by multiple objective functions rather than as a single normalization rule.
3. Computational scaling and algorithmic realizations
Recent SDSN work is largely driven by the cost of exact normalization. Dense Sinkhorn on an 3 matrix is intrinsically quadratic per iteration, and this becomes the main systems bottleneck in long-sequence attention or large-graph learning. The literature therefore develops sparse, truncated, low-rank, one-shot, and mixed-precision alternatives.
| Setting | Normalized object | Stated scaling |
|---|---|---|
| Dense Gaussian-kernel Sinkhorn | 4 or 5 | 6 per iteration; usually 7 iterations (Landa et al., 2022) |
| Sparse Sinkhorn / DOSNES scaling | Sparse similarity matrix | 8; typically 9 for 0–1 tolerance (Lu et al., 2016) |
| DsmNet / DsmNet-compensate | 2 | 3 time, 4 memory (Hu et al., 16 Apr 2026) |
| A-DSSC active-set projection | Doubly stochastic affinity on support 5 | 6 per dual evaluation; near-linear if final affinity has 7 nonzeros (Lim et al., 2020) |
| ESPFormer | Expected sliced transport plan 8 | 9, fully parallel over slices (Shahbazi et al., 11 Feb 2025) |
| LOTFormer | Low-rank glued coupling | 0 time and memory, with 1 (Shahbazi et al., 27 Sep 2025) |
| FRAM SDSN | FRA projection | 2; mixed precision gives up to 3 speedup over CPU-FP64 (Shen et al., 26 Jul 2025) |
Hu et al. replace an exact inverse 4, whose direct computation is 5, by a truncated Neumann expansion
6
which can be applied to features through sequential sparse multiplications in 7 time (Hu et al., 16 Apr 2026). Because truncation leaks probability mass, they add a residual correction
8
thereby restoring exact row-stochasticity while keeping the hard 9-hop cutoff off the diagonal (Hu et al., 16 Apr 2026).
In subspace clustering, the active-set solver for A-DSSC exploits the sparsity of the final affinity. Each restricted dual solve costs 0 rather than 1, and if the final support has 2 nonzeros the overall routine becomes near-linear (Lim et al., 2020). In attention, ESPFormer replaces iterative Sinkhorn with a one-shot aggregation of sliced transport plans, while LOTFormer factorizes the transport through a learned pivot measure of size 3, so that one never materializes the full 4 attention map (Shahbazi et al., 11 Feb 2025, Shahbazi et al., 27 Sep 2025). In FRAM, the SDSN subroutine itself remains 5 per iteration, but the authors place different steps in FP64, TF32, and FP32 to obtain substantial hardware acceleration without changing the converged SDSN solution (Shen et al., 26 Jul 2025).
4. Spectral behavior, effective depth, and the “Homogeneity Trap”
The most explicit spectral analysis is developed for Sinkhorn-projected deep networks. Once a layer’s mixing matrix 6 is doubly stochastic, Perron–Frobenius guarantees spectral radius 7 with eigenvector 8, and all remaining singular values lie in 9. The key quantity is the subdominant singular value
0
which governs contraction on the detail subspace 1. Repeated application satisfies 2, leading to the spectral effective-depth scale
3
The central claim is that maximum-entropy Sinkhorn projections drive 4 toward the uniform barycenter 5, forcing 6 and thus collapsing effective depth (Liu, 5 Jan 2026).
This collapse has a Fourier interpretation. For a doubly stochastic operator on the complete graph, all nonconstant harmonics lie in the detail subspace and are contracted by at most 7. As 8 in the entropy maximization, or as the Sinkhorn temperature 9 grows, 0, so the mean mode survives and all higher-frequency modes are annihilated. The paper terms this behavior the “Homogeneity Trap”: repeated mixing does not enlarge a useful receptive field, but instead removes detail until only the average remains (Liu, 5 Jan 2026).
The same work also analyzes LayerNorm in a noise-dominated regime. Writing the pre-normalized feature as
1
the detail-subspace SNR is
2
Their finite-3 bound states that once 4, LayerNorm produces nearly orthogonal outputs from distinct inputs with overwhelming probability; affine rescaling cannot restore the lost direction because it rescales signal and noise equally (Liu, 5 Jan 2026).
Hu et al. obtain a closely related but graph-specific picture. For the exact operator 5, repeated application empirically collapses Dirichlet energy to zero, which they identify as over-smoothing. By contrast, 6 plateaus too soon because of mass leakage, yielding under-mixing, while the compensated 7 maintains a high nonzero energy floor. They also characterize the second-largest eigenvalue of the compensated operator, with 8 as 9 and 00 as 01 (Hu et al., 16 Apr 2026). Taken together, these analyses show that doubly stochasticity alone does not determine the usable spectral regime; the construction method and regularization strength do.
5. Geometry, density, boundary inference, and graph diffusion
In manifold inference, doubly stochastic Gaussian kernels are used as a robust alternative to standard row-stochastic normalization. The core result of Landa and Cheng is that, under a compact 02-dimensional manifold model with high-dimensional sub-Gaussian noise, the doubly stochastic affinity matrix and its scaling factors concentrate around population forms with explicit finite-sample probabilistic error bounds. From this, they derive a robust density estimator, estimators for pointwise noise magnitude, pointwise signal magnitude, and pairwise clean Euclidean distances, and a family of robust graph Laplacians 03 that converge to a Schrödinger-type operator; the choice 04 yields the Laplace–Beltrami operator (Landa et al., 2022).
The boundary-estimation extension sharpens the geometric role of the scaling factor. Landa and Cheng show that the continuous analogue 05 depends systematically on boundary distance and curvature, and they combine the doubly stochastic kernel with local PCA to estimate a boundary direction 06 and detect boundary points by thresholding 07. Their simulations show that the resulting boundary-point estimates outperform the standard Gaussian-kernel approach, particularly under noisy conditions (Kohli et al., 2024).
In graph neural networks, Hu et al. derive a different doubly stochastic operator directly from graph topology: 08 This 09 is doubly stochastic and encodes continuous multi-hop proximity together with strict local centrality. The associated DsmNet and DsmNet-compensate architectures operate in 10 time, mitigate over-smoothing by bounding Dirichlet-energy decay, perform strongly on homophilic benchmarks, and can also supply continuous structural encodings for Graph Transformers. The same paper also states the theoretical boundaries of such DSM diffusion on heterophilic topologies, where any low-pass diffusion operator suffers performance drops and DsmNet variants help only marginally (Hu et al., 16 Apr 2026).
6. Clustering, visualization, attention, and assignment matching
In subspace clustering, doubly stochastic normalization is used not merely as post-processing but as part of the learning objective. The feasible set
11
appears in J-DSSC, a convex joint model that learns both a self-expressive representation and a normalized affinity. The paper identifies several consequences of enforcing 12: exact Laplacian normalization for symmetric 13, row/column balancing, a connectivity guarantee because every row sums to one, and a convex feasible set. A faster approximation, A-DSSC, separates the self-expressive step from the doubly stochastic projection and solves the latter with a sparsity-aware active-set dual method; on GPU, full A-DSSC on MNIST with 14 takes less than 15 seconds to reach 16 clustering accuracy (Lim et al., 2020).
In data visualization, Lu et al. use doubly stochastic normalization to equalize total similarities before running SNE. They report that the doubly stochasticity constraint often leads to embeddings that are approximately spherical, motivating the replacement of a flat embedding space by spheres. In their account, the spherical embedding eliminates the discrepancy between center and periphery and thereby resolves the crowding problem caused by highly imbalanced similarities (Lu et al., 2016).
In Transformer attention, SDSN is used to balance incoming and outgoing token mass. ESPFormer constructs queries and keys as empirical measures, projects them along 17 directions, computes one-dimensional transport plans, and aggregates them into
18
The paper states that each slice plan 19 is doubly stochastic and that the convex combination 20 therefore remains in the transport polytope. Because the slices can be processed fully in parallel, the stated complexity is 21, and the reported speedups over Sinkhorn-based attention on long sequences are 22–23 (Shahbazi et al., 11 Feb 2025).
LOTFormer attacks the same problem through low-rank entropic transport. With a learnable pivot measure 24, it solves two entropic OT subproblems, 25 and 26, and glues them into
27
Under uniform query and key marginals, the paper verifies 28 and 29; after rescaling by 30, the attention matrix is exactly doubly stochastic. The same factorization yields 31 and 32 application to values. Empirically, LOTFormer reports a 33 average on Long Range Arena with depthwise convolution, 34 top-1 on ImageNet-1K with a DeiT-Tiny backbone at 35 resolution, and 36 BLEU on IWSLT’14 De37En after fine-tuning (Shahbazi et al., 27 Sep 2025).
In graph matching, FRAM uses SDSN as the inner solver for Frobenius-regularized assignment matching. Its alternating-projection scheme splits the problem into 38, which enforces row and column sums but allows negatives, and 39, which enforces nonnegativity. The resulting SDSN solution is scale-invariant because the input is normalized by 40 before solving the FRA problem. The paper also proves an assignment-gap bound 41, identifies the barycentric limit as 42, and shows convergence to the minimum-Frobenius-norm point of the optimal face as 43. With GPU mixed precision, the full FRAM system achieves up to 44 speedup over its CPU-FP64 counterpart, with negligible loss in solution accuracy (Shen et al., 26 Jul 2025).
7. Design trade-offs, parameter regimes, and recurrent misconceptions
The most important practical trade-off concerns entropy. In the Sinkhorn-based deep-network analysis, increasing the entropy strength 45, raising Sinkhorn temperature 46, or letting the number of Sinkhorn iterations 47 improves numerical stability but shrinks 48 and shortens effective depth. The paper therefore recommends moderate 49, monitoring 50 against 51, and choosing the “elbow” before 52 plummets. It also recommends finite Sinkhorn iteration counts, explicitly suggesting early stopping such as 53 instead of 54, small learnable post-projection scaling that keeps the underlying SNR above the 55 threshold, and interleaving DSM blocks with non-DSM expressive layers such as un-normalized self-attention or residual MLPs (Liu, 5 Jan 2026).
Graph diffusion models exhibit an analogous locality-versus-mixing trade-off. In DsmNet, 56 controls the transition from local 57-hop structure to global diffusion; on homophilic benchmarks such as Cora, CiteSeer, PubMed, Amazon, and Coauthor, 58 often maximizes accuracy, while on heterophilic datasets including Chameleon, Squirrel, Texas, Wisconsin, and Cornell, any low-pass diffusion operator suffers performance drops. The same paper notes that residual mass compensation always improves centrality preservation and helps keep node representations distinct in deep layers (Hu et al., 16 Apr 2026). A related misconception is that exact normalization by itself guarantees preservation of informative geometry; the heterophily results show that normalization can still be mismatched to the graph regime.
Kernel-based SDSN carries its own parameter sensitivities. In the manifold and boundary papers, bandwidth is tied to the usual bias-variance and locality-noise balance: 59 may be chosen as the median 60-th nearest-neighbor distance, tolerances of 61–62 are described as adequate for Sinkhorn stopping, and practical numerical stabilization includes log-domain computation, per-row rescaling, or discarding pairs beyond a fixed multiple of 63 when the Gaussian kernel underflows (Kohli et al., 2024, Landa et al., 2022). These recommendations indicate that SDSN is rarely a drop-in normalization; it is a parameterized operator whose spectral, geometric, and numerical behavior must be co-designed with the downstream task.
A recurrent misconception in deep learning is that post-normalization layers can repair whatever information doubly stochastic mixing suppresses. The finite-64 SNR theorem for LayerNorm directly contradicts that view in noise-dominated regimes, and the GNN diffusion results similarly show that exact normalization can yield over-smoothing unless the operator is truncated or compensated (Liu, 5 Jan 2026, Hu et al., 16 Apr 2026). The broader implication is that the central question in SDSN is not whether to impose doubly stochasticity, but how to do so without collapsing the nontrivial spectrum that carries task-relevant structure.