arrow
Return

Pruning during training by network efficacy modeling

delete2023-03-14
delete1
delete
OA
AI
M
Mohit Rajpal *
Y
Yehong Zhang
B
Bryan Kian Hsiang Low
DOI:10.1007/s10994-023-06304-1delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Deep neural networks (DNNs) are costly to train. Pruning, an approach to alleviate model complexity by zeroing out or pruning DNN elements, has shown promise in reducing training costs for DNNs with little to no efficacy at a given task. This paper presents a novel method to perform early pruning of DNN elements (e.g., neurons or convolutional filters) during the training process while minimizing losses to model performance. To achieve this, we model the efficacy of DNN elements in a Bayesian manner conditioned upon efficacy data collected during the training and prune DNN elements with low predictive efficacy after training completion. Empirical evaluations show that the proposed Bayesian early pruning improves the computational efficiency of DNN training while better preserving model performance compared to other tested pruning approaches.
Keywords:
Early pruning
Network efficacy modeling
Network saliency
Multi-output Gaussian process
Foresight pruning

Journal

Machine Learning cover
Machine Learning
IF:
2.9
Papers:
2.7K
Citations:
3.4W

Organization

P
Peng Cheng Laboratory
Scholars:
1.7K
Papers: 1.8K
Citations: 2.0K
N
National University of Singapore
Scholars:
7.6W
Papers: 6.5W
Citations: 11.4W
Cited Papers

Cited Papers

Memory cells specific for myelin oligodendrocyte glycoprotein (MOG) govern the transfer of experimental autoimmune encephalomyelitis
err2011-05-01
err0
errOAAI
errJessica L. Williams; Aaron P. Kithcart; Kristen M. Smith; Todd Shawler; Gina M. Cox; Caroline C. Whitacre
errShare
errSave
Predictors of walking capacity in peripheral arterial disease patients
err2013-04-01
err0
errOAAI
errBreno Quintella Farah; João Paulo dos Anjos Souza Barbosa; Gabriel Grizzo Cucato; Marcel da Rocha Chehuen; Luis Alberto Gobbo; Nelson Wolosker; Cláudia Lúcia de Moraes Forjaz; Raphael Mendes Ritti-Dias
errShare
errSave
Relative alpha desynchronization and synchronization during speech perception
err1997-06-01
err0
PREAI
errChristina M Krause; Bodil Pörn; A.Heikki Lang; Matti Laine
errShare
errSave
errShare
errSave
researcher View more