Abstract:We present a novel set of rigorous and computationally efficient topology-based complexity notions that exhibit a strong correlation with the generalization gap in modern deep neural networks (DNNs). DNNs show remarkable generalization properties, yet the source of these capabilities remains elusive, defying the established statistical learning theory. Recent studies have revealed that properties of training trajectories can be indicative of generalization. Building on this insight, state-of-the-art methods have leveraged the topology of these trajectories, particularly their fractal dimension, to quantify generalization. Most existing works compute this quantity by assuming continuous- or infinite-time training dynamics, complicating the development of practical estimators capable of accurately predicting generalization without access to test data. In this paper, we respect the discrete-time nature of training trajectories and investigate the underlying topological quantities that can be amenable to topological data analysis tools. This leads to a new family of reliable topological complexity measures that provably bound the generalization error, eliminating the need for restrictive geometric assumptions. These measures are computationally friendly, enabling us to propose simple yet effective algorithms for computing generalization indices. Moreover, our flexible framework can be extended to different domains, tasks, and architectures. Our experimental results demonstrate that our new complexity measures correlate highly with generalization error in industry-standards architectures such as transformers and deep graph networks. Our approach consistently outperforms existing topological bounds across a wide range of datasets, models, and optimizers, highlighting the practical relevance and effectiveness of our complexity measures.

On Characterizing the Evolution of Embedding Space of Neural Networks using Algebraic Topology

Functional Loops: Monitoring Functional Organization of Deep Neural Networks Using Algebraic Topology

Experimental Observations of the Topology of Convolutional Neural Network Activations

Topological exploration of artificial neuronal network dynamics

Topological Measurement of Deep Neural Networks Using Persistent Homology

Topological Data Analysis of Neural Network Layer Representations

Topological Generalization Bounds for Discrete-Time Stochastic Optimization Algorithms

Topological learning in multiclass data sets

A rank decomposition for the topological classification of neural representations

Characterization of topological structures in different neural network architectures

Geometric and Topological Inference for Deep Representations of Complex Networks

Exposition and Interpretation of the Topology of Neural Networks

Algebraic Topological Networks via the Persistent Local Homology Sheaf

The importance of the whole: Topological data analysis for the network neuroscientist

Topological Learning in Multi-Class Data Sets

A Comprehensive Review of Deep Neural Network Interpretation Using Topological Data Analysis

Two's company, three (or more) is a simplex: Algebraic-topological tools for understanding higher-order structure in neural data

A topological description of loss surfaces based on Betti Numbers

Intrinsic Dimension, Persistent Homology and Generalization in Neural Networks

Beyond Multilayer Perceptrons: Investigating Complex Topologies in Neural Networks

Attending to Topological Spaces: The Cellular Transformer