Search papers, labs, and topics across Lattice.
This paper introduces Colorist, a novel data augmentation strategy that utilizes classical statistical color matching to enhance the robustness of medical image classifiers against domain shifts. By applying global mean-standard deviation matching in the RGB color space, Colorist generates domain variations that maintain structural integrity and color fidelity without the computational overhead of deep generative models. The method significantly improves balanced accuracy by up to 9% over existing domain generalization techniques and 13% over unaugmented baselines across various medical datasets, highlighting its effectiveness and efficiency.
Colorist outperforms complex generative models by safely generating clinically relevant domain variations while preserving anatomical structures.
Hardware shifts, color variations, and changing patient characteristics between development and deployment routinely break trained medical image classifiers. Existing remedies fall short: standard color jittering provides insufficient diversity, while deep generative style transfer algorithms hallucinate features, destroy clinically relevant structures, and waste massive compute resources. To address this, we revisit classical statistical color matching and repurpose it as Colorist, a highly efficient data augmentation strategy that applies global mean-standard deviation matching directly in the RGB color space. We demonstrate that this training-free, fully interpretable approach safely generates structurally intact domain variations, outperforming deep generative models in structural fidelity and color alignment. Across out-of-distribution histopathology, peripheral blood, dermatology, and retinal datasets, it improves balanced accuracy by up to +9% over state-of-the-art domain generalization regularizers and by +13% over an unaugmented baseline. Moreover, by avoiding neural networks in the augmentation loop, Colorist preserves anatomical structure, minimizes carbon footprint, and integrates seamlessly into standard dataloaders. Together, these findings establish statistical matching as a safe, interpretable, yet overlooked alternative to deep architectures for clinical robustness. Source code is available at https://github.com/sdoerrich97/colorist.
