COS3D: Collaborative Open-Vocabulary 3D Segmentation
Runsong Zhu, Ka-Hei Hui, Zhengzhe Liu, Qianyi Wu, Weiliang Tang, Shi Qiu, Pheng-Ann Heng, Chi-Wing Fu
TL;DR
COS3D tackles open-vocabulary 3D segmentation by introducing a collaborative field that jointly leverages instance (segmentation-aware) and language (text-aligned) cues. It learns an instance-to-language mapping to build a meaningful language field during training, and uses adaptive language-to-instance prompt refinement during inference to produce high-quality 3D segmentations from text prompts. The method achieves state-of-the-art results on LeRF and ScanNetv2, while delivering improved training efficiency and supporting versatile applications such as novel image-based segmentation, explicit hierarchical queries, and robotics. By integrating 3D-Gaussian Splatting, 2D foundation models (SAM/CLIP), and a two-stage learning strategy, COS3D offers a practical, scalable solution for open-vocabulary 3D perception in real-world scenarios.
Abstract
Open-vocabulary 3D segmentation is a fundamental yet challenging task, requiring a mutual understanding of both segmentation and language. However, existing Gaussian-splatting-based methods rely either on a single 3D language field, leading to inferior segmentation, or on pre-computed class-agnostic segmentations, suffering from error accumulation. To address these limitations, we present COS3D, a new collaborative prompt-segmentation framework that contributes to effectively integrating complementary language and segmentation cues throughout its entire pipeline. We first introduce the new concept of collaborative field, comprising an instance field and a language field, as the cornerstone for collaboration. During training, to effectively construct the collaborative field, our key idea is to capture the intrinsic relationship between the instance field and language field, through a novel instance-to-language feature mapping and designing an efficient two-stage training strategy. During inference, to bridge distinct characteristics of the two fields, we further design an adaptive language-to-instance prompt refinement, promoting high-quality prompt-segmentation inference. Extensive experiments not only demonstrate COS3D's leading performance over existing methods on two widely-used benchmarks but also show its high potential to various applications,~\ie, novel image-based 3D segmentation, hierarchical segmentation, and robotics. The code is publicly available at \href{https://github.com/Runsong123/COS3D}{https://github.com/Runsong123/COS3D}.
