Open-CRB: Towards Open World Active Learning for 3D Object Detection

Zhuoxiao Chen, Yadan Luo, Zixin Wang, Zijian Wang, Xin Yu +1 more
10/16/2023
cs.CVcs.AIcs.LG

Official code: Luoyadan/CRB-active-3Ddet (71 stars, Apache-2.0, last commit 2023-11-13).

Abstract

LiDAR-based 3D object detection has recently seen significant advancements through active learning (AL), attaining satisfactory performance by training on a small fraction of strategically selected point clouds. However, in real-world deployments where streaming point clouds may include unknown or novel objects, the ability of current AL methods to capture such objects remains unexplored. This paper investigates a more practical and challenging research task: Open World Active Learning for 3D Object Detection (OWAL-3D), aimed at acquiring informative point clouds with new concepts. To tackle this challenge, we propose a simple yet effective strategy called Open Label Conciseness (OLC), which mines novel 3D objects with minimal annotation costs. Our empirical results show that OLC successfully adapts the 3D detection model to the open world scenario with just a single round of selection. Any generic AL policy can then be integrated with the proposed OLC to efficiently address the OWAL-3D problem. Based on this, we introduce the Open-CRB framework, which seamlessly integrates OLC with our preliminary AL method, CRB, designed specifically for 3D object detection. We develop a comprehensive codebase for easy reproducing and future research, supporting 15 baseline methods (\textit{i.e.}, active learning, out-of-distribution detection and open world detection), 2 types of modern 3D detectors (\textit{i.e.}, one-stage SECOND and two-stage PV-RCNN) and 3 benchmark 3D datasets (\textit{i.e.}, KITTI, nuScenes and Waymo). Extensive experiments evidence that the proposed Open-CRB demonstrates superiority and flexibility in recognizing both novel and known classes with very limited labeling costs, compared to state-of-the-art baselines. Source code is available at \url{https://github.com/Luoyadan/CRB-active-3Ddet/tree/Open-CRB}.

View on arXivView PDF

Code Implementations(1)

Apache-2.0

The official PyTorch implementation of "Exploring Active 3D Object Detection from a Generalization Perspective" (ICLR Spotlight 2023 & TPAMI 2025).

719C, C++May 19, 20222 years agoApache-2.0
3d-detectionactive-learning

Similar Papers

Cite this paper

@article{chen2023opencrb,
  title  = {Open-CRB: Towards Open World Active Learning for 3D Object Detection},
  author = {Zhuoxiao Chen and Yadan Luo and Zixin Wang and Zijian Wang and Xin Yu and Zi Huang},
  year   = {2023},
  eprint = {2310.10391},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2310.10391v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2310.10391.svg)](https://codeofpaper.com/paper/2310.10391)

Discussion