vineeths96 / Compressed-Transformers
In this repository, we explore model compression for transformer architectures via quantization. We specifically explore quantization aware training of the linear layers and demonstrate the performance for 8 bits, 4 bits, 2 bits and 1 bit (binary) quantization.
☆24Updated 3 years ago
Alternatives and similar repositories for Compressed-Transformers:
Users that are interested in Compressed-Transformers are comparing it to the libraries listed below
- [ICASSP'22] Integer-only Zero-shot Quantization for Efficient Speech Recognition☆31Updated 3 years ago
- Position-based Scaled Gradient for Model Quantization and Pruning Code (NeurIPS 2020)☆26Updated 4 years ago
- Code for High-Capacity Expert Binary Networks (ICLR 2021).☆27Updated 3 years ago
- Implementation of NeurIPS 2019 paper "Normalization Helps Training of Quantized LSTM"☆31Updated 9 months ago
- Post-training sparsity-aware quantization☆34Updated 2 years ago
- This repository containts the pytorch scripts to train mixed-precision networks for microcontroller deployment, based on the memory contr…☆49Updated 11 months ago
- ☆28Updated 3 years ago
- ☆17Updated 2 years ago
- The official PyTorch implementation of the NeurIPS2022 (spotlight) paper, Outlier Suppression: Pushing the Limit of Low-bit Transformer L…☆47Updated 2 years ago
- [ICML 2022] "Coarsening the Granularity: Towards Structurally Sparse Lottery Tickets" by Tianlong Chen, Xuxi Chen, Xiaolong Ma, Yanzhi Wa…☆32Updated 2 years ago
- BitSplit Post-trining Quantization☆49Updated 3 years ago
- DeiT implementation for Q-ViT☆24Updated this week
- Reproducing Quantization paper PACT☆63Updated 2 years ago
- [TMLR] Official PyTorch implementation of paper "Efficient Quantization-aware Training with Adaptive Coreset Selection"☆30Updated 8 months ago
- [ICLR 2022 Oral] F8Net: Fixed-Point 8-bit Only Multiplication for Network Quantization☆94Updated 2 years ago
- Towards Compact CNNs via Collaborative Compression☆11Updated 3 years ago
- [UNDER CONSTRUCTION]unofficial implementation of ABC-Net☆12Updated 7 years ago
- Implementation of a Quantized Transformer Model☆18Updated 6 years ago
- Training Quantized Neural Networks with a Full-precision Auxiliary Module☆13Updated 4 years ago
- Layer-wise Pruning of Transformer Heads for Efficient Language Modeling☆21Updated 3 years ago
- Pytorch implementations of the BNN, XNOR-Net and BiReal-Net☆15Updated 4 years ago
- Quantize pytorch model, support post-training quantization and quantization aware training methods☆13Updated last year
- [ICLR 2021 Spotlight] "CPT: Efficient Deep Neural Network Training via Cyclic Precision" by Yonggan Fu, Han Guo, Meng Li, Xin Yang, Yinin…☆30Updated last year
- An 8bit automated quantization conversion tool for the pytorch (Post-training quantization based on KL divergence)☆33Updated 5 years ago
- This is an implementation of YOLO using LSQ network quantization method.☆23Updated 3 years ago
- Improving Post Training Neural Quantization: Layer-wise Calibration and Integer Programming☆34Updated last year
- PyTorch implementation of "Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inference"☆55Updated 5 years ago
- A Out-of-box PyTorch Scaffold for Neural Network Quantization-Aware-Training (QAT) Research. Website: https://github.com/zhutmost/neuralz…☆26Updated 2 years ago
- [TMLR] Official PyTorch implementation of paper "Quantization Variation: A New Perspective on Training Transformers with Low-Bit Precisio…☆44Updated 6 months ago
- Code for ICML 2022 paper "SPDY: Accurate Pruning with Speedup Guarantees"☆18Updated last year