Sinkhorn Normalization of Diffusion Kernels
Not on the map. Why?
Abstract (arXiv)
Smoothing a signal based on local neighborhoods is a core operation in machine learning and geometry processing. On well-structured domains such as vector spaces and manifolds, the Laplace operator derived from differential geometry offers a principled approach to smoothing via heat diffusion, with strong theoretical guarantees. However, constructing such Laplacians requires a carefully defined domain structure, which is not always available. Most practitioners thus rely on simple convolution kernels and message-passing layers, which are biased against the boundaries of the domain. We bridge this gap by introducing a broad class of smoothing operators, derived from general similarity or adjacency matrices, and demonstrate that they can be normalized into diffusion-like operators that inherit desirable properties from Laplacians. Our approach relies on a symmetric variant of the Sinkhorn algorithm, which rescales positive smoothing operators to match the structural behavior of heat diffusion. This construction enables Laplacian-like smoothing and processing of irregular data such as point clouds, sparse voxel grids or mixture of Gaussians. We show that the resulting operators not only approximate heat diffusion but also retain spectral information from the Laplacian itself, with applications to shape analysis and matching.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 26 of its 91 references that are in the map. First posted on arXiv 8 Jul 2025.
Builds on
26 of 91 references.
References · 26
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2017Graph Attention Networkscited by 8,923
- 2017Inductive Representation Learning on Large Graphscited by 6,320
- 20233D Gaussian Splatting for Real-Time Radiance Field Renderingcited by 6,226
- 2017Neural Message Passing for Quantum Chemistrycited by 3,487
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
- 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,316
- 2021Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gaugescited by 1,167
- 2022Diffusion Models: A Comprehensive Survey of Methods and Applicationscited by 1,114
- 2018Computational Optimal Transportcited by 780
- 2016Stein Variational Gradient Descent: A General Purpose Bayesian Inference Algorithmcited by 507
- 2019Maximum Mean Discrepancy Gradient Flowcited by 127
14 more
- 2020Deep Geometric Functional Maps: Robust Feature Learning for Shape Correspondencecited by 125
- 2020Kernel Operations on the GPU, with Autodiff, without Memory Overflowscited by 104
- 2018Continuous and Orientation-preserving Correspondences via Functional Mapscited by 77
- 2021DPFM: Deep Partial Functional Mapscited by 58
- 2023Unsupervised Learning of Robust Spectral Shape Matchingcited by 57
- 2018The Vector Heat Methodcited by 31
- 2020Spectral convergence of diffusion maps: improved error bounds and an alternative normalisationcited by 26
- 2017Efficient Deformable Shape Correspondence via Kernel Matchingcited by 25
- 2017Manifold learning with bi-stochastic kernelscited by 20
- 2022TIDE: Time Derivative Diffusion for Deep Learning on Graphscited by 17
- 2022Bi-stochastically normalized graph Laplacian: convergence to manifold Laplacian and robustness to outlier noisecited by 7
- 2024Synchronous Diffusion for Unsupervised Smooth Non-Rigid 3D Shape Matchingcited by 7
- 2025Laplace-Beltrami Operator for Gaussian Splattingcited by 5
- 2024An Intrinsic Vector Heat Networkcited by 1
Built on by
Not cited yet.