Data-Independent Structured Pruning of Neural Networks via Coresets

By Ben Mussay, Dan Feldman, Samson Zhou, Vladimir Braverman, Margarita Osadchy. 2020

28 citations recorded by the source corpus.

Model compression is crucial for the deployment of neural networks on devices with limited computational and memory resources. Many different methods show comparable accuracy of the compressed model and similar compression rates. However, the majority of the compression methods are based on heuristics and offer no worst case guarantees on the tradeoff between the compression rate and the approximation error for an arbitrarily new sample. We propose the first efficient structured pruning algorithm with a provable tradeoff between its compression rate and the approximation error for any future test sample. Our method is based on the coreset framework, and it approximates the output of a layer of neurons/filters by a coreset of neurons/filters in the previous layer and discards the rest. We apply this framework in a layer-by-layer fashion from the bottom to the top. Unlike previous works, our coreset is data-independent, meaning that it provably guarantees the accuracy of the function for any input $x\in \mathbb {R}^{d}$ , including an adversarial one.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Efficiency & Compression

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Simon Lacoste-Julien
  2. Marwa El Halabi
  3. Suraj Srinivas
  4. Peter Richtárik
  5. Francis R. Bach
  6. Fabian Pedregosa
  7. Martin Jaggi
  8. Gauthier Gidel

In Computer Vision and Pattern Recognition

  1. Simon Lacoste-Julien
  2. Marwa El Halabi
  3. Suraj Srinivas
  4. Peter Richtárik
  5. Ivan Laptev
  6. Jonathan Frankle
  7. Michael Carbin
  8. Gauthier Gidel

In Efficiency & Compression

  1. Simon Lacoste-Julien
  2. Marwa El Halabi
  3. Suraj Srinivas
  4. Jonathan Frankle
  5. Michael Carbin

Citations in the corpus