What Matters in Practical Learned Image Compression
For practitioners deploying image compression on mobile devices, this work provides a practical codec that significantly improves the speed-quality tradeoff.
This paper identifies key modeling choices for practical learned image codecs optimized for perceptual quality and runtime, and uses neural architecture search to find efficient models. The resulting codec achieves 2.3-3x bitrate savings over traditional codecs and 20-40% over learned alternatives, while encoding 12MP images in 230ms on an iPhone 17 Pro Max.
One of the major differentiators unlocked by learned codecs relative to their hard-coded traditional counterparts is their ability to be optimized directly to appeal to the human visual system. Despite this potential, a perceptual yet practical image codec is yet to be proposed. In this work, we aim to close this gap. We conduct a comprehensive study of the key modeling choices that govern the design of a practical learned image codec, jointly optimized for perceptual quality and runtime -- including within the ablations several novel techniques. We then perform performance-aware neural architecture search over millions of backbone configurations to identify models that achieve the target on-device runtime while maximizing compression performance as captured by perceptual metrics. We combine the various optimizations to construct a new codec that achieves a significantly improved tradeoff between speed and perceptual quality. Based on rigorous subjective user studies, it provides 2.3-3x bitrate savings against AV1, AV2, VVC, ECM and JPEG-AI, and 20-40% bitrate savings against the best learned codec alternatives. At the same time, on an iPhone 17 Pro Max, it encodes 12MP images as fast as 230ms, and decodes them in 150ms -- faster than most top ML-based codecs run on a V100 GPU.