Giter VIP home page Giter VIP logo

adaptive-quantization-modules's Introduction

Online Learned Continual Compression with Adaptive Quantization Modules (ICML 2020)

Stacking Quantization blocks for efficient lifelong online compression
Code for reproducing all results in our paper which can be found here
You can find a quick demo on Google Colab here

(key) Requirements

  • Python 3.7
  • Pytorch 1.4.0

Structure

├── Common 
    ├── modular.py          # Module (QLayer) and Stacked modules (QStack). Includes most key ops, such as adaptive buffer        
    ├── quantize.py         # Discretization Ops (GumbelSoftmax, Vector/Tensor Quantization and Argmax Quantization)
    ├── model.py            # Encoder, Decoder, Classifier Blocks 
├── config                  # .yaml files specifying different AQM architectures and hyperparameters used in the paper 
├── Lidar
    ├── ....                # files to run LiDAR experiments 
├── Utils             
    ├── args.py             # Contains command-line args
    ├── buffer.py           # Basic buffer implementation. Handled raw and compressed representations
    ├── data.py             # CL datasets and dataloaders
    ├── utils.py            # Logging / Saving & Loading Models, Args, point cloud processing
    
├── gen_main.py             # files to run the offline classification (e.g. Imagenet) experiments 
├── eval.py                 # evaluation loops for drift, test acc / mse, and lidar
├── cls_main.py             # files to run the online classification (e.g. CIFAR) experiments

├── reproduce.txt           # All command and information to reproduce the results in the paper

Acknowledgements

We would like to thank authors of the following repositories (from which we borrowed code) for making the code public.
Gradient Episodic Memory
VQ-VAE
VQ-VAE-2
MIR

Contact

For any questions / comments / concerns, feel free to open an issue via github, or to send me an email at
[email protected].

We strongly believe in fully reproducible research. To that end, if you find any discrepancy between our code and the paper, please let us know, and we will make sure to address it.

Happy streaming compression :)

Citation

If you find this code useful please cite us in your work.

@article{caccia2019online,
  title={Online Learned Continual Compression with Adaptive Quantization Modules},
  author={Caccia, Lucas and Belilovsky, Eugene and Caccia, Massimo and Pineau, Joelle},
  journal={Proceedings of the 37th International Conference on Machine Learning},
  year={2020}
}

adaptive-quantization-modules's People

Contributors

pclucas14 avatar eugenium avatar anonsubmitter2 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.