Exploring The Neural Burden In Pruned Models: An Insight Inspired By Neuroscience

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Wang, Zeyu, Dai, Weichen, Zhou, Xiangyu, Qi, Ji, Zhou, Yi
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914889527721984
author Wang, Zeyu
Dai, Weichen
Zhou, Xiangyu
Qi, Ji
Zhou, Yi
author_facet Wang, Zeyu
Dai, Weichen
Zhou, Xiangyu
Qi, Ji
Zhou, Yi
contents Vision Transformer and its variants have been adopted in many visual tasks due to their powerful capabilities, which also bring significant challenges in computation and storage. Consequently, researchers have introduced various compression methods in recent years, among which the pruning techniques are widely used to remove a significant fraction of the network. Therefore, these methods can reduce significant percent of the FLOPs, but often lead to a decrease in model performance. To investigate the underlying causes, we focus on the pruning methods specifically belonging to the pruning-during-training category, then drew inspiration from neuroscience and propose a new concept for artificial neural network models named Neural Burden. We investigate its impact in the model pruning process, and subsequently explore a simple yet effective approach to mitigate the decline in model performance, which can be applied to any pruning-during-training technique. Extensive experiments indicate that the neural burden phenomenon indeed exists, and show the potential of our method. We hope that our findings can provide valuable insights for future research. Code will be made publicly available after this paper is published.
format Preprint
id arxiv_https___arxiv_org_abs_2407_16716
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Exploring The Neural Burden In Pruned Models: An Insight Inspired By Neuroscience
Wang, Zeyu
Dai, Weichen
Zhou, Xiangyu
Qi, Ji
Zhou, Yi
Neural and Evolutionary Computing
Computer Vision and Pattern Recognition
Machine Learning
Vision Transformer and its variants have been adopted in many visual tasks due to their powerful capabilities, which also bring significant challenges in computation and storage. Consequently, researchers have introduced various compression methods in recent years, among which the pruning techniques are widely used to remove a significant fraction of the network. Therefore, these methods can reduce significant percent of the FLOPs, but often lead to a decrease in model performance. To investigate the underlying causes, we focus on the pruning methods specifically belonging to the pruning-during-training category, then drew inspiration from neuroscience and propose a new concept for artificial neural network models named Neural Burden. We investigate its impact in the model pruning process, and subsequently explore a simple yet effective approach to mitigate the decline in model performance, which can be applied to any pruning-during-training technique. Extensive experiments indicate that the neural burden phenomenon indeed exists, and show the potential of our method. We hope that our findings can provide valuable insights for future research. Code will be made publicly available after this paper is published.
title Exploring The Neural Burden In Pruned Models: An Insight Inspired By Neuroscience
topic Neural and Evolutionary Computing
Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2407.16716