Score: 2

Perception-Inspired Color Space Design for Photo White Balance Editing

Published: December 10, 2025 | arXiv ID: 2512.09383v2

By: Yang Cheng , Ziteng Cui , Shenghan Su and more

Potential Business Impact:

Fixes photos' colors under tricky lights.

Business Areas:
Photo Editing Content and Publishing, Media and Entertainment

White balance (WB) is a key step in the image signal processor (ISP) pipeline that mitigates color casts caused by varying illumination and restores the scene's true colors. Currently, sRGB-based WB editing for post-ISP WB correction is widely used to address color constancy failures in the ISP pipeline when the original camera RAW is unavailable. However, additive color models (e.g., sRGB) are inherently limited by fixed nonlinear transformations and entangled color channels, which often impede their generalization to complex lighting conditions. To address these challenges, we propose a novel framework for WB correction that leverages a perception-inspired Learnable HSI (LHSI) color space. Built upon a cylindrical color model that naturally separates luminance from chromatic components, our framework further introduces dedicated parameters to enhance this disentanglement and learnable mapping to adaptively refine the flexibility. Moreover, a new Mamba-based network is introduced, which is tailored to the characteristics of the proposed LHSI color space. Experimental results on benchmark datasets demonstrate the superiority of our method, highlighting the potential of perception-inspired color space design in computational photography. The source code is available at https://github.com/YangCheng58/WB_Color_Space.

Country of Origin
🇨🇳 🇯🇵 Japan, China

Repos / Data Links

Page Count
9 pages

Category
Computer Science:
CV and Pattern Recognition