Lossy Source Coding with Focal Loss
Alex Dytso, Martina Cardone
TL;DR
This work introduces focal loss as a distortion measure for lossy source coding with soft reconstruction, unifying ideas from machine learning with rate-distortion theory. It derives single-shot converse and achievability bounds and proves that the asymptotic distortion-rate function $\mathsf{D}(\mathsf{R};\gamma)$ coincides with the log-loss case, i.e., $[H(X) - \mathsf{R}]^+$, while demonstrating that the focal term influences finite-blocklength performance. The analysis relies on an entropy-like quantity $H_\gamma(P_X)$ and a maximum-entropy bound $h_\gamma(|\mathcal{X}|)$, and introduces an auxiliary distribution $F_X$ to optimize the achievability bound, with the optimal choice asymptotically being $F_X = R_X$. Numerical examples with small alphabets and a Binomial source illustrate non-asymptotic differences induced by the focal term and highlight potential privacy implications for negative $\gamma$. Overall, the results provide a principled framework for focal-loss-based distortion in rate-distortion theory and point to future work on non-asymptotic tightening, excess-distortion analysis, multiterminal extensions, and privacy-oriented applications.
Abstract
Focal loss has recently gained significant popularity, particularly in tasks like object detection where it helps to address class imbalance by focusing more on hard-to-classify examples. This work proposes the focal loss as a distortion measure for lossy source coding. The paper provides single-shot converse and achievability bounds. These bounds are then used to characterize the distortion-rate trade-off in the infinite blocklength, which is shown to be the same as that for the log loss case. In the non-asymptotic case, the difference between focal loss and log loss is illustrated through a series of simulations.
