SplineCam: Exact Visualization and Characterization of Deep Network Geometry and Decision Boundaries

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Humayun, Ahmed Imtiaz, Balestriero, Randall, Balakrishnan, Guha, Baraniuk, Richard
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909218576007168
author Humayun, Ahmed Imtiaz
Balestriero, Randall
Balakrishnan, Guha
Baraniuk, Richard
author_facet Humayun, Ahmed Imtiaz
Balestriero, Randall
Balakrishnan, Guha
Baraniuk, Richard
contents Current Deep Network (DN) visualization and interpretability methods rely heavily on data space visualizations such as scoring which dimensions of the data are responsible for their associated prediction or generating new data features or samples that best match a given DN unit or representation. In this paper, we go one step further by developing the first provably exact method for computing the geometry of a DN's mapping - including its decision boundary - over a specified region of the data space. By leveraging the theory of Continuous Piece-Wise Linear (CPWL) spline DNs, SplineCam exactly computes a DNs geometry without resorting to approximations such as sampling or architecture simplification. SplineCam applies to any DN architecture based on CPWL nonlinearities, including (leaky-)ReLU, absolute value, maxout, and max-pooling and can also be applied to regression DNs such as implicit neural representations. Beyond decision boundary visualization and characterization, SplineCam enables one to compare architectures, measure generalizability and sample from the decision boundary on or off the manifold. Project Website: bit.ly/splinecam.
format Preprint
id arxiv_https___arxiv_org_abs_2302_12828
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle SplineCam: Exact Visualization and Characterization of Deep Network Geometry and Decision Boundaries
Humayun, Ahmed Imtiaz
Balestriero, Randall
Balakrishnan, Guha
Baraniuk, Richard
Computer Vision and Pattern Recognition
Machine Learning
Current Deep Network (DN) visualization and interpretability methods rely heavily on data space visualizations such as scoring which dimensions of the data are responsible for their associated prediction or generating new data features or samples that best match a given DN unit or representation. In this paper, we go one step further by developing the first provably exact method for computing the geometry of a DN's mapping - including its decision boundary - over a specified region of the data space. By leveraging the theory of Continuous Piece-Wise Linear (CPWL) spline DNs, SplineCam exactly computes a DNs geometry without resorting to approximations such as sampling or architecture simplification. SplineCam applies to any DN architecture based on CPWL nonlinearities, including (leaky-)ReLU, absolute value, maxout, and max-pooling and can also be applied to regression DNs such as implicit neural representations. Beyond decision boundary visualization and characterization, SplineCam enables one to compare architectures, measure generalizability and sample from the decision boundary on or off the manifold. Project Website: bit.ly/splinecam.
title SplineCam: Exact Visualization and Characterization of Deep Network Geometry and Decision Boundaries
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2302.12828