BIROn - Birkbeck Institutional Research Online

    Ranking-based color constancy with limited training samples

    Li, B. and Qin, H. and Xiong, W. and Li, Y. and Feng, S. and Hu, W. and Maybank, Stephen (2023) Ranking-based color constancy with limited training samples. IEEE Transactions on Pattern Analysis and Machine Intelligence , ISSN 0162-8828.

    RCC.pdf - Author's Accepted Manuscript

    Download (10MB) | Preview


    Computational color constancy is an important component of Image Signal Processors (ISP) for white balancing in many imaging devices. Recently deep convolutional neural networks (CNN) have been introduced for color constancy. They achieve prominent performance improvements comparing with those statistics or shallow learning-based methods. However, the need for a large number of training samples, a high computational cost and a huge model size make CNN-based methods unsuitable for deployment on low-resource ISPs for real time applications. In order to overcome these limitations and to achieve comparable perforance to CNN-based methods, an efficient method is defined for selecting the optimal simple statistics-based method (SM) for each image. To this end, we propose a novel ranking-based color constancy method (RCC) that formulates the selection of the optimal SM method as a label ranking problem. RCC designs a specific ranking loss function, and uses a low rank constraint to control the model complexity and a grouped sparse constraint for feature selection. Finally, we apply the RCC model to predict the order of the candidate SM methods for a test image, and then estimate its illumination using the predicted optimal SM method (or fusing the results estimated by the top k SM methods). Comprehensive experiment results show that the proposed RCC outperforms nearly all the shallow learning-based methods and achieves comparable performance to (sometimes even better performance than) deep CNN-based methods with only 1/2000 of the model size and training time. RCC also shows good robustness to limited training samples and good generalization crossing cameras. Furthermore, to remove the dependence on the ground truth illumination, we extend RCC to obtain a novel ranking-based method without ground truth illumination (RCC_NO) that learns the ranking model using simple partial binary preference annotations provided by untrained annotators rather than experts. RCC_NO also achieves better performance than the SM methods and most shallow learning-based methods with low costs of sample collection and illumination measurement.


    Item Type: Article
    Keyword(s) / Subject(s): Color constancy, illumination estimation, ranking, feature selection
    School: Birkbeck Faculties and Schools > Faculty of Science > School of Computing and Mathematical Sciences
    Depositing User: Steve Maybank
    Date Deposited: 24 May 2023 15:39
    Last Modified: 09 Aug 2023 12:54


    Activity Overview
    6 month trend
    6 month trend

    Additional statistics are available via IRStats2.

    Archive Staff Only (login required)

    Edit/View Item Edit/View Item