English
Related papers

Related papers: When Layers Play the Lottery, all Tickets Win at I…

200 papers

Singular-value-decomposition-based coherent integrated photonic neural networks (SC-IPNNs) have a large footprint, suffer from high static power consumption for training and inference, and cannot be pruned using conventional DNN pruning…

Emerging Technologies · Computer Science 2021-12-15 Sanmitra Banerjee , Mahdi Nikdast , Sudeep Pasricha , Krishnendu Chakrabarty

We study whether a neural network optimizes to the same, linearly connected minimum under different samples of SGD noise (e.g., random data order and augmentation). We find that standard vision models become stable to SGD noise in this way…

Machine Learning · Computer Science 2020-07-21 Jonathan Frankle , Gintare Karolina Dziugaite , Daniel M. Roy , Michael Carbin

We present a filter pruning approach for deep model compression, using a multitask network. Our approach is based on learning a a pruner network to prune a pre-trained target network. The pruner is essentially a multitask deep neural…

Computer Vision and Pattern Recognition · Computer Science 2020-01-17 Vinay Kumar Verma , Pravendra Singh , Vinay P. Namboodiri , Piyush Rai

Magnitude pruning is a common, effective technique to identify sparse subnetworks at little cost to accuracy. In this work, we ask whether a particular architecture's accuracy-sparsity tradeoff can be improved by combining pruning…

Computer Vision and Pattern Recognition · Computer Science 2021-05-03 Rajiv Movva , Jonathan Frankle , Michael Carbin

Modern deep networks have millions to billions of parameters, which leads to high memory and energy requirements during training as well as during inference on resource-constrained edge devices. Consequently, pruning techniques have been…

Machine Learning · Computer Science 2020-03-06 Sourjya Roy , Priyadarshini Panda , Gopalakrishnan Srinivasan , Anand Raghunathan

Pre-training serves as a broadly adopted starting point for transfer learning on various downstream tasks. Recent investigations of lottery tickets hypothesis (LTH) demonstrate such enormous pre-trained models can be replaced by extremely…

Machine Learning · Computer Science 2022-06-13 Tianlong Chen , Zhenyu Zhang , Sijia Liu , Yang Zhang , Shiyu Chang , Zhangyang Wang

With the emergence of the Software 3.0 era, there is a growing trend of compressing and integrating large models into software systems, with significant societal implications. Regrettably, in numerous instances, model compression techniques…

Software Engineering · Computer Science 2024-07-04 Xuanqi Gao , Weipeng Jiang , Juan Zhai , Shiqing Ma , Xiaoyu Zhang , Chao Shen

The lottery ticket hypothesis (LTH) is well-studied for convolutional neural networks but has been validated only empirically for graph neural networks (GNNs), for which theoretical findings are largely lacking. In this paper, we identify…

Machine Learning · Computer Science 2025-06-05 Lorenz Kummer , Samir Moustafa , Anatol Ehrlich , Franka Bause , Nikolaus Suess , Wilfried N. Gansterer , Nils M. Kriege

The computer vision world has been re-gaining enthusiasm in various pre-trained models, including both classical ImageNet supervised pre-training and recently emerged self-supervised pre-training such as simCLR and MoCo. Pre-trained weights…

Machine Learning · Computer Science 2021-03-31 Tianlong Chen , Jonathan Frankle , Shiyu Chang , Sijia Liu , Yang Zhang , Michael Carbin , Zhangyang Wang

The exponential growth in numbers of parameters of neural networks over the past years has been accompanied by an increase in performance across several fields. However, due to their sheer size, the networks not only became difficult to…

Machine Learning · Computer Science 2022-02-18 Nils Koster , Oliver Grothe , Achim Rettinger

Pruning filters is an effective method for accelerating deep neural networks (DNNs), but most existing approaches prune filters on a pre-trained network directly which limits in acceleration. Although each filter has its own effect in DNNs,…

Computer Vision and Pattern Recognition · Computer Science 2019-05-29 Zhengguang Zhou , Wengang Zhou , Richang Hong , Houqiang Li

Pruning is a well-known mechanism for reducing the computational cost of deep convolutional networks. However, studies have shown the potential of pruning as a form of regularization, which reduces overfitting and improves generalization.…

Computer Vision and Pattern Recognition · Computer Science 2021-11-29 Artur Jordao , Helio Pedrini

Overparameterized Neural Networks (NN) display state-of-the-art performance. However, there is a growing need for smaller, energy-efficient, neural networks tobe able to use machine learning applications on devices with limited…

Machine Learning · Statistics 2021-05-21 Soufiane Hayou , Jean-Francois Ton , Arnaud Doucet , Yee Whye Teh

Deep neural networks are often highly overparameterized, prohibiting their use in compute-limited systems. However, a line of recent works has shown that the size of deep networks can be considerably reduced by identifying a subset of…

Machine Learning · Computer Science 2020-06-30 Minsu Cho , Ameya Joshi , Chinmay Hegde

Deep generative adversarial networks (GANs) have gained growing popularity in numerous scenarios, while usually suffer from high parameter complexities for resource-constrained real-world applications. However, the compression of GANs has…

Machine Learning · Computer Science 2021-06-02 Xuxi Chen , Zhenyu Zhang , Yongduo Sui , Tianlong Chen

Deep Reinforcement Learning (RL) is a powerful framework for solving complex real-world problems. Large neural networks employed in the framework are traditionally associated with better generalization capabilities, but their increased size…

Machine Learning · Computer Science 2022-01-03 Samin Yeasar Arnob , Riyasat Ohib , Sergey Plis , Doina Precup

We explore the applications of random matrix theory (RMT) in the training of deep neural networks (DNNs), focusing on layer pruning that is reducing the number of DNN parameters (weights). Our numerical results show that this pruning leads…

Machine Learning · Computer Science 2024-09-10 Leonid Berlyand , Etienne Sandier , Yitzchak Shmalo , Lei Zhang

In this paper, we study the importance of pruning in Deep Networks (DNs) and the yin & yang relationship between (1) pruning highly overparametrized DNs that have been trained from random initialization and (2) training small DNs that have…

When large scale training data is available, one can obtain compact and accurate networks to be deployed in resource-constrained environments effectively through quantization and pruning. However, training data are often protected due to…

Machine Learning · Computer Science 2020-10-16 Chen Zhu , Zheng Xu , Ali Shafahi , Manli Shu , Amin Ghiasi , Tom Goldstein

Recent work on deep neural network pruning has shown there exist sparse subnetworks that achieve equal or improved accuracy, training time, and loss using fewer network parameters when compared to their dense counterparts. Orthogonal to…

Machine Learning · Computer Science 2019-12-06 Justin Cosentino , Federico Zaiter , Dan Pei , Jun Zhu