arXiv:2409.03543cs.CVcs.AI2024-09被引 2

分析了真实场景下模型在分布偏移下的表现与置信度可靠性。

Prediction Accuracy & Reliability: Classification and Object Localization under Distribution Shift

  • 对比不同模型在天气变化和异常情况下的表现
  • 重雨对分类影响大于定位,而浓雾则相反
  • 仅在特定层加入蒙特卡洛丢弃可提升性能与置信度

自然分布偏移导致卷积神经网络(CNN)感知性能下降。本研究基于公开自动驾驶数据集构建新数据集,涵盖单一目标裁剪图像(含类别与边界框标注),以及六类分布偏移数据:恶劣天气、模拟雨雾、边缘案例和分布外数据。分析了分布偏移及天气增强对检测质量与置信度估计的影响,评估了分类与目标定位任务的模型性能,并在自然与接近自然分布偏移下基准化两种不确定性量化方法:集成与多种蒙特卡洛丢弃变体。结果表明:ConvNeXt-Tiny 比 EfficientNet-B0 更鲁棒;重雨显著降低分类性能,而对定位影响较小,与浓雾情况相反;仅在特定层引入蒙特卡洛丢弃可提升任务表现与置信度估计,且最优层依赖于偏移类型与任务类型。

原文摘要 · Abstract (English)

Natural distribution shift causes a deterioration in the perception performance of convolutional neural networks (CNNs). This comprehensive analysis for real-world traffic data addresses: 1) investigating the effect of natural distribution shift and weather augmentations on both detection quality and confidence estimation, 2) evaluating model performance for both classification and object localization, and 3) benchmarking two common uncertainty quantification methods - Ensembles and different variants of Monte-Carlo (MC) Dropout - under natural and close-to-natural distribution shift. For this purpose, a novel dataset has been curated from publicly available autonomous driving datasets. The in-distribution (ID) data is based on cutouts of a single object, for which both class and bounding box annotations are available. The six distribution-shift datasets cover adverse weather scenarios, simulated rain and fog, corner cases, and out-of-distribution data. A granular analysis of CNNs under distribution shift allows to quantize the impact of different types of shifts on both, task performance and confidence estimation: ConvNeXt-Tiny is more robust than EfficientNet-B0; heavy rain degrades classification stronger than localization, contrary to heavy fog; integrating MC-Dropout into selected layers only has the potential to enhance task performance and confidence estimation, whereby the identification of these layers depends on the type of distribution shift and the considered task.

分布偏移置信度估计自动驾驶模型鲁棒性

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