Effects of Data Geometry in Early Deep Learning
Not on the map. Why?
Abstract (arXiv)
Deep neural networks can approximate functions on different types of data, from images to graphs, with varied underlying structure. This underlying structure can be viewed as the geometry of the data manifold. By extending recent advances in the theoretical understanding of neural networks, we study how a randomly initialized neural network with piece-wise linear activation splits the data manifold into regions where the neural network behaves as a linear function. We derive bounds on the density of boundary of linear regions and the distance to these boundaries on the data manifold. This leads to insights into the expressivity of randomly initialized deep neural networks on non-Euclidean data sets. We empirically corroborate our theoretical results using a toy supervised learning problem. Our experiments demonstrate that number of linear regions varies across manifolds and the results hold with changing neural network architectures. We further demonstrate how the complexity of linear regions is different on the low dimensional manifold of images as compared to the Euclidean space, using the MetFaces dataset.
Key facts. Cited by 12 papers in the map (data through 7 Oct 2026). Builds on 56 of its 87 references that are in the map. First posted on arXiv 29 Dec 2022.
Builds on
56 of 87 references.
References · 56
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2014Explaining and Harnessing Adversarial Examplescited by 7,287
- 2018A Style-Based Generator Architecture for Generative Adversarial Networkscited by 5,986
- 2016PointNet: Deep Learning on Point Sets for 3D Classification and Segmentationcited by 5,223
- 2013Intriguing properties of neural networkscited by 5,114
- 2019Analyzing and Improving the Image Quality of StyleGANcited by 3,580
- 2019A Comprehensive Survey on Graph Neural Networkscited by 2,906
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2018Dynamic Graph CNN for Learning on Point Cloudscited by 2,220
- 2016Understanding deep learning requires rethinking generalizationcited by 1,657
- 2018On the Spectral Bias of Neural Networkscited by 1,321
38 more
- 2020Training Generative Adversarial Networks with Limited Datacited by 1,184
- 2021Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gaugescited by 1,131
- 2016Geometric deep learning: going beyond Euclidean datacited by 1,042
- 2017A Closer Look at Memorization in Deep Networkscited by 936
- 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 784
- 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 748
- 2019Wide Neural Networks of Any Depth Evolve as Linear Models Under Gradient Descentcited by 732
- 2016Geometric deep learning on graphs and manifolds using mixture model CNNscited by 468
- 2019Implicit Regularization in Deep Matrix Factorizationcited by 402
- 2018Hyperbolic Neural Networkscited by 393
- 2014On the Number of Linear Regions of Deep Neural Networkscited by 378
- 2017A PAC-Bayesian Approach to Spectrally-Normalized Margin Bounds for Neural Networkscited by 309
- 2018Stronger generalization bounds for deep nets via a compression approachcited by 306
- 2016On the Expressive Power of Deep Neural Networkscited by 302
- 2017Empirical Analysis of the Hessian of Over-Parametrized Neural Networkscited by 283
- 2016Exponential expressivity in deep neural networks through transient chaoscited by 260
- 2015Deep Convolutional Networks on Graph-Structured Datacited by 246
- 2015A Differential Equation for Modeling Nesterov's Accelerated Gradient Method: Theory and Insightscited by 203
- 2018Sensitivity and Generalization in Neural Networks: an Empirical Studycited by 187
- 2014Qualitatively characterizing neural network optimization problemscited by 186
- 2018A Convergence Analysis of Gradient Descent for Deep Linear Neural Networkscited by 180
- 2018Algorithmic Regularization in Learning Deep Homogeneous Models: Layers are Automatically Balancedcited by 172
- 2019Complexity of Linear Regions in Deep Networkscited by 166
- 2017Bounding and Counting Linear Regions of Deep Neural Networkscited by 161
- 2019Deep ReLU Networks Have Surprisingly Few Activation Patternscited by 160
- 2017Generalization in Deep Learningcited by 146
- 2017A Bayesian Perspective on Generalization and Stochastic Gradient Descentcited by 134
- 2020Flows for simultaneous manifold learning and density estimationcited by 117
- 2019Finite Depth and Width Corrections to the Neural Tangent Kernelcited by 114
- 2017Universal Function Approximation by Deep Neural Nets with Bounded Width and ReLU Activationscited by 104
- 2017Fisher-Rao Metric, Geometry, and Complexity of Neural Networkscited by 99
- 2020Neural Networks Fail to Learn Periodic Functions and How to Fix Itcited by 97
- 2020Neural Mechanics: Symmetry and Broken Conservation Laws in Deep Learning Dynamicscited by 93
- 2015Representation Benefits of Deep Feedforward Networkscited by 84
- 2019Deep ReLU network approximation of functions on a manifoldcited by 73
- 2016Deep vs. shallow networks : An approximation theory perspectivecited by 63
- 2015Provable approximation properties for deep neural networkscited by 60
- 2018Products of Many Large Random Matrices and Gradients in Deep Neural Networkscited by 49
50 most cited of 56 All 56 (JSON)
Built on by
Citing papers per year: 2023: 3; 2024: 2; 2025: 2; 2026: 5.
12 papers.
- 2023When Deep Learning Meets Polyhedral Theory: A Surveycited by 39
- 2025Geometry of Neural Reinforcement Learning in Continuous State and Action Spacescited by 3
- 2024Defining Neural Network Architecture through Polytope Structures of Datasetcited by 2
- 2025Generalization Below the Edge of Stability: The Role of Data Geometrycited by 2
- 2023Data Topology-Dependent Upper Bounds of Neural Network Widthscited by 1
- 2023On original and latent space connectivity in deep neural networkscited by 1
- 2026Training-Time Batch Normalization Reshapes Local Partition Geometry in Piecewise-Affine Networkscited by 1
- 2026AffineLens: Capturing the Continuous Piecewise Affine Functions of Neural Networkscited by 1
- 2026Region Seeding via Pre-Activation Regularization: A Geometric View of Piecewise Affine Neural Networkscited by 1
- 2026The Geometry of Projection Heads: Conditioning, Invariance, and Collapsecited by 1
- 2024Data-induced multiscale losses and efficient multirate gradient descent schemescited by 0
- 2026Expressivity Saturation: Reduced Affine Region Usage Under Increasing Task Complexitycited by 0
- 2026Expressivity Saturation: Reduced Affine Region Usage Under Increasing Task Complexity19 Jun 2026
- 2026The Geometry of Projection Heads: Conditioning, Invariance, and Collapse16 May 2026
- 2026Region Seeding via Pre-Activation Regularization: A Geometric View of Piecewise Affine Neural Networks7 May 2026
- 2026AffineLens: Capturing the Continuous Piecewise Affine Functions of Neural Networks7 May 2026
- 2026Training-Time Batch Normalization Reshapes Local Partition Geometry in Piecewise-Affine Networks6 May 2026
- 2025Generalization Below the Edge of Stability: The Role of Data Geometry20 Oct 2025
- 2025Geometry of Neural Reinforcement Learning in Continuous State and Action Spaces28 Jul 2025
- 2024Data-induced multiscale losses and efficient multirate gradient descent schemes5 Feb 2024
- 2024Defining Neural Network Architecture through Polytope Structures of Dataset4 Feb 2024
- 2023On original and latent space connectivity in deep neural networks12 Nov 2023
- 2023Data Topology-Dependent Upper Bounds of Neural Network Widths25 May 2023
- 2023When Deep Learning Meets Polyhedral Theory: A Survey29 Apr 2023