ISSN :2582-9793

Improving CNN Robustness to Color Shifts via Color Balancing and Spatial Dropout

Original Research (Published On: 31-Mar-2025 )
DOI : https://dx.doi.org/10.54364/AAIML.2025.51205

Pradyumna Elavarthi and Anca Ralescu

Adv. Artif. Intell. Mach. Learn., 5 (1):3605-3626

Pradyumna Elavarthi : University of Cincinnati

Anca Ralescu : University of Cincinnati

Download PDF Here

DOI: https://dx.doi.org/10.54364/AAIML.2025.51205

Article History: Received on: 01-Jan-25, Accepted on: 24-Mar-25, Published on: 31-Mar-25

Corresponding Author: Pradyumna Elavarthi

Email: elavarpa@mail.uc.edu

Citation: Pradyumna Elavarthi, Anca Ralescu. (2025). Improving CNN Robustness to Color Shifts via Color Balancing and Spatial Dropout. Adv. Artif. Intell. Mach. Learn., 5 (1):3605-3626.


Abstract

    

Convolutional neural networks (CNNs) have demonstrated remarkable success in vision-related tasks. However, their susceptibility to failing when inputs deviate from the training distribution is well-documented. Recent studies suggest that CNNs exhibit a bias toward texture instead of object shape in image classification tasks, and that background information may affect predictions. This paper investigates the ability of CNNs to adapt to different color distributions of an image while maintaining context and background. The results of our experiments on modified MNIST, CIFAR10 and CIFAR 100 data demonstrate that changes in color can substantially affect classification accuracy. The paper explores the effects of various regularization techniques on generalization error across datasets and proposes an architectural modification using in a novel way color balancing and spatial dropout regularization. This enhances the model reliance on color-invariant intensity-based features for improved classification accuracy. Overall, this work contributes to ongoing efforts to understand the limitations and challenges of CNNs in image classification tasks and offers a potential solution to improve their performance.

Statistics

   Article View: 38
   PDF Downloaded: 1