Accelerating Relative Entropy Coding with Space Partitioning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: He, Jiajun, Flamich, Gergely, Hernández-Lobato, José Miguel
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914996497154048
author He, Jiajun
Flamich, Gergely
Hernández-Lobato, José Miguel
author_facet He, Jiajun
Flamich, Gergely
Hernández-Lobato, José Miguel
contents Relative entropy coding (REC) algorithms encode a random sample following a target distribution $Q$, using a coding distribution $P$ shared between the sender and receiver. Sadly, general REC algorithms suffer from prohibitive encoding times, at least on the order of $2^{D_{\text{KL}}[Q||P]}$, and faster algorithms are limited to very specific settings. This work addresses this issue by introducing a REC scheme utilizing space partitioning to reduce runtime in practical scenarios. We provide theoretical analyses of our method and demonstrate its effectiveness with both toy examples and practical applications. Notably, our method successfully handles REC tasks with $D_{\text{KL}}[Q||P]$ about three times greater than what previous methods can manage, and reduces the bitrate by approximately 5-15% in VAE-based lossless compression on MNIST and INR-based lossy compression on CIFAR-10, compared to previous methods, significantly improving the practicality of REC for neural compression.
format Preprint
id arxiv_https___arxiv_org_abs_2405_12203
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Accelerating Relative Entropy Coding with Space Partitioning
He, Jiajun
Flamich, Gergely
Hernández-Lobato, José Miguel
Information Theory
Machine Learning
Relative entropy coding (REC) algorithms encode a random sample following a target distribution $Q$, using a coding distribution $P$ shared between the sender and receiver. Sadly, general REC algorithms suffer from prohibitive encoding times, at least on the order of $2^{D_{\text{KL}}[Q||P]}$, and faster algorithms are limited to very specific settings. This work addresses this issue by introducing a REC scheme utilizing space partitioning to reduce runtime in practical scenarios. We provide theoretical analyses of our method and demonstrate its effectiveness with both toy examples and practical applications. Notably, our method successfully handles REC tasks with $D_{\text{KL}}[Q||P]$ about three times greater than what previous methods can manage, and reduces the bitrate by approximately 5-15% in VAE-based lossless compression on MNIST and INR-based lossy compression on CIFAR-10, compared to previous methods, significantly improving the practicality of REC for neural compression.
title Accelerating Relative Entropy Coding with Space Partitioning
topic Information Theory
Machine Learning
url https://arxiv.org/abs/2405.12203