OoDHDR-Codec: Out-of-Distribution Generalization for HDR Image Compression


  • Linfeng Cao Shanghai Jiao Tong University
  • Aofan Jiang Shanghai Jiao Tong University
  • Wei Li Huawei Noah's Ark Lab
  • Huaying Wu Shanghai Jiao Tong University
  • Nanyang Ye Shanghai Jiao Tong University




Computer Vision (CV)


Recently, deep learning has been proven to be a promising approach in standard dynamic range (SDR) image compression. However, due to the wide luminance distribution of high dynamic range (HDR) images and the lack of large standard datasets, developing a deep model for HDR image compression is much more challenging. To tackle this issue, we view HDR data as distributional shifts of SDR data and the HDR image compression can be modeled as an out-of-distribution generalization (OoD) problem. Herein, we propose a novel out-of-distribution (OoD) HDR image compression framework (OoDHDR-codec). It learns the general representation across HDR and SDR environments, and allows the model to be trained effectively using a large set of SDR datases supplemented with much fewer HDR samples. Specifically, OoDHDR-codec consists of two branches to process the data from two environments. The SDR branch is a standard blackbox network. For the HDR branch, we develop a hybrid system that models luminance masking and tone mapping with white-box modules and performs content compression with black-box neural networks. To improve the generalization from SDR training data on HDR data, we introduce an invariance regularization term to learn the common representation for both SDR and HDR compression. Extensive experimental results show that the OoDHDR codec achieves strong competitive in-distribution performance and state-of-the-art OoD performance. To the best of our knowledge, our proposed approach is the first work to model HDR compression as OoD generalization problems and our OoD generalization algorithmic framework can be applied to any deep compression model in addition to the network architectural choice demonstrated in the paper. Code available at https://github.com/caolinfeng/OoDHDR-codec.




How to Cite

Cao, L., Jiang, A., Li, W., Wu, H., & Ye, N. (2022). OoDHDR-Codec: Out-of-Distribution Generalization for HDR Image Compression. Proceedings of the AAAI Conference on Artificial Intelligence, 36(1), 158-166. https://doi.org/10.1609/aaai.v36i1.19890



AAAI Technical Track on Computer Vision I