mlsanityData to 7 Oct 2026

What to Do When Your Discrete Optimization Is the Size of a Neural Network?

arXiv 2402.10339 ↗First posted 15 Feb 2024cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Oftentimes, machine learning applications using neural networks involve solving discrete optimization problems, such as in pruning, parameter-isolation-based continual learning and training of binary networks. Still, these discrete problems are combinatorial in nature and are also not amenable to gradient-based optimization. Additionally, classical approaches used in discrete settings do not scale well to large neural networks, forcing scientists and empiricists to rely on alternative methods. Among these, two main distinct sources of top-down information can be used to lead the model to good solutions: (1) extrapolating gradient information from points outside of the solution set (2) comparing evaluations between members of a subset of the valid solutions. We take continuation path (CP) methods to represent using purely the former and Monte Carlo (MC) methods to represent the latter, while also noting that some hybrid methods combine the two. The main goal of this work is to compare both approaches. For that purpose, we first overview the two classes while also discussing some of their drawbacks analytically. Then, on the experimental section, we compare their performance, starting with smaller microworld experiments, which allow more fine-grained control of problem variables, and gradually moving towards larger problems, including neural network regression and neural network pruning for image classification, where we additionally compare against magnitude-based pruning.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 48 of its 132 references that are in the map. First posted on arXiv 15 Feb 2024.

Builds on

48 of 132 references.

References · 48

  1. 2017Attention Is All You Needcited by 59,932
  2. 2020Language Models are Few-Shot Learnerscited by 32,199
  3. 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 6,979
  4. 2013Auto-Encoding Variational Bayescited by 3,980
  5. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
  6. 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,708
  7. 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
  8. 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,031
  9. 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
  10. 2018Soft Actor-Critic Algorithms and Applicationscited by 1,179
  11. 2016The Concrete Distribution: A Continuous Relaxation of Discrete Random Variablescited by 1,135
  12. 2018Overcoming catastrophic forgetting with hard attention to the taskcited by 672
36 more
  1. 2017To prune, or not to prune: exploring the efficacy of pruning for model compressioncited by 596
  2. 2017Learning Sparse Neural Networks through L0 Regularizationcited by 571
  3. 2021Sparsity in Deep Learning: Pruning and growth for efficient inference and training in neural networkscited by 521
  4. 2019The State of Sparsity in Deep Neural Networkscited by 471
  5. 2020Movement Pruning: Adaptive Sparsity by Fine-Tuningcited by 398
  6. 2020Multi-Task Learning with Deep Neural Networks: A Surveycited by 374
  7. 2017PathNet: Evolution Channels Gradient Descent in Super Neural Networkscited by 362
  8. 2019Monte Carlo Gradient Estimation in Machine Learningcited by 281
  9. 2019Deconstructing Lottery Tickets: Zeros, Signs, and the Supermaskcited by 206
  10. 2017Deep Rewiring: Training very sparse deep networkscited by 136
  11. 2020Training Binary Neural Networks with Real-to-Binary Convolutionscited by 111
  12. 2017Backpropagation through the Void: Optimizing control variates for black-box gradient estimationcited by 104
  13. 2019One ticket to win them all: generalizing lottery ticket initializations across datasets and optimizerscited by 102
  14. 2019Winning the Lottery with Continuous Sparsificationcited by 99
  15. 2019A Signal Propagation Perspective for Pruning Neural Networks at Initializationcited by 96
  16. 2020VarGrad: A Low-Variance Gradient Estimator for Variational Inferencecited by 73
  17. 2020Gradient Estimation with Stochastic Softmax Trickscited by 66
  18. 2018Variance reduction properties of the reparameterization trickcited by 41
  19. 2020Rao-Blackwellizing the Straight-Through Gumbel-Softmax Gradient Estimatorcited by 37
  20. 2015MuProp: Unbiased Backpropagation for Stochastic Neural Networkscited by 36
  21. 2022Bayesian Optimization over Discrete and Mixed Spaces via Probabilistic Reparameterizationcited by 35
  22. 2020High-Capacity Expert Binary Networkscited by 34
  23. 2016The Generalized Reparameterization Gradientcited by 30
  24. 2020Growing Efficient Deep Networks by Structured Continuous Sparsificationcited by 28
  25. 2019Sparse Transfer Learning via Winning Lottery Ticketscited by 27
  26. 2020DisARM: An Antithetic Gradient Estimator for Binary Latent Variablescited by 27
  27. 2019Back to Simplicity: How to Train Accurate BNNs from Scratch?cited by 24
  28. 2019Improved training of binary networks for human pose estimation and image recognitioncited by 22
  29. 2022Gradient Estimation with Discrete Stein Operatorscited by 22
  30. 2020Probabilistic Best Subset Selection via Gradient-Based Optimizationcited by 17
  31. 2019Structured Pruning of a BERT-based Question Answering Modelcited by 16
  32. 2020Learned Threshold Pruningcited by 16
  33. 2021Model-free Policy Learning with Reward Gradientscited by 9
  34. 2021Double Control Variates for Gradient Estimation in Discrete Latent Variable Modelscited by 8
  35. 2018ARM: Augment-REINFORCE-Merge Gradient for Stochastic Binary Networkscited by 3
  36. 2018Improved Gradient-Based Optimization Over Discrete Distributionscited by 3

Built on by

Not cited yet.