Skip to content

Commit

Permalink
Added Data Generation Arxiv paper reference
Browse files Browse the repository at this point in the history
Added new paper to the references list and under data generation

- https://arxiv.org/abs/2410.22110
  • Loading branch information
ServiAmirPM authored Nov 20, 2024
1 parent c72cd0f commit 71eb106
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -92,6 +92,7 @@ ________________________________________________________________________________
__________________________________________________________________________________________________________
### Data-free quantization (Data Generation) [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/sony/model_optimization/blob/main/tutorials/notebooks/mct_features_notebooks/pytorch/example_pytorch_data_generation.ipynb)
Generates synthetic images based on the statistics stored in the model's batch normalization layers, according to your specific needs, for when image data isn’t available. See [Data Generation Library](https://github.com/sony/model_optimization/blob/main/model_compression_toolkit/data_generation/README.md) for more.
The specifications of the method are detailed in the paper: _"**Data Generation for Hardware-Friendly Post-Training Quantization**"_ [5].
__________________________________________________________________________________________________________
### Structured Pruning [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/sony/model_optimization/blob/main/tutorials/notebooks/mct_features_notebooks/pytorch/example_pytorch_pruning_mnist.ipynb)
Reduces model size/complexity and ensures better channels utilization by removing redundant input channels from layers and reconstruction of layer weights. Read more ([Pytorch API](https://sony.github.io/model_optimization/docs/api/api_docs/methods/pytorch_pruning_experimental.html) / [Keras API](https://sony.github.io/model_optimization/docs/api/api_docs/methods/keras_pruning_experimental.html)).
Expand Down Expand Up @@ -210,3 +211,5 @@ MCT is licensed under Apache License Version 2.0. By contributing to the project
[3] [TORCHVISION.MODELS](https://pytorch.org/vision/stable/models.html)

[4] Gordon, O., Cohen, E., Habi, H. V., & Netzer, A., 2024. [EPTQ: Enhanced Post-Training Quantization via Hessian-guided Network-wise Optimization. arXiv preprint](https://arxiv.org/abs/2309.11531)

[5] Dikstein, L., Lapid, A., Habi, H. V., & Netzer, A., 2024. [Data Generation for Hardware-Friendly Post-Training Quantization](https://arxiv.org/abs/2410.22110)

0 comments on commit 71eb106

Please sign in to comment.