Integer-Only Discrete Flows (ICML 2022) [arxiv]
This repository contains Pytorch implementation of experiments from the paper Fast Lossless Neural Compression with Integer-Only Discrete Flows. The implementation is based on Integer Discrete Flows. rANS entropy coding in C language is based on local bits back.
- Python >= 3.7
- Pytorch 1.9.0
- TensorRT 8.2.0.6 + CUDA 10.2
-
Follow training procedure described by Algorithm.1 in the paper. Refer to commands.sh for detailed scripts.
-
For TensorRT Implementation, switch to branch trt.
Please open an issue.
@inproceedings{wang2022fast,
title={Fast Lossless Neural Compression with Integer-Only Discrete Flows},
author={Wang, Siyu and Chen, Jianfei and Li, Chongxuan and Zhu, Jun and Zhang, Bo},
booktitle={International Conference on Machine Learning},
year={2022},
organization={PMLR}
}