arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2401.05925cs.CVcs.AI

通过高效特征反投影学习分段3D高斯以实现零样本神经场景分割

Learning Segmented 3D Gaussians via Efficient Feature Unprojection for Zero-shot Neural Scene Segmentation

Institute of Artificial Intelligence and Robotics, Xi’an Jiaotong University(西安交通大学人工智能与机器人研究所)

Bin Dou, Tianyu Zhang, Zhaohui Wang, Yongjia Ma, Zejian Yuan

AI总结:

提出CoSegGaussians,通过Feature Unprojection and Fusion模块和CoSeg Loss解决3D高斯零样本分割中冗余属性导致的3D不一致性问题,mIoU提升约10%。

摘要:

零样本神经场景分割在无需人工标注的情况下重建3D神经分割场,是场景理解的有效方式。然而,现有模型,尤其是高效的基于3D高斯的方法,难以产生紧凑的分割结果。该问题主要源于分配在单个高斯上的冗余可学习属性,导致对零样本生成的原始标签中的3D不一致性缺乏鲁棒性。为解决此问题,我们的工作名为Compact Segmented 3D Gaussians (CoSegGaussians),提出将Feature Unprojection and Fusion模块作为分割场,该模块利用基于高级特征、对所有高斯具有泛化能力的浅层解码器。具体而言,利用学习到的高斯几何参数,通过我们的反投影技术将语义感知的基于图像的特征引入场景。提升后的特征连同空间信息被输入到多尺度聚合解码器中,为所有高斯生成分割标识。此外,我们设计了CoSeg Loss以提升模型对3D不一致噪声的鲁棒性。实验结果表明,我们的模型在零样本语义分割任务上超越了基线,mIoU比最佳基线提高了约10%。代码和更多结果将在https://David-Dou.github.io/CoSegGaussians发布。

英文摘要:

Zero-shot neural scene segmentation, which reconstructs 3D neural segmentation field without manual annotations, serves as an effective way for scene understanding. However, existing models, especially the efficient 3D Gaussian-based methods, struggle to produce compact segmentation results. This issue stems primarily from their redundant learnable attributes assigned on individual Gaussians, leading to a lack of robustness against the 3D-inconsistencies in zero-shot generated raw labels. To address this problem, our work, named Compact Segmented 3D Gaussians (CoSegGaussians), proposes the Feature Unprojection and Fusion module as the segmentation field, which utilizes a shallow decoder generalizable for all Gaussians based on high-level features. Specifically, leveraging the learned Gaussian geometric parameters, semantic-aware image-based features are introduced into the scene via our unprojection technique. The lifted features, together with spatial information, are fed into the multi-scale aggregation decoder to generate segmentation identities for all Gaussians. Furthermore, we design CoSeg Loss to boost model robustness against 3D-inconsistent noises. Experimental results show that our model surpasses baselines on zero-shot semantic segmentation task, improving by ~10% mIoU over the best baseline. Code and more results will be available at https://David-Dou.github.io/CoSegGaussians.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑