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

MiniGPT-3D: Efficiently Aligning 3D Point Clouds with Large Language Models using 2D Priors

About

Large 2D vision-language models (2D-LLMs) have gained significant attention by bridging Large Language Models (LLMs) with images using a simple projector. Inspired by their success, large 3D point cloud-language models (3D-LLMs) also integrate point clouds into LLMs. However, directly aligning point clouds with LLM requires expensive training costs, typically in hundreds of GPU-hours on A100, which hinders the development of 3D-LLMs. In this paper, we introduce MiniGPT-3D, an efficient and powerful 3D-LLM that achieves multiple SOTA results while training for only 27 hours on one RTX 3090. Specifically, we propose to align 3D point clouds with LLMs using 2D priors from 2D-LLMs, which can leverage the similarity between 2D and 3D visual information. We introduce a novel four-stage training strategy for modality alignment in a cascaded way, and a mixture of query experts module to adaptively aggregate features with high efficiency. Moreover, we utilize parameter-efficient fine-tuning methods LoRA and Norm fine-tuning, resulting in only 47.8M learnable parameters, which is up to 260x fewer than existing methods. Extensive experiments show that MiniGPT-3D achieves SOTA on 3D object classification and captioning tasks, with significantly cheaper training costs. Notably, MiniGPT-3D gains an 8.12 increase on GPT-4 evaluation score for the challenging object captioning task compared to ShapeLLM-13B, while the latter costs 160 total GPU-hours on 8 A800. We are the first to explore the efficient 3D-LLM, offering new insights to the community. Code and weights are available at https://github.com/TangYuan96/MiniGPT-3D.

Yuan Tang, Xu Han, Xianzhi Li, Qiao Yu, Yixue Hao, Long Hu, Min Chen• 2024

Related benchmarks

TaskDatasetResultRank
3D Object ClassificationObjaverse
Average Accuracy60.25
19
3D Object ClassificationModelNet40 (test)
Average Accuracy60.86
17
6DoF object manipulation trajectory generationHOT3D
3D Positional ADE0.281
12
3DoF object manipulation trajectory generationHOT3D
3D ADE0.299
12
3D Object Captioning3D Objects
Sentence-BERT Score47.64
11
3D Object Recognition3D Objects
Recognition Accuracy53.52
11
3D Object CaptioningObjaverse
GPT-4 Performance Score57.06
10
3D Object CaptioningBrep2Text 1.0 (test)
Qwen-Max Score56.58
6
4D Object CaptioningObjaverse 4,000 IDs (test)
GPT-4 Score54.7
6
4D Object Question AnsweringObjaverse 4,000 object IDs (test)
GPT-4 Score59.08
6
Showing 10 of 11 rows

Other info

Code

Follow for update