arXiv:2411.18860cs.CV2024-11被引 1

提升自动驾驶目标检测的鲁棒性,通过改进批量归一化增强测试时自适应能力

Improving Batch Normalization with TTA for Robust Object Detection in Self-Driving

论文配图:Improving Batch Normalization with TTA for Robust Object Detection in Self-Driving
图 1 · 摘自论文原文
  • 引入可学习的批归一化层,动态调整统计参数以适应不同输入
  • 提出双阶段语义一致性自适应策略,消除不稳定样本并优化搜索过程
  • 在多个恶劣场景下提升检测性能,最大增益达8%,适合自动驾驶感知系统

在当前开放真实世界自动驾驶场景中,传感器故障和极端天气等导致训练与测试数据间存在领域偏移,严重影响感知模型的泛化能力。随着模型参数规模增大,传统测试时自适应(TTA)方法变得不稳定,常导致性能下降。本文提出两种新方法,改进基于批量归一化的TTA在自动驾驶目标检测中的表现:(1) 提出基于广义搜索熵最小化(GSEM)的可学习批归一化(LearnableBN)层,通过引入辅助可学习参数,使批归一化层能根据输入数据动态更新统计量;(2) 设计一种基于语义一致性的双阶段自适应策略,引导模型迭代寻找最优解,并在适应过程中剔除不稳定样本。在NuScenes-C数据集上的大量实验表明,以BEVFormer为基线模型,本方法在六种损坏类型、三种严重程度下最高实现约8%的性能提升。源代码即将公开。

原文摘要 · Abstract (English)

In current open real-world autonomous driving scenarios, challenges such as sensor failure and extreme weather conditions hinder the generalization of most autonomous driving perception models to these unseen domain due to the domain shifts between the test and training data. As the parameter scale of autonomous driving perception models grows, traditional test-time adaptation (TTA) methods become unstable and often degrade model performance in most scenarios. To address these challenges, this paper proposes two new robust methods to improve the Batch Normalization with TTA for object detection in autonomous driving: (1) We introduce a LearnableBN layer based on Generalized-search Entropy Minimization (GSEM) method. Specifically, we modify the traditional BN layer by incorporating auxiliary learnable parameters, which enables the BN layer to dynamically update the statistics according to the different input data. (2) We propose a new semantic-consistency based dual-stage-adaptation strategy, which encourages the model to iteratively search for the optimal solution and eliminates unstable samples during the adaptation process. Extensive experiments on the NuScenes-C dataset shows that our method achieves a maximum improvement of about 8% using BEVFormer as the baseline model across six corruption types and three levels of severity. We will make our source code available soon.

自动驾驶目标检测测试时自适应批归一化

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