AssemLM: A Spatial Reasoning Multimodal Large Language Model for Robotic Assembly
About
Spatial reasoning is a fundamental capability for embodied intelligence, especially for fine-grained manipulation tasks such as robotic assembly. Recent methods based on vision-language models (VLMs) largely rely on coarse 2D perception and struggle to perform accurate reasoning over complex 3D geometry. To address this limitation, we propose AssemLM, a spatial multimodal large language model for robotic assembly that integrates assembly manuals, point clouds, and textual instructions to predict task-critical 6D assembly poses with explicit geometric understanding. To bridge raw 3D perception and high-level linguistic reasoning, AssemLM employs a specialized point cloud encoder to extract fine-grained geometric and rotational features for accurate 3D spatial reasoning in assembly tasks. In addition, we introduce AssemBench, a large-scale benchmark for assembly-oriented spatial reasoning with over 900K multimodal samples and precise 6D pose annotations, extending evaluation from 2D grounding to full 3D geometric inference. Extensive experiments and real-robot evaluations demonstrate that AssemLM achieves state-of-the-art 6D pose reasoning performance and effectively supports fine-grained, multi-step assembly tasks in real-world settings. Code, models, and the AssemBench dataset will be made publicly available.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Structural Furniture Assembly | IKEA Bench category | RMSE (Translation)0.033 | 4 | |
| Structural Furniture Assembly | IKEA Chair category | RMSE (Translation)0.0205 | 4 | |
| Structural Furniture Assembly | IKEA Desk category | RMSE (Translation)0.0255 | 4 | |
| Structural Furniture Assembly | IKEA All categories | RMSE (Translation)0.0263 | 4 | |
| Assembly Pose Prediction | AssemBench Furniture 1.0 | -- | 4 | |
| Build Blocks | Real-world Assembly Assets | Success Rate9 | 2 | |
| Insert Flower | Real-world Assembly Assets | Success Rate20 | 2 | |
| Insert Plug | Real-world Assembly Assets | Success Rate13 | 2 | |
| Overall Assembly Tasks | Real-world Assembly Assets | Success Rate50.8 | 2 | |
| Store Cans | Real-world Assembly Assets | Success Rate19 | 2 |