摘要
arXiv:2502.12418v1 交叉类型: cross
摘要: 色度恒定估计光源色度以校正色彩偏差的图像。近年来,基于深度神经网络的色度恒定(DNNCC)模型取得了显著进展。然而,由于深度神经网络的脆弱性,DNNCC 中的风险尚未被探索。在本文中,我们首次从鲁棒性的角度调查了亮度这一色度恒定的关键因素对 DNNCC 的影响。我们的评估表明,尽管主流的 DNNCC 模型专注于色度估计,但它们对亮度的高敏感性仍然存在。这揭示了现有 DNNCC 模型的一个潜在限制:它们对亮度的敏感性可能因现实世界数据集中普遍存在的亮度变化而影响其性能。从我们分析的见解出发,我们提出了一种简单而有效的亮度鲁棒性增强策略,该策略适用于 DNNCC 模型,称为 BRE。BRE 的核心基于自适应步长对抗亮度增广技术,该技术识别高风险亮度变化并通过显式亮度调整生成增广图像。随后,BRE 开发了一种亮度鲁棒性感知的模型优化策略,该策略结合了对抗亮度训练和亮度对比损失,大幅增强了 DNNCC 模型的亮度鲁棒性。BRE 无超参数,并且可以在现有的 DNNCC 模型中集成,而不会在测试阶段增加额外的开销。在两个公开的色度恒定数据集——ColorChecker 和 Cube+ 上的实验表明,所提出的 BRE 一致地提高了现有 DNNCC 模型的光源估计性能,平均降低估计误差 5.04%,突显了增强亮度鲁棒性在这些模型中的关键作用。