Cargando…

Spectral pruning of fully connected layers

Training of neural networks can be reformulated in spectral space, by allowing eigenvalues and eigenvectors of the network to act as target of the optimization instead of the individual weights. Working in this setting, we show that the eigenvalues can be used to rank the nodes’ importance within th...

Descripción completa

Detalles Bibliográficos
Autores principales: Buffoni, Lorenzo, Civitelli, Enrico, Giambagli, Lorenzo, Chicchi, Lorenzo, Fanelli, Duccio
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Nature Publishing Group UK 2022
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9249877/
https://www.ncbi.nlm.nih.gov/pubmed/35778586
http://dx.doi.org/10.1038/s41598-022-14805-7
_version_ 1784739685617106944
author Buffoni, Lorenzo
Civitelli, Enrico
Giambagli, Lorenzo
Chicchi, Lorenzo
Fanelli, Duccio
author_facet Buffoni, Lorenzo
Civitelli, Enrico
Giambagli, Lorenzo
Chicchi, Lorenzo
Fanelli, Duccio
author_sort Buffoni, Lorenzo
collection PubMed
description Training of neural networks can be reformulated in spectral space, by allowing eigenvalues and eigenvectors of the network to act as target of the optimization instead of the individual weights. Working in this setting, we show that the eigenvalues can be used to rank the nodes’ importance within the ensemble. Indeed, we will prove that sorting the nodes based on their associated eigenvalues, enables effective pre- and post-processing pruning strategies to yield massively compacted networks (in terms of the number of composing neurons) with virtually unchanged performance. The proposed methods are tested for different architectures, with just a single or multiple hidden layers, and against distinct classification tasks of general interest.
format Online
Article
Text
id pubmed-9249877
institution National Center for Biotechnology Information
language English
publishDate 2022
publisher Nature Publishing Group UK
record_format MEDLINE/PubMed
spelling pubmed-92498772022-07-03 Spectral pruning of fully connected layers Buffoni, Lorenzo Civitelli, Enrico Giambagli, Lorenzo Chicchi, Lorenzo Fanelli, Duccio Sci Rep Article Training of neural networks can be reformulated in spectral space, by allowing eigenvalues and eigenvectors of the network to act as target of the optimization instead of the individual weights. Working in this setting, we show that the eigenvalues can be used to rank the nodes’ importance within the ensemble. Indeed, we will prove that sorting the nodes based on their associated eigenvalues, enables effective pre- and post-processing pruning strategies to yield massively compacted networks (in terms of the number of composing neurons) with virtually unchanged performance. The proposed methods are tested for different architectures, with just a single or multiple hidden layers, and against distinct classification tasks of general interest. Nature Publishing Group UK 2022-07-01 /pmc/articles/PMC9249877/ /pubmed/35778586 http://dx.doi.org/10.1038/s41598-022-14805-7 Text en © The Author(s) 2022 https://creativecommons.org/licenses/by/4.0/Open AccessThis article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made. The images or other third party material in this article are included in the article's Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article's Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by/4.0/ (https://creativecommons.org/licenses/by/4.0/) .
spellingShingle Article
Buffoni, Lorenzo
Civitelli, Enrico
Giambagli, Lorenzo
Chicchi, Lorenzo
Fanelli, Duccio
Spectral pruning of fully connected layers
title Spectral pruning of fully connected layers
title_full Spectral pruning of fully connected layers
title_fullStr Spectral pruning of fully connected layers
title_full_unstemmed Spectral pruning of fully connected layers
title_short Spectral pruning of fully connected layers
title_sort spectral pruning of fully connected layers
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9249877/
https://www.ncbi.nlm.nih.gov/pubmed/35778586
http://dx.doi.org/10.1038/s41598-022-14805-7
work_keys_str_mv AT buffonilorenzo spectralpruningoffullyconnectedlayers
AT civitellienrico spectralpruningoffullyconnectedlayers
AT giambaglilorenzo spectralpruningoffullyconnectedlayers
AT chicchilorenzo spectralpruningoffullyconnectedlayers
AT fanelliduccio spectralpruningoffullyconnectedlayers