seulkiyeom / LRP_pruning

LRP based Structured Pruning

Requirements

Pytorch 1.6.0+
Python 3.8+
pandas

Model

ResNet-18, ResNet-50
VGG-16, AlexNet

Toy Experiment

https://github.com/seulkiyeom/LRP_Pruning_toy_example

Reference

If you make use of, or are inspired by our work and code, please cite our paper

@article{yeom2021pruning,
  title={Pruning by explaining: A novel criterion for deep neural network pruning},
  author={Yeom, Seul-Ki and
          Seegerer, Philipp and
          Lapuschkin, Sebastian and
          Binder, Alexander and
          Wiedemann, Simon and
          M{\"u}ller, Klaus-Robert and
          Samek, Wojciech},
  journal={Pattern Recognition},
  pages={107899},
  year={2021},
  publisher={Elsevier}
}