NegROI: Click-Centric Uncertainty-Guided Refinement with Scene-Conditioned Negative Prompts for Robust Interactive 3D Segmentation

· Source: cs.CV updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Robotics & Autonomous Systems, Computer Vision · Depth: Expert, long

Summary

NegROI is a novel transformer-based interactive framework designed for robust 3D segmentation in point clouds, addressing issues like coarse voxel resolution blurring fine boundaries and hard false positives from confusing background structures. It integrates click-centric multi-resolution refinement with scene-conditioned negative prompts. The system refines a local Region Of Interest (ROI) around the current click on a finer grid (fine scale η=2) and fuses these refined logits back to the coarse mask. It employs uncertainty-driven selective refinement (threshold τ=0.20), models background patterns via K=8 scene-conditioned negative prompts, and uses boundary-aware hard negative mining (k=256 voxels) with a diversity regularizer. Experiments on ScanNet, S3DIS, and KITTI demonstrate improved click efficiency, reduced false positives, and stronger cross-dataset robustness compared to baselines.

Key takeaway

For Machine Learning Engineers developing interactive 3D segmentation systems, NegROI offers a robust approach to improve boundary accuracy and reduce false positives. You should consider integrating click-centric multi-resolution ROI refinement and scene-conditioned negative prompts into your models. This strategy enhances cross-dataset generalization and click efficiency, especially for sparse interactions, by focusing computation on ambiguous regions and explicitly modeling background distractors.

Key insights

NegROI enhances interactive 3D segmentation by combining local, fine-grid refinement with explicit background modeling using scene-conditioned negative prompts.

Principles

Method

NegROI uses a sparse voxel backbone and two-way transformer. It decodes coarse logits, then refines a click-centric ROI on a finer grid, fusing results. Scene-conditioned negative prompts are appended to interaction tokens, supervised by boundary-aware hard negative mining and diversity regularization.

In practice

Topics

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.