Diffusion Operator Geometry of Feedforward Representations
Not on the map. Why?
Abstract (arXiv)
Feedforward neural networks transform data through learned representations whose geometry shapes how classes separate and relate across successive layers. We study that geometry through diffusion operators. Each feature-cloud snapshot is assigned a Gaussian-kernel Markov operator, giving a smooth description of one-step transport between classes from which spectral, boundary, and local geometric information can be read. We define the empirical class chain, state the condition under which it is an exact Markov quotient, and derive both the corresponding population transition and a simpler overlap chain based on expected class affinities. For balanced shared-covariance Gaussian class-conditional snapshots these affinities have closed forms controlled by a regularized Mahalanobis separation, which yields explicit expressions for leakage and coarse spectral behaviour. We further show that operator observables vary smoothly under feature perturbations, whereas hard neighborhood graphs are controlled by neighbor-order margins. Experiments on CIFAR-10 and CIFAR-100 ResNet-18 representations find that class transport becomes increasingly persistent with depth while retaining structured relations between classes, and that the diffusion class chain is more stable than its k-nearest-neighbor counterpart under matched perturbations.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 28 of its 57 references that are in the map. First posted on arXiv 1 May 2026.
Builds on
28 of 57 references.
References · 28
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2019Similarity of Neural Network Representations Revisitedcited by 1,755
- 2016Understanding intermediate layers using linear classifier probescited by 1,353
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 719
- 2017Deep Neural Networks as Gaussian Processescited by 622
- 2016Kernel Mean Embedding of Distributions: A Review and Beyondcited by 364
- 2018Insights on representational similarity in neural networks with canonical correlationcited by 324
- 2019Intrinsic dimension of data representations in deep neural networkscited by 282
- 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 227
- 2016Deep Information Propagationcited by 195
- 2023A mathematical perspective on Transformerscited by 164
16 more
- 2021Neural Collapse Under MSE Loss: Proximity to and Dynamics on the Central Pathcited by 158
- 2023The emergence of clusters in self-attention dynamicscited by 140
- 2016Toward Deeper Understanding of Neural Networks: The Power of Initialization and a Dual View on Expressivitycited by 138
- 2020Topology of deep neural networkscited by 112
- 2022Self-Consistent Dynamical Field Theory of Kernel Evolution in Wide Neural Networkscited by 105
- 2019Community Detection on Networks with Ricci Flowcited by 62
- 2024Clustering in Causal Attention Maskingcited by 32
- 2025The Mean-Field Dynamics of Transformerscited by 30
- 2020Lipschitz regularity of graph Laplacians on random data cloudscited by 26
- 2023Assessing Neural Network Representations During Training Using Noise-Resilient Diffusion Spectral Entropycited by 13
- 2020Ollivier-Ricci curvature convergence in random geometric graphscited by 11
- 2026Computing Diffusion Geometrycited by 7
- 2023Continuum Limits of Ollivier's Ricci Curvature on data clouds: pointwise consistency and global lower boundscited by 6
- 2024Deep Learning as Ricci Flowcited by 4
- 2024Exploring the Manifold of Neural Networks Using Diffusion Geometrycited by 2
- 2025Neural Feature Geometry Evolves as Discrete Ricci Flowcited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.