English
Related papers

Related papers: One-shot Network Pruning at Initialization with Di…

200 papers

We introduce a pruning algorithm that provably sparsifies the parameters of a trained model in a way that approximately preserves the model's predictive accuracy. Our algorithm uses a small batch of input points to construct a data-informed…

Machine Learning · Computer Science 2021-03-16 Cenk Baykal , Lucas Liebenwein , Igor Gilitschenski , Dan Feldman , Daniela Rus

Spiking Neural Networks (SNNs) have gained significant attention due to the energy-efficient and multiplication-free characteristics. Despite these advantages, deploying large-scale SNNs on edge hardware is challenging due to limited…

Neural and Evolutionary Computing · Computer Science 2024-11-22 Shuo Chen , Boxiao Liu , Zeshi Liu , Haihang You

Deep learning methods are highly effective for many image reconstruction tasks. However, the performance of supervised learned models can degrade when applied to distinct experimental settings at test time or in the presence of distribution…

Image and Video Processing · Electrical Eng. & Systems 2024-12-30 Shijun Liang , Evan Bell , Avrajit Ghosh , Saiprasad Ravishankar

Network pruning can significantly reduce the computation and memory footprint of large neural networks. To achieve a good trade-off between model size and performance, popular pruning techniques usually rely on hand-crafted heuristics and…

Computer Vision and Pattern Recognition · Computer Science 2021-01-19 Wenyuan Zeng , Yuwen Xiong , Raquel Urtasun

DNN pruning reduces memory footprint and computational work of DNN-based solutions to improve performance and energy-efficiency. An effective pruning scheme should be able to systematically remove connections and/or neurons that are…

Machine Learning · Computer Science 2019-06-07 Marc Riera , Jose-Maria Arnau , Antonio Gonzalez

Semantic segmentation of multichannel images is a fundamental task for many applications. Selecting an appropriate channel combination from the original multichannel image can improve the accuracy of semantic segmentation and reduce the…

Computer Vision and Pattern Recognition · Computer Science 2023-05-09 Yuanzhi Cai , Jagannath Aryal , Yuan Fang , Hong Huang , Lei Fan

Developmental plasticity plays a prominent role in shaping the brain's structure during ongoing learning in response to dynamically changing environments. However, the existing network compression methods for deep artificial neural networks…

Neural and Evolutionary Computing · Computer Science 2024-10-29 Bing Han , Feifei Zhao , Yi Zeng , Guobin Shen

Recently, a race towards the simplification of deep networks has begun, showing that it is effectively possible to reduce the size of these models with minimal or no performance loss. However, there is a general lack in understanding why…

Machine Learning · Computer Science 2022-12-29 Enzo Tartaglione , Andrea Bragagnolo , Marco Grangetto

Structured pruning is a promising approach for reducing the inference costs of large vision and language models. By removing carefully chosen structures, e.g., neurons or attention heads, the improvements from this approach can be realized…

Computer Vision and Pattern Recognition · Computer Science 2024-03-21 Xiang Meng , Shibal Ibrahim , Kayhan Behdin , Hussein Hazimeh , Natalia Ponomareva , Rahul Mazumder

Current techniques for deep neural network (DNN) pruning often involve intricate multi-step processes that require domain-specific expertise, making their widespread adoption challenging. To address the limitation, the Only-Train-Once (OTO)…

Computer Vision and Pattern Recognition · Computer Science 2024-03-25 Xidong Wu , Shangqian Gao , Zeyu Zhang , Zhenzhen Li , Runxue Bao , Yanfu Zhang , Xiaoqian Wang , Heng Huang

Artificial neural networks (ANNs) especially deep convolutional networks are very popular these days and have been proved to successfully offer quite reliable solutions to many vision problems. However, the use of deep neural networks is…

Machine Learning · Computer Science 2020-07-28 Yangzi Guo , Yiyuan She , Adrian Barbu

Text-to-image (T2I) diffusion models have demonstrated impressive image generation capabilities. Still, their computational intensity prohibits resource-constrained organizations from deploying T2I models after fine-tuning them on their…

Computer Vision and Pattern Recognition · Computer Science 2025-02-12 Alireza Ganjdanesh , Reza Shirkavand , Shangqian Gao , Heng Huang

Structured pruning is an effective approach for compressing large pre-trained neural networks without significantly affecting their performance. However, most current structured pruning methods do not provide any performance guarantees, and…

Machine Learning · Computer Science 2023-02-14 Marwa El Halabi , Suraj Srinivas , Simon Lacoste-Julien

Existing high-performance deep learning models require very intensive computing. For this reason, it is difficult to embed a deep learning model into a system with limited resources. In this paper, we propose the novel idea of the network…

Machine Learning · Computer Science 2019-02-13 Dae-Woong Jeong , Jaehun Kim , Youngseok Kim , Tae-Ho Kim , Myungsu Chae

When deploying pre-trained neural network models in real-world applications, model consumers often encounter resource-constraint platforms such as mobile and smart devices. They typically use the pruning technique to reduce the size and…

Machine Learning · Computer Science 2025-06-19 Mark Huasong Meng , Guangdong Bai , Sin Gee Teo , Jin Song Dong

Attributing the output of a neural network to the contribution of given input elements is a way of shedding light on the black-box nature of neural networks. Due to the complexity of current network architectures, current gradient-based…

Computer Vision and Pattern Recognition · Computer Science 2020-03-10 Ashkan Khakzar , Soroosh Baselizadeh , Saurabh Khanduja , Christian Rupprecht , Seong Tae Kim , Nassir Navab

Sparsification-based pruning has been an important category in model compression. Existing methods commonly set sparsity-inducing penalty terms to suppress the importance of dropped weights, which is regarded as the suppressed…

Computer Vision and Pattern Recognition · Computer Science 2024-03-12 Shengji Tang , Weihao Lin , Hancheng Ye , Peng Ye , Chong Yu , Baopu Li , Tao Chen

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

Existing structured pruning methods typically rely on multi-stage training procedures that incur high computational costs. Pruning at initialization aims to reduce this burden but often suffers from degraded performance. To address these…

Computer Vision and Pattern Recognition · Computer Science 2025-12-18 Deepak Ghimire , Dayoung Kil , Seonghwan Jeong , Jaesik Park , Seong-heum Kim

One-shot neural architecture search allows joint learning of weights and network architecture, reducing computational cost. We limit our search space to the depth of residual networks and formulate an analytically tractable variational…

Machine Learning · Statistics 2020-04-03 Javier Antorán , James Urquhart Allingham , José Miguel Hernández-Lobato