Neural Persistence: A Complexity Measure for Deep Neural Networks Using Algebraic Topology

Code

• github.com/BorgwardtLab/Neural-Persistence
  OfficialIn papertf★ 0
• github.com/explainableml/deep-graph-persistence
  pytorch★ 4

Abstract

While many approaches to make neural networks more fathomable have been proposed, they are restricted to interrogating the network with input data. Measures for characterizing and monitoring structural properties, however, have not been developed. In this work, we propose neural persistence, a complexity measure for neural network architectures based on topological data analysis on weighted stratified graphs. To demonstrate the usefulness of our approach, we show that neural persistence reflects best practices developed in the deep learning community such as dropout and batch normalization. Moreover, we derive a neural persistence-based stopping criterion that shortens the training process while achieving comparable accuracies as early stopping based on validation loss.

Tasks

Reproductions