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

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision · Depth: Expert, quick

Summary

NegROI is a novel transformer-based interactive framework designed for robust 3D segmentation in point clouds, addressing limitations of existing methods such as coarse voxel resolution and persistent false positives. It tackles challenges posed by density and scale shifts across datasets like RGB-D reconstructions and LiDAR scans. The framework employs click-centric multi-resolution refinement, focusing on a local Region Of Interest (ROI) around user clicks on a finer grid and integrating refined logits into the coarse mask. To enhance efficiency and robustness, NegROI introduces uncertainty-driven selective refinement, prioritizing ambiguous areas. It also models difficult background patterns using scene-conditioned negative prompts, which are stabilized by a diversity regularizer. Furthermore, boundary-aware hard negative mining supervises negative-prompt attention towards high-confidence false positives near boundaries. Experiments on ScanNet, S3DIS, and KITTI demonstrate improved click efficiency, fewer false positives, and superior cross-dataset robustness over current state-of-the-art baselines.

Key takeaway

For Machine Learning Engineers developing interactive 3D segmentation systems, especially when facing challenges with fine boundaries or persistent false positives across diverse datasets, NegROI presents a robust solution. You should consider integrating its click-centric multi-resolution refinement and scene-conditioned negative prompts. This method improves click efficiency and reduces false positives, offering stronger cross-dataset robustness than current baselines.

Key insights

NegROI refines 3D segmentation using click-centric multi-resolution processing and scene-conditioned negative prompts for robustness across diverse datasets.

Principles

Method

NegROI refines coarse voxel predictions by focusing on a local ROI around clicks on a finer grid, fusing logits. It uses uncertainty-driven selective refinement and scene-conditioned negative prompts with diversity regularization and boundary-aware hard negative mining.

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 Artificial Intelligence.