Interpretable Neuron Structuring with Graph Spectral Regularization
Authors:
Alexander Tong,
David van Dijk,
Jay S. Stanley III,
Matthew Amodio,
Kristina Yim,
Rebecca Muhle,
James Noonan,
Guy Wolf,
Smita Krishnaswamy
Abstract:
While neural networks are powerful approximators used to classify or embed data into lower dimensional spaces, they are often regarded as black boxes with uninterpretable features. Here we propose Graph Spectral Regularization for making hidden layers more interpretable without significantly impacting performance on the primary task. Taking inspiration from spatial organization and localization of…
▽ More
While neural networks are powerful approximators used to classify or embed data into lower dimensional spaces, they are often regarded as black boxes with uninterpretable features. Here we propose Graph Spectral Regularization for making hidden layers more interpretable without significantly impacting performance on the primary task. Taking inspiration from spatial organization and localization of neuron activations in biological networks, we use a graph Laplacian penalty to structure the activations within a layer. This penalty encourages activations to be smooth either on a predetermined graph or on a feature-space graph learned from the data via co-activations of a hidden layer of the neural network. We show numerous uses for this additional structure including cluster indication and visualization in biological and image data sets.
△ Less
Submitted 14 February, 2020; v1 submitted 30 September, 2018;
originally announced October 2018.
