Neural Distributed Compression

Elza Erkip

Camera arrays, sensor networks and edge devices often compress data that is correlated with data held elsewhere. In the Wyner–Ziv setting, the decoder has correlated side information and the encoder does not. Information theory shows that the encoder can still benefit, by binning: it assigns the same index to source values that are far apart, and the decoder uses its side information to tell them apart. Practical compressors rarely exploit this, and standard neural compressors do not learn it.

This project develops learned distributed compressors that are both practical and interpretable. Trained only from data, our neural Wyner–Ziv compressors recover binning and approach theoretical rate–distortion limits. They outperform neural transform coding that is given the same side information. Related work covers robust compression when the side information may be missing (the Heegard–Berger problem), image compression with side information, and joint source-channel coding. The same ideas power learned compress-and-forward relaying.

The Wyner–Ziv work is a collaboration with Johannes Ballé (Google Research) and was supported in part by the Google Research Collabs Program.

Distributed source codingWyner–Ziv codingLearned compressionBinningInterpretable machine learning
Wyner–Ziv compression: the decoder alone sees correlated side information. Learned compressors discover binning, reusing one index on intervals far apart.
Rate–distortion for a Laplacian source whose sign is known at the decoder. Neural transform coding, even when given the side information, stays close to the optimal quantizer that ignores it. The learned Wyner–Ziv compressor reaches the optimal scalar quantizer with side information. Data from Özyılkan, Ballé and Erkip, IEEE JSAIT 2024.
Learned compressor for a Gaussian source with correlated side information y. Colored bands are the quantization regions, one color per index; each curve is the decoder's estimate for one index as a function of y. The same index is reused on intervals far apart, the binning used in information-theoretic proofs (Özyılkan, Ballé and Erkip, IEEE JSAIT 2024).

Faculty

Elza Erkip

Students

Ezgi Özyılkan