Skip to content

Commit

Permalink
Merge remote-tracking branch 'origin/main' into xquant
Browse files Browse the repository at this point in the history
  • Loading branch information
reuvenp committed Jun 17, 2024
2 parents 8e35f8b + c77aee7 commit 9272a97
Show file tree
Hide file tree
Showing 163 changed files with 6,217 additions and 1,697 deletions.
44 changes: 22 additions & 22 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -6,20 +6,20 @@ This project provides researchers, developers, and engineers tools for optimizin

Specifically, this project aims to apply quantization to compress neural networks.

<img src="docsrc/images/mct_block_diagram.svg" width="10000">
<img src="https://github.com/sony/model_optimization/raw/main/docsrc/images/mct_block_diagram.svg" width="10000">

MCT is developed by researchers and engineers working at Sony Semiconductor Israel.



## Table of Contents

- [Getting Started](#getting-started)
- [Supported features](#supported-features)
- [Results](#results)
- [Troubleshooting](#trouble-shooting)
- [Contributions](#contributions)
- [License](#license)
- [Getting Started](https://github.com/sony/model_optimization?tab=readme-ov-file#getting-started)
- [Supported features](https://github.com/sony/model_optimization?tab=readme-ov-file#supported-features)
- [Results](https://github.com/sony/model_optimization?tab=readme-ov-file#results)
- [Troubleshooting](https://github.com/sony/model_optimization?tab=readme-ov-file#trouble-shooting)
- [Contributions](https://github.com/sony/model_optimization?tab=readme-ov-file#contributions)
- [License](https://github.com/sony/model_optimization?tab=readme-ov-file#license)


## Getting Started
Expand All @@ -33,17 +33,17 @@ To install the latest stable release of MCT, run the following command:
pip install model-compression-toolkit
```

For installing the nightly version or installing from source, refer to the [installation guide](INSTALLATION.md).
For installing the nightly version or installing from source, refer to the [installation guide](https://github.com/sony/model_optimization/blob/main/INSTALLATION.md).


### Quick start & tutorials

Explore the Model Compression Toolkit (MCT) through our tutorials,
covering compression techniques for Keras and PyTorch models. Access interactive [notebooks](tutorials/README.md)
covering compression techniques for Keras and PyTorch models. Access interactive [notebooks](https://github.com/sony/model_optimization/blob/main/tutorials/README.md)
for hands-on learning. For example:
* [Keras MobileNetV2 post training quantization](tutorials/notebooks/imx500_notebooks/keras/example_keras_mobilenetv2_for_imx500.ipynb)
* [Post training quantization with PyTorch](tutorials/notebooks/mct_features_notebooks/pytorch/example_pytorch_ptq_mnist.ipynb)
* [Data Generation for ResNet18 with PyTorch](tutorials/notebooks/mct_features_notebooks/pytorch/example_pytorch_data_generation.ipynb).
* [Keras MobileNetV2 post training quantization](https://github.com/sony/model_optimization/blob/main/tutorials/notebooks/imx500_notebooks/keras/example_keras_mobilenetv2_for_imx500.ipynb)
* [Post training quantization with PyTorch](https://github.com/sony/model_optimization/blob/main/tutorials/notebooks/mct_features_notebooks/pytorch/example_pytorch_ptq_mnist.ipynb)
* [Data Generation for ResNet18 with PyTorch](https://github.com/sony/model_optimization/blob/main/tutorials/notebooks/mct_features_notebooks/pytorch/example_pytorch_data_generation.ipynb).


### Supported Versions
Expand All @@ -67,15 +67,15 @@ Currently, MCT is being tested on various Python, Pytorch and TensorFlow version
## Supported Features
MCT offers a range of powerful features to optimize neural network models for efficient deployment. These supported features include:

### Data Generation [*](#experimental-features)
### Data Generation [*](https://github.com/sony/model_optimization?tab=readme-ov-file#experimental-features)
MCT provides tools for generating synthetic images based on the statistics stored in a model's batch normalization layers. These generated images are valuable for various compression tasks where image data is required, such as quantization and pruning.
You can customize data generation configurations to suit your specific needs. [Go to the Data Generation page.](model_compression_toolkit/data_generation/README.md)
You can customize data generation configurations to suit your specific needs. [Go to the Data Generation page.](https://github.com/sony/model_optimization/blob/main/model_compression_toolkit/data_generation/README.md)

### Quantization
MCT supports different quantization methods:
* Post-training quantization (PTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/api_docs/methods/keras_post_training_quantization.html), [PyTorch API](https://sony.github.io/model_optimization/docs/api/api_docs/methods/pytorch_post_training_quantization.html)
* Gradient-based post-training quantization (GPTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/api_docs/methods/keras_gradient_post_training_quantization.html), [PyTorch API](https://sony.github.io/model_optimization/docs/api/api_docs/methods/pytorch_gradient_post_training_quantization.html)
* Quantization-aware training (QAT) [*](#experimental-features)
* Quantization-aware training (QAT) [*](https://github.com/sony/model_optimization?tab=readme-ov-file#experimental-features)


| Quantization Method | Complexity | Computational Cost |
Expand All @@ -97,20 +97,20 @@ Main features:
* <ins>Advanced quantization algorithms:</ins> To prevent a performance degradation some algorithms are applied such as:
* <ins>Shift negative correction:</ins> Symmetric activation quantization can hurt the model's performance when some layers output both negative and positive activations, but their range is asymmetric. For more details please visit [1].
* <ins>Outliers filtering:</ins> Computing z-score for activation statistics to detect and remove outliers.
* <ins>Clustering:</ins> Using non-uniform quantization grid to quantize the weights and activations to match their distributions.[*](#experimental-features)
* <ins>Clustering:</ins> Using non-uniform quantization grid to quantize the weights and activations to match their distributions.[*](https://github.com/sony/model_optimization?tab=readme-ov-file#experimental-features)
* <ins>Mixed-precision search:</ins> Assigning quantization bit-width per layer (for weights/activations), based on the layer's sensitivity to different bit-widths.
* <ins>Visualization:</ins> You can use TensorBoard to observe useful information for troubleshooting the quantized model's performance (for example, the model in different phases of the quantization, collected statistics, similarity between layers of the float and quantized model and bit-width configuration for mixed-precision quantization). For more details, please read the [visualization documentation](https://sony.github.io/model_optimization/docs/guidelines/visualization.html).
* <ins>Target Platform Capabilities:</ins> The Target Platform Capabilities (TPC) describes the target platform (an edge device with dedicated hardware). For more details, please read the [TPC README](model_compression_toolkit/target_platform_capabilities/README.md).
* <ins>Target Platform Capabilities:</ins> The Target Platform Capabilities (TPC) describes the target platform (an edge device with dedicated hardware). For more details, please read the [TPC README](https://github.com/sony/model_optimization/blob/main/model_compression_toolkit/target_platform_capabilities/README.md).

### Enhanced Post-Training Quantization (EPTQ)
As part of the GPTQ we provide an advanced optimization algorithm called EPTQ.

The specifications of the algorithm are detailed in the paper: _"**EPTQ: Enhanced Post-Training Quantization via Label-Free Hessian**"_ [4].

More details on the how to use EPTQ via MCT can be found in the [EPTQ guidelines](model_compression_toolkit/gptq/README.md).
More details on the how to use EPTQ via MCT can be found in the [EPTQ guidelines](https://github.com/sony/model_optimization/blob/main/model_compression_toolkit/gptq/README.md).


### Structured Pruning [*](#experimental-features)
### Structured Pruning [*](https://github.com/sony/model_optimization?tab=readme-ov-file#experimental-features)
MCT introduces a structured and hardware-aware model pruning.
This pruning technique is designed to compress models for specific hardware architectures,
taking into account the target platform's Single Instruction, Multiple Data (SIMD) capabilities.
Expand All @@ -132,7 +132,7 @@ For more details, we highly recommend visiting our project website where experim
Graph of [MobileNetV2](https://keras.io/api/applications/mobilenet/) accuracy on ImageNet vs average bit-width of weights, using
single-precision quantization, mixed-precision quantization, and mixed-precision quantization with GPTQ.

<img src="docsrc/images/mbv2_accuracy_graph.png">
<img src="https://github.com/sony/model_optimization/raw/main/docsrc/images/mbv2_accuracy_graph.png">

For more results, please see [1]

Expand Down Expand Up @@ -168,11 +168,11 @@ Check out the [FAQ](https://github.com/sony/model_optimization/tree/main/FAQ.md)
## Contributions
MCT aims at keeping a more up-to-date fork and welcomes contributions from anyone.

*You will find more information about contributions in the [Contribution guide](CONTRIBUTING.md).
*You will find more information about contributions in the [Contribution guide](https://github.com/sony/model_optimization/blob/main/CONTRIBUTING.md).


## License
[Apache License 2.0](LICENSE.md).
[Apache License 2.0](https://github.com/sony/model_optimization/blob/main/LICENSE.md).

## References

Expand Down
2 changes: 1 addition & 1 deletion docs/.buildinfo
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Sphinx build info version 1
# This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done.
config: 000d2f7b9e75da9f4efddff5202e396f
config: 5cc6500fd7d8a78cea89aef3268de5c5
tags: 645f666f9bcd5a90fca523b33c5a78b7
2 changes: 1 addition & 1 deletion docs/_sources/index.rst.txt
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,7 @@ Quickstart
====================================
Take a look of how you can start using MCT in just a few minutes!

Visit our `notebooks <https://github.com/sony/model_optimization/tree/main/tutorials/notebooks>`_ and `MCT quick start. <https://github.com/sony/model_optimization/tree/main/tutorials/quick_start>`_
Visit our `notebooks <https://github.com/sony/model_optimization/tree/main/tutorials/notebooks>`_


API Documentation
Expand Down
6 changes: 3 additions & 3 deletions docs/api/api_docs/classes/DataGenerationConfig.html
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
<meta charset="utf-8" />
<meta name="viewport" content="width=device-width, initial-scale=1.0" /><meta name="generator" content="Docutils 0.18.1: http://docutils.sourceforge.net/" />

<title>Data Generation Configuration &#8212; MCT Documentation: ver 2.0.0</title>
<title>Data Generation Configuration &#8212; MCT Documentation: ver 2.1.0</title>
<link rel="stylesheet" type="text/css" href="../../../static/pygments.css" />
<link rel="stylesheet" type="text/css" href="../../../static/bizstyle.css" />
<link rel="stylesheet" type="text/css" href="../../../static/css/custom.css" />
Expand All @@ -31,7 +31,7 @@ <h3>Navigation</h3>
<li class="right" style="margin-right: 10px">
<a href="../../../genindex.html" title="General Index"
accesskey="I">index</a></li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.0.0</a> &#187;</li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.1.0</a> &#187;</li>
<li class="nav-item nav-item-this"><a href="">Data Generation Configuration</a></li>
</ul>
</div>
Expand Down Expand Up @@ -212,7 +212,7 @@ <h3>Navigation</h3>
<li class="right" style="margin-right: 10px">
<a href="../../../genindex.html" title="General Index"
>index</a></li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.0.0</a> &#187;</li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.1.0</a> &#187;</li>
<li class="nav-item nav-item-this"><a href="">Data Generation Configuration</a></li>
</ul>
</div>
Expand Down
6 changes: 3 additions & 3 deletions docs/api/api_docs/classes/DefaultDict.html
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
<meta charset="utf-8" />
<meta name="viewport" content="width=device-width, initial-scale=1.0" /><meta name="generator" content="Docutils 0.18.1: http://docutils.sourceforge.net/" />

<title>DefaultDict Class &#8212; MCT Documentation: ver 2.0.0</title>
<title>DefaultDict Class &#8212; MCT Documentation: ver 2.1.0</title>
<link rel="stylesheet" type="text/css" href="../../../static/pygments.css" />
<link rel="stylesheet" type="text/css" href="../../../static/bizstyle.css" />
<link rel="stylesheet" type="text/css" href="../../../static/css/custom.css" />
Expand All @@ -31,7 +31,7 @@ <h3>Navigation</h3>
<li class="right" style="margin-right: 10px">
<a href="../../../genindex.html" title="General Index"
accesskey="I">index</a></li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.0.0</a> &#187;</li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.1.0</a> &#187;</li>
<li class="nav-item nav-item-this"><a href="">DefaultDict Class</a></li>
</ul>
</div>
Expand Down Expand Up @@ -114,7 +114,7 @@ <h3>Navigation</h3>
<li class="right" style="margin-right: 10px">
<a href="../../../genindex.html" title="General Index"
>index</a></li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.0.0</a> &#187;</li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.1.0</a> &#187;</li>
<li class="nav-item nav-item-this"><a href="">DefaultDict Class</a></li>
</ul>
</div>
Expand Down
6 changes: 3 additions & 3 deletions docs/api/api_docs/classes/FrameworkInfo.html
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
<meta charset="utf-8" />
<meta name="viewport" content="width=device-width, initial-scale=1.0" /><meta name="generator" content="Docutils 0.18.1: http://docutils.sourceforge.net/" />

<title>FrameworkInfo Class &#8212; MCT Documentation: ver 2.0.0</title>
<title>FrameworkInfo Class &#8212; MCT Documentation: ver 2.1.0</title>
<link rel="stylesheet" type="text/css" href="../../../static/pygments.css" />
<link rel="stylesheet" type="text/css" href="../../../static/bizstyle.css" />
<link rel="stylesheet" type="text/css" href="../../../static/css/custom.css" />
Expand All @@ -31,7 +31,7 @@ <h3>Navigation</h3>
<li class="right" style="margin-right: 10px">
<a href="../../../genindex.html" title="General Index"
accesskey="I">index</a></li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.0.0</a> &#187;</li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.1.0</a> &#187;</li>
<li class="nav-item nav-item-this"><a href="">FrameworkInfo Class</a></li>
</ul>
</div>
Expand Down Expand Up @@ -144,7 +144,7 @@ <h3>Navigation</h3>
<li class="right" style="margin-right: 10px">
<a href="../../../genindex.html" title="General Index"
>index</a></li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.0.0</a> &#187;</li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.1.0</a> &#187;</li>
<li class="nav-item nav-item-this"><a href="">FrameworkInfo Class</a></li>
</ul>
</div>
Expand Down
6 changes: 3 additions & 3 deletions docs/api/api_docs/classes/GradientPTQConfig.html
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
<meta charset="utf-8" />
<meta name="viewport" content="width=device-width, initial-scale=1.0" /><meta name="generator" content="Docutils 0.18.1: http://docutils.sourceforge.net/" />

<title>GradientPTQConfig Class &#8212; MCT Documentation: ver 2.0.0</title>
<title>GradientPTQConfig Class &#8212; MCT Documentation: ver 2.1.0</title>
<link rel="stylesheet" type="text/css" href="../../../static/pygments.css" />
<link rel="stylesheet" type="text/css" href="../../../static/bizstyle.css" />
<link rel="stylesheet" type="text/css" href="../../../static/css/custom.css" />
Expand All @@ -31,7 +31,7 @@ <h3>Navigation</h3>
<li class="right" style="margin-right: 10px">
<a href="../../../genindex.html" title="General Index"
accesskey="I">index</a></li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.0.0</a> &#187;</li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.1.0</a> &#187;</li>
<li class="nav-item nav-item-this"><a href="">GradientPTQConfig Class</a></li>
</ul>
</div>
Expand Down Expand Up @@ -142,7 +142,7 @@ <h3>Navigation</h3>
<li class="right" style="margin-right: 10px">
<a href="../../../genindex.html" title="General Index"
>index</a></li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.0.0</a> &#187;</li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.1.0</a> &#187;</li>
<li class="nav-item nav-item-this"><a href="">GradientPTQConfig Class</a></li>
</ul>
</div>
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
<meta charset="utf-8" />
<meta name="viewport" content="width=device-width, initial-scale=1.0" /><meta name="generator" content="Docutils 0.18.1: http://docutils.sourceforge.net/" />

<title>MixedPrecisionQuantizationConfig &#8212; MCT Documentation: ver 2.0.0</title>
<title>MixedPrecisionQuantizationConfig &#8212; MCT Documentation: ver 2.1.0</title>
<link rel="stylesheet" type="text/css" href="../../../static/pygments.css" />
<link rel="stylesheet" type="text/css" href="../../../static/bizstyle.css" />
<link rel="stylesheet" type="text/css" href="../../../static/css/custom.css" />
Expand All @@ -31,7 +31,7 @@ <h3>Navigation</h3>
<li class="right" style="margin-right: 10px">
<a href="../../../genindex.html" title="General Index"
accesskey="I">index</a></li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.0.0</a> &#187;</li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.1.0</a> &#187;</li>
<li class="nav-item nav-item-this"><a href="">MixedPrecisionQuantizationConfig</a></li>
</ul>
</div>
Expand Down Expand Up @@ -94,7 +94,7 @@ <h3>Navigation</h3>
<li class="right" style="margin-right: 10px">
<a href="../../../genindex.html" title="General Index"
>index</a></li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.0.0</a> &#187;</li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.1.0</a> &#187;</li>
<li class="nav-item nav-item-this"><a href="">MixedPrecisionQuantizationConfig</a></li>
</ul>
</div>
Expand Down
6 changes: 3 additions & 3 deletions docs/api/api_docs/classes/MpDistanceWeighting.html
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
<meta charset="utf-8" />
<meta name="viewport" content="width=device-width, initial-scale=1.0" /><meta name="generator" content="Docutils 0.18.1: http://docutils.sourceforge.net/" />

<title>MpDistanceWeighting &#8212; MCT Documentation: ver 2.0.0</title>
<title>MpDistanceWeighting &#8212; MCT Documentation: ver 2.1.0</title>
<link rel="stylesheet" type="text/css" href="../../../static/pygments.css" />
<link rel="stylesheet" type="text/css" href="../../../static/bizstyle.css" />
<link rel="stylesheet" type="text/css" href="../../../static/css/custom.css" />
Expand All @@ -31,7 +31,7 @@ <h3>Navigation</h3>
<li class="right" style="margin-right: 10px">
<a href="../../../genindex.html" title="General Index"
accesskey="I">index</a></li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.0.0</a> &#187;</li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.1.0</a> &#187;</li>
<li class="nav-item nav-item-this"><a href="">MpDistanceWeighting</a></li>
</ul>
</div>
Expand Down Expand Up @@ -83,7 +83,7 @@ <h3>Navigation</h3>
<li class="right" style="margin-right: 10px">
<a href="../../../genindex.html" title="General Index"
>index</a></li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.0.0</a> &#187;</li>
<li class="nav-item nav-item-0"><a href="../../../index.html">MCT Documentation: ver 2.1.0</a> &#187;</li>
<li class="nav-item nav-item-this"><a href="">MpDistanceWeighting</a></li>
</ul>
</div>
Expand Down
Loading

0 comments on commit 9272a97

Please sign in to comment.