Universal Rate-Distortion-Classification Representations for Lossy Compression
Nam Nguyen, Thuan Nguyen, Thinh Nguyen, Bella Bose
TL;DR
This work tackles multi-task lossy compression by introducing a universal RDC framework where a fixed encoder is paired with task-specific decoders to realize a range of distortion and classification constraints. For Gaussian sources under MSE, the authors prove zero rate penalty, showing that the full distortion-classification region can be achieved with a single encoder; for general sources they characterize the achievable region and bound the penalty via minimum mean-squared error and Wasserstein distance analyses. Theoretical results are complemented by experiments on MNIST that validate the universality concept and demonstrate distortion performance on par with task-specific encoders, aided by Wasserstein-regularized training. The findings support a practical universal representation approach for multi-task compression, enabling flexible deployment without retraining encoders across tasks.
Abstract
In lossy compression, Wang et al. [1] recently introduced the rate-distortion-perception-classification function, which supports multi-task learning by jointly optimizing perceptual quality, classification accuracy, and reconstruction fidelity. Building on the concept of a universal encoder introduced in [2], we investigate the universal representations that enable a broad range of distortion-classification tradeoffs through a single shared encoder coupled with multiple task-specific decoders. We establish, through both theoretical analysis and numerical experiments, that for Gaussian source under mean squared error (MSE) distortion, the entire distortion-classification tradeoff region can be achieved using a single universal encoder. For general sources, we characterize the achievable region and identify conditions under which encoder reuse results in negligible distortion penalty. The experimental result on the MNIST dataset further supports our theoretical findings. We show that universal encoders can obtain distortion performance comparable to task-specific encoders. These results demonstrate the practicality and effectiveness of the proposed universal framework in multi-task compression scenarios.
