统一多数据集3D检测框架,提升跨域泛化能力
Uni$^2$Det: Unified and Universal Framework for Prompt-Guided Multi-dataset 3D Detection
- 用分阶段提示模块适配不同数据集特征差异
- 在KITTI、Waymo、nuScenes上显著优于现有方法
- 支持零样本跨数据集迁移,适合工业级部署
我们提出Uni$^2$Det,一种统一且通用的多数据集3D检测训练框架,可在不同领域间实现稳健性能并具备未见领域的泛化能力。由于不同领域间数据分布差异大、分类体系不一致,简单合并数据集训练面临巨大挑战。为此,我们设计了多阶段提示模块,利用各数据集特性生成提示,有效缓解差异。该设计可无缝嵌入各类先进3D检测框架,统一集成且易于扩展至多数据集通用场景。在KITTI、Waymo、nuScenes等多组数据融合实验中,Uni$^2$Det显著超越现有方法;零样本跨数据集迁移结果验证了其强大泛化能力。
原文摘要 · Abstract (English)
We present Uni$^2$Det, a brand new framework for unified and universal multi-dataset training on 3D detection, enabling robust performance across diverse domains and generalization to unseen domains. Due to substantial disparities in data distribution and variations in taxonomy across diverse domains, training such a detector by simply merging datasets poses a significant challenge. Motivated by this observation, we introduce multi-stage prompting modules for multi-dataset 3D detection, which leverages prompts based on the characteristics of corresponding datasets to mitigate existing differences. This elegant design facilitates seamless plug-and-play integration within various advanced 3D detection frameworks in a unified manner, while also allowing straightforward adaptation for universal applicability across datasets. Experiments are conducted across multiple dataset consolidation scenarios involving KITTI, Waymo, and nuScenes, demonstrating that our Uni$^2$Det outperforms existing methods by a large margin in multi-dataset training. Notably, results on zero-shot cross-dataset transfer validate the generalization capability of our proposed method.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。