English
Related papers

Related papers: Rare Gems: Finding Lottery Tickets at Initializati…

200 papers

Graph Neural Networks (GNNs) demonstrate superior performance in various graph learning tasks, yet their wider real-world application is hindered by the computational overhead when applied to large-scale graphs. To address the issue, the…

Machine Learning · Computer Science 2024-12-11 Yanwei Yue , Guibin Zhang , Haoran Yang , Dawei Cheng

Sparse Neural Networks (NNs) can match the generalization of dense NNs using a fraction of the compute/storage for inference, and also have the potential to enable efficient training. However, naively training unstructured sparse NNs from…

Machine Learning · Computer Science 2022-03-17 Utku Evci , Yani A. Ioannou , Cem Keskin , Yann Dauphin

Network pruning is a method for reducing test-time computational resource requirements with minimal performance degradation. Conventional wisdom of pruning algorithms suggests that: (1) Pruning methods exploit information from training data…

Machine Learning · Computer Science 2020-10-23 Jingtong Su , Yihang Chen , Tianle Cai , Tianhao Wu , Ruiqi Gao , Liwei Wang , Jason D. Lee

The underlying loss landscapes of deep neural networks have a great impact on their training, but they have mainly been studied theoretically due to computational constraints. This work vastly reduces the time required to compute such loss…

Machine Learning · Computer Science 2021-12-17 Robert Bain

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

Modern deep neural networks require a significant amount of computing time and power to train and deploy, which limits their usage on edge devices. Inspired by the iterative weight pruning in the Lottery Ticket Hypothesis, we propose…

Machine Learning · Computer Science 2022-07-15 John Tan Chong Min , Mehul Motani

Sparse shrunk additive models and sparse random feature models have been developed separately as methods to learn low-order functions, where there are few interactions between variables, but neither offers computational efficiency. On the…

Machine Learning · Computer Science 2021-12-09 Yuege Xie , Bobby Shi , Hayden Schaeffer , Rachel Ward

The Lottery Ticket Hypothesis (LTH) posits the existence of a sparse subnetwork (a.k.a. winning ticket) that can generalize comparably to its over-parameterized counterpart when trained from scratch. The common approach to finding a winning…

Machine Learning · Computer Science 2025-04-09 Junghun Oh , Sungyong Baik , Kyoung Mu Lee

Deep neural networks are effective feature extractors but they are prohibitively large for deployment scenarios. Due to the huge number of parameters, interpretability of parameters in different layers is not straight-forward. This is why…

Computation and Language · Computer Science 2021-12-23 Saeed Damadi

Recently, Diffenderfer and Kailkhura proposed a new paradigm for learning compact yet highly accurate binary neural networks simply by pruning and quantizing randomly weighted full precision neural networks. However, the accuracy of these…

Machine Learning · Computer Science 2022-09-27 Hao Cheng , Pu Zhao , Yize Li , Xue Lin , James Diffenderfer , Ryan Goldhahn , Bhavya Kailkhura

Neural network pruning is a widely used strategy for reducing model storage and computing requirements. It allows to lower the complexity of the network by introducing sparsity in the weights. Because taking advantage of sparse matrices is…

Computer Vision and Pattern Recognition · Computer Science 2022-03-14 Nathan Hubens , Matei Mancas , Bernard Gosselin , Marius Preda , Titus Zaharia

Large Transformer-based models were shown to be reducible to a smaller number of self-attention heads and layers. We consider this phenomenon from the perspective of the lottery ticket hypothesis, using both structured and magnitude…

Computation and Language · Computer Science 2020-10-27 Sai Prasanna , Anna Rogers , Anna Rumshisky

This work introduces a novel approach to pruning deep learning models by using distilled data. Unlike conventional strategies which primarily focus on architectural or algorithmic optimization, our method reconsiders the role of data in…

Machine Learning · Computer Science 2023-08-10 Luke McDermott , Daniel Cummings

The hypothesis that sub-network initializations (lottery) exist within the initializations of over-parameterized networks, which when trained in isolation produce highly generalizable models, has led to crucial insights into network…

Machine Learning · Statistics 2020-10-01 Bindya Venkatesh , Jayaraman J. Thiagarajan , Kowshik Thopalli , Prasanna Sattigeri

We study the generalization properties of pruned neural networks that are the winners of the lottery ticket hypothesis on datasets of natural images. We analyse their potential under conditions in which training data is scarce and comes…

Computer Vision and Pattern Recognition · Computer Science 2020-11-23 Matthia Sabatelli , Mike Kestemont , Pierre Geurts

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

Over-parameterized neural networks incur prohibitive memory and computational costs for resource-constrained deployment. The Strong Lottery Ticket (SLT) hypothesis suggests that randomly initialized networks contain sparse subnetworks…

Machine Learning · Computer Science 2026-03-11 Itamar Tsayag , Ofir Lindenbaum

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

The design of sparse neural networks, i.e., of networks with a reduced number of parameters, has been attracting increasing research attention in the last few years. The use of sparse models may significantly reduce the computational and…

Machine Learning · Computer Science 2025-01-22 Giulia Fracastoro , Sophie M. Fosson , Andrea Migliorati , Giuseppe C. Calafiore

Discovering a high-performing sparse network within a massive neural network is advantageous for deploying them on devices with limited storage, such as mobile phones. Additionally, model explainability is essential to fostering trust in…

Computer Vision and Pattern Recognition · Computer Science 2023-07-27 Shantanu Ghosh , Kayhan Batmanghelich