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

多视图分层图神经网络用于基于草图的3D形状检索

Multi-View Hierarchical Graph Neural Network for Sketch-Based 3D Shape Retrieval

Tsinghua Shenzhen International Graduate School, Tsinghua University(清华大学深圳国际研究生院);Control and Simulation Center, Harbin Institute of Technology(哈尔滨工业大学控制仿真中心)

Hang Cheng, Muyan He, Mingyu Fan, Chengfeng Xie, Xi Cheng, Long Zeng

AI总结:

本文提出MV-HGNN框架,通过多视图图神经网络和视图选择器提升3D形状检索的表示能力,结合CLIP文本嵌入实现类别无关对齐,实验表明在类别级和零样本场景中优于现有方法。

摘要:

基于草图的3D形状检索(SBSR)旨在检索与输入手绘草图类别一致的3D形状。该任务的核心挑战在于现有方法通常采用简化聚合策略独立编码3D多视图特征,忽略了视图间的几何关系和多级细节,导致3D表示薄弱。同时,传统SBSR方法受限于可见类别限制,在零样本场景中表现不佳。为解决这些问题,我们提出多视图分层图神经网络(MV-HGNN)框架。具体而言,我们构建视图级图并利用局部图卷积和全局注意力捕捉相邻几何依赖和跨视图信息传递。进一步引入视图选择器执行层次图粗化,使图卷积具有逐步扩大的感受野并缓解冗余视图干扰,从而获得更具判别性的层次3D表示。为实现类别无关对齐并缓解对已见类别的过拟合,我们利用CLIP文本嵌入作为语义原型,并将草图和3D特征投影到共享语义空间。我们采用两阶段训练策略进行类别级检索,并在相同模型架构下采用一阶段策略进行零样本检索。在类别级和零样本设置下,大量实验表明MV-HGNN在两个公开基准上优于现有方法。

英文摘要:

Sketch-based 3D shape retrieval (SBSR) aims to retrieve 3D shapes that are consistent with the category of the input hand-drawn sketch. The core challenge of this task lies in two aspects: existing methods typically employ simplified aggregation strategies for independently encoded 3D multi-view features, which ignore the geometric relationships between views and multi-level details, resulting in weak 3D representation. Simultaneously, traditional SBSR methods are constrained by visible category limitations, leading to poor performance in zero-shot scenarios. To address these challenges, we propose Multi-View Hierarchical Graph Neural Network (MV-HGNN), a novel framework for SBSR. Specifically, we construct a view-level graph and capture adjacent geometric dependencies and cross-view message passing via local graph convolution and global attention. A view selector is further introduced to perform hierarchical graph coarsening, enabling a progressively larger receptive field for graph convolution and mitigating the interference of redundant views, which leads to more discriminate discriminative hierarchical 3D representation. To enable category agnostic alignment and mitigate overfitting to seen classes, we leverage CLIP text embeddings as semantic prototypes and project both sketch and 3D features into a shared semantic space. We use a two-stage training strategy for category-level retrieval and a one-stage strategy for zero-shot retrieval under the same model architecture. Under both category-level and zero-shot settings, extensive experiments on two public benchmarks demonstrate that MV-HGNN outperforms state-of-the-art methods.

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

↑