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

RI-Mamba:用于鲁棒文本到形状检索的旋转不变Mamba

RI-Mamba: Rotation-Invariant Mamba for Robust Text-to-Shape Retrieval

The University of Western Australia(西澳大学)

Khanh Nguyen, Dasith de Silva Edirimuni, Ghulam Mubashar Hassan, Ajmal Mian

AI总结:

RI-Mamba是一种旋转不变的状态空间模型,用于提升文本到形状检索的鲁棒性和表现力。

摘要:

随着虚拟现实和游戏的日益流行,3D资产在数量和多样性上迅速增长。因此,在大型存储库中进行直观搜索时,文本到形状检索已成为必不可少的。然而,现有方法需要标准姿态并支持少数物体类别,限制了其在现实中的应用,因为物体可以属于不同的类别并以随机方向出现。为了解决这一挑战,我们提出了RI-Mamba,这是第一个针对点云的旋转不变状态空间模型。RI-Mamba定义了全局和局部参考框架,以分离姿态和几何,并利用Hilbert排序构建具有有意义几何结构的token序列,同时保持旋转不变性。我们进一步引入了一种新的策略来计算方向嵌入,并通过特征级线性调制重新整合它们,有效地恢复空间上下文并增强模型的表达能力。我们的策略本质上与状态空间模型兼容,并且在线性时间内运行。为了扩大检索规模,我们采用跨模态对比学习与自动三元组生成,允许在无需人工标注的情况下训练于多样化的数据集。广泛的实验表明,RI-Mamba在表示能力和鲁棒性方面均优于现有方法,在OmniObject3D基准测试中,超过200个物体类别下任意方向下均实现了最先进的性能。我们的代码将在https://github.com/ndkhanh360/RI-Mamba.git上公开。

英文摘要:

3D assets have rapidly expanded in quantity and diversity due to the growing popularity of virtual reality and gaming. As a result, text-to-shape retrieval has become essential in facilitating intuitive search within large repositories. However, existing methods require canonical poses and support few object categories, limiting their real-world applicability where objects can belong to diverse classes and appear in random orientations. To address this challenge, we propose RI-Mamba, the first rotation-invariant state-space model for point clouds. RI-Mamba defines global and local reference frames to disentangle pose from geometry and uses Hilbert sorting to construct token sequences with meaningful geometric structure while maintaining rotation invariance. We further introduce a novel strategy to compute orientational embeddings and reintegrate them via feature-wise linear modulation, effectively recovering spatial context and enhancing model expressiveness. Our strategy is inherently compatible with state-space models and operates in linear time. To scale up retrieval, we adopt cross-modal contrastive learning with automated triplet generation, allowing training on diverse datasets without manual annotation. Extensive experiments demonstrate RI-Mamba's superior representational capacity and robustness, achieving state-of-the-art performance on the OmniObject3D benchmark across more than 200 object categories under arbitrary orientations. Our code will be made available at https://github.com/ndkhanh360/RI-Mamba.git.

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

↑