Topological Data Analysis of Decision Boundaries with Application to Model Selection

Code

• github.com/nrkarthikeyan/topology-decision-boundaries
  OfficialIn papernone★ 0

Abstract

We propose the labeled Cech complex, the plain labeled Vietoris-Rips complex, and the locally scaled labeled Vietoris-Rips complex to perform persistent homology inference of decision boundaries in classification tasks. We provide theoretical conditions and analysis for recovering the homology of a decision boundary from samples. Our main objective is quantification of deep neural network complexity to enable matching of datasets to pre-trained models; we report results for experiments using MNIST, FashionMNIST, and CIFAR10.

Tasks

Reproductions