English
Related papers

Related papers: Routing the Lottery: Adaptive Subnetworks for Hete…

200 papers

$\textit{RigL}$, a sparse training algorithm, claims to directly train sparse networks that match or exceed the performance of existing dense-to-sparse training techniques (such as pruning) for a fixed parameter count and compute budget. We…

Machine Learning · Computer Science 2021-03-31 Varun Sundar , Rajat Vadiraj Dwaraknath

Sparse neural networks are effective approaches to reduce the resource requirements for the deployment of deep neural networks. Recently, the concept of adaptive sparse connectivity, has emerged to allow training sparse neural networks from…

Many applications require sparse neural networks due to space or inference time restrictions. There is a large body of work on training dense networks to yield sparse networks for inference, but this limits the size of the largest trainable…

Machine Learning · Computer Science 2021-07-26 Utku Evci , Trevor Gale , Jacob Menick , Pablo Samuel Castro , Erich Elsen

Pruning the weights of randomly initialized neural networks plays an important role in the context of lottery ticket hypothesis. Ramanujan et al. (2020) empirically showed that only pruning the weights can achieve remarkable performance…

Machine Learning · Computer Science 2022-04-06 Daiki Chijiwa , Shin'ya Yamaguchi , Yasutoshi Ida , Kenji Umakoshi , Tomohiro Inoue

We propose a unified framework for adaptive routing in multitask, multimodal prediction settings where data heterogeneity and task interactions vary across samples. Motivated by applications in psychotherapy where structured assessments and…

The advancement of deep learning has led to the development of neural decoders for low latency communications. However, neural decoders can be very complex which can lead to increased computation and latency. We consider iterative pruning…

Machine Learning · Computer Science 2022-11-17 Vikrant Malik , Rohan Ghosh , Mehul Motani

Neural network pruning is a popular technique used to reduce the inference costs of modern, potentially overparameterized, networks. Starting from a pre-trained network, the process is as follows: remove redundant parameters, retrain, and…

Machine Learning · Computer Science 2021-03-05 Lucas Liebenwein , Cenk Baykal , Brandon Carter , David Gifford , Daniela Rus

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

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

Despite their impressive performance, Deep Neural Networks (DNNs) typically underperform Gradient Boosting Trees (GBTs) on many tabular-dataset learning tasks. We propose that applying a different regularization coefficient to each weight…

Machine Learning · Statistics 2018-10-25 Ira Shavitt , Eran Segal

The lottery ticket hypothesis posits that dense networks contain sparse subnetworks, ``winning tickets,'' that, when rewound to their initial weights and retrained in isolation, match the performance of the full model. We ask a more…

Machine Learning · Computer Science 2026-05-19 Alon Bebchuk , Nir Shavit

Estimating individualized treatment rules (ITRs) is fundamental to precision medicine, where the goal is to tailor treatment decisions to individual patient characteristics. While numerous methods have been developed for ITR estimation,…

Methodology · Statistics 2026-05-15 Eun Jeong Oh , Min Qian

Large-scale pretrained models such as LXMERT are becoming popular for learning cross-modal representations on text-image pairs for vision-language tasks. According to the lottery ticket hypothesis, NLP and computer vision models contain…

Computer Vision and Pattern Recognition · Computer Science 2023-10-25 Maryam Hashemi , Ghazaleh Mahmoudi , Sara Kodeiri , Hadi Sheikhi , Sauleh Eetemadi

Vision transformers have revolutionized computer vision, but their computational demands present challenges for training and deployment. This paper introduces LOTUS (LOttery Transformers with Ultra Sparsity), a novel method that leverages…

Computer Vision and Pattern Recognition · Computer Science 2024-05-03 Ojasw Upadhyay

Deep Neural Networks (DNNs) are known to be vulnerable to adversarial attacks, i.e., an imperceptible perturbation to the input can mislead DNNs trained on clean images into making erroneous predictions. To tackle this, adversarial training…

Machine Learning · Computer Science 2025-01-07 Yonggan Fu , Qixuan Yu , Yang Zhang , Shang Wu , Xu Ouyang , David Cox , Yingyan Celine Lin

A striking observation about iterative magnitude pruning (IMP; Frankle et al. 2020) is that $\unicode{x2014}$ after just a few hundred steps of dense training $\unicode{x2014}$ the method can find a sparse sub-network that can be trained to…

Machine Learning · Computer Science 2022-06-06 Mansheej Paul , Brett W. Larsen , Surya Ganguli , Jonathan Frankle , Gintare Karolina Dziugaite

We propose a novel neural model compression strategy combining data augmentation, knowledge transfer, pruning, and quantization for device-robust acoustic scene classification (ASC). Specifically, we tackle the ASC task in a low-resource…

Current state-of-the-art results in Music Information Retrieval are largely dominated by deep learning approaches. These provide unprecedented accuracy across all tasks. However, the consistently overlooked downside of these models is their…

Machine Learning · Computer Science 2020-08-03 Philippe Esling , Theis Bazin , Adrien Bitton , Tristan Carsault , Ninon Devis

Given a public transportation network of stations and connections, we want to find a minimum subset of stations such that each connection runs through a selected station. Although this problem is NP-hard in general, real-world instances are…

Social and Information Networks · Computer Science 2019-05-30 Thomas Bläsius , Philipp Fischbeck , Tobias Friedrich , Martin Schirneck

A wide range of empirical and theoretical works have shown that overparameterisation can amplify the performance of neural networks. According to the lottery ticket hypothesis, overparameterised networks have an increased chance of…

Machine Learning · Statistics 2024-10-10 Stefano Sarao Mannelli , Yaraslau Ivashynka , Andrew Saxe , Luca Saglietti