解决3D目标检测中框回归不准的问题,提升定位精度。
Decoupled and Interactive Regression Modeling for High-performance One-stage 3D Object Detection
- 解耦中心偏移与框属性回归,增强长距离预测能力。
- 融合分类任务优化低质量样本的置信度预测,提升可靠性。
- 在Waymo和ONCE数据集上达顶尖性能,推理开销几乎不变。
一阶段3D目标检测中,回归任务的边界框建模不足限制了性能提升。本研究发现其根源在于两点:(1) 中心偏移预测能力有限,导致大量响应最高位置显著偏离物体中心;(2) 回归任务忽略低质量样本,造成不可靠的交并比(IoU)修正。为此,提出解耦且交互的回归建模(DIRM)。具体地,解耦属性回归(DAR)通过自适应多样本分配策略,实现对中心属性的长范围回归建模;同时,交互质量预测(IQP)将分类任务与质量预测联合优化,利用分类擅长处理负样本的优势,提升低质量结果的IoU预测可靠性。在Waymo和ONCE数据集上的大量实验表明,DIRM显著提升多个先进方法的性能,且额外推理延迟极小。值得注意的是,该方法在两个数据集上均达到当前最优水平。
原文摘要 · Abstract (English)
Inadequate bounding box modeling in regression tasks constrains the performance of one-stage 3D object detection. Our study reveals that the primary reason lies in two aspects: (1) The limited center-offset prediction seriously impairs the bounding box localization since many highest response positions significantly deviate from object centers. (2) The low-quality sample ignored in regression tasks significantly impacts the bounding box prediction since it produces unreliable quality (IoU) rectification. To tackle these problems, we propose Decoupled and Interactive Regression Modeling (DIRM) for one-stage detection. Specifically, Decoupled Attribute Regression (DAR) is implemented to facilitate long regression range modeling for the center attribute through an adaptive multi-sample assignment strategy that deeply decouples bounding box attributes. On the other hand, to enhance the reliability of IoU predictions for low-quality results, Interactive Quality Prediction (IQP) integrates the classification task, proficient in modeling negative samples, with quality prediction for joint optimization. Extensive experiments on Waymo and ONCE datasets demonstrate that DIRM significantly improves the performance of several state-of-the-art methods with minimal additional inference latency. Notably, DIRM achieves state-of-the-art detection performance on both the Waymo and ONCE datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。