Current browse context:
cond-mat.dis-nn
Change to browse by:
References & Citations
Condensed Matter > Disordered Systems and Neural Networks
Title: Spectral pruning of fully connected layers: ranking the nodes based on the eigenvalues
(Submitted on 2 Aug 2021 (this version), latest version 26 Jan 2022 (v2))
Abstract: Training of neural networks can be reformulated in dual space, by allowing eigenvalues and eigenvectors to act as target of the optimization. Working in this setting, we show that the eigenvalues can be used to rank the nodes' relevance, within the collection. By sorting the nodes based on their associated eigenvalues enabled us to devise effective pre- and post-processing trimming strategies to yield compact networks (in terms of the number of composing neurons) with virtually unchanged performance, as compared to their untrimmed homologues. The proposed methods are tested for different architectures, with just a single or multiple hidden layers, and against distinct classification datasets of valuable pedagogical interest.
Submission history
From: Lorenzo Buffoni [view email][v1] Mon, 2 Aug 2021 14:41:14 GMT (2357kb,D)
[v2] Wed, 26 Jan 2022 18:25:14 GMT (5798kb,D)
Link back to: arXiv, form interface, contact.