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...
Autores principales: | , , , , |
---|---|
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 |