arXiv:2506.01346cs.CV2025-06被引 2

优化图像直方图分布,提升模型在恶劣天气下的分类准确率

Rethinking Image Histogram Matching for Image Classification

  • 设计可微分的参数化直方图匹配方法,动态学习最优目标分布
  • 在恶劣天气下分类准确率显著优于传统直方图均衡化方法
  • 适合需要鲁棒图像分类的自动驾驶、安防等实际场景

本文重新思考图像直方图匹配(HM)技术,提出一种可微且参数化的HM预处理方法,用于下游分类器。卷积神经网络在分类任务中表现优异,但在低对比度图像(如恶劣天气拍摄)上性能下降。为缓解此问题,常采用直方图均衡化(HE),其本质是将像素值分布对齐至均匀分布的特殊HM形式。本文聚焦于目标像素值分布的形状,发现单一精心设计的分布可能比均匀分布更优。基于此假设,提出一种利用下游分类器损失函数优化目标分布的可微分参数化HM方法。该方法通过将输入图像任意分布转换为针对分类器优化的目标分布,缓解像素值失衡问题。实验表明,仅在正常天气图像上训练,使用本方法的分类器在恶劣天气条件下表现优于传统预处理方法。

原文摘要 · Abstract (English)

This paper rethinks image histogram matching (HM) and proposes a differentiable and parametric HM preprocessing for a downstream classifier. Convolutional neural networks have demonstrated remarkable achievements in classification tasks. However, they often exhibit degraded performance on low-contrast images captured under adverse weather conditions. To maintain classifier performance under low-contrast images, histogram equalization (HE) is commonly used. HE is a special case of HM using a uniform distribution as a target pixel value distribution. In this paper, we focus on the shape of the target pixel value distribution. Compared to a uniform distribution, a single, well-designed distribution could have potential to improve the performance of the downstream classifier across various adverse weather conditions. Based on this hypothesis, we propose a differentiable and parametric HM that optimizes the target distribution using the loss function of the downstream classifier. This method addresses pixel value imbalances by transforming input images with arbitrary distributions into a target distribution optimized for the classifier. Our HM is trained on only normal weather images using the classifier. Experimental results show that a classifier trained with our proposed HM outperforms conventional preprocessing methods under adverse weather conditions.

图像增强分类模型可微处理鲁棒性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。