Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Multi-branch Collaborative Learning Network for 3D Visual Grounding

About

3D referring expression comprehension (3DREC) and segmentation (3DRES) have overlapping objectives, indicating their potential for collaboration. However, existing collaborative approaches predominantly depend on the results of one task to make predictions for the other, limiting effective collaboration. We argue that employing separate branches for 3DREC and 3DRES tasks enhances the model's capacity to learn specific information for each task, enabling them to acquire complementary knowledge. Thus, we propose the MCLN framework, which includes independent branches for 3DREC and 3DRES tasks. This enables dedicated exploration of each task and effective coordination between the branches. Furthermore, to facilitate mutual reinforcement between these branches, we introduce a Relative Superpoint Aggregation (RSA) module and an Adaptive Soft Alignment (ASA) module. These modules significantly contribute to the precise alignment of prediction results from the two branches, directing the module to allocate increased attention to key positions. Comprehensive experimental evaluation demonstrates that our proposed method achieves state-of-the-art performance on both the 3DREC and 3DRES tasks, with an increase of 2.05% in Acc@0.5 for 3DREC and 3.96% in mIoU for 3DRES.

Zhipeng Qian, Yiwei Ma, Zhekai Lin, Jiayi Ji, Xiawu Zheng, Xiaoshuai Sun, Rongrong Ji• 2024

Related benchmarks

TaskDatasetResultRank
3D Visual GroundingScanRefer (val)
Overall Accuracy @ IoU 0.5045.7
155
3D Visual GroundingNr3D (test)
Overall Success Rate59.8
88
3D Visual GroundingSr3D (test)
Overall Accuracy68.4
73
3D Visual GroundingScanRefer Unique
Acc@0.25 (IoU=0.25)86.9
24
3D Visual GroundingScanRefer (test)
Unique Accuracy86.9
21
3D Visual GroundingScanRefer Multiple (val)
Accuracy @ IoU 0.2552
15
3D Visual GroundingScanRefer ScanNet v2 (val)
Unique Acc93.4
5
3D Visual GroundingARKitScenes (test)
Unique Success Rate61.2
5
Showing 8 of 8 rows

Other info

Follow for update