用强化学习统一处理图像、文本、点云,实现多模态3D建模新突破
cadrille: Multi-modal CAD Reconstruction with Reinforcement Learning
- 三模态输入联合建模,借鉴大模型训练思路分两阶段优化
- 在DeepCAD上三项指标均超单模态方法,真实数据集达新纪录
- 首次验证强化学习适用于CAD生成,适合工业设计与AI辅助建模场景
计算机辅助设计(CAD)在工程制造中至关重要,可创建精确可编辑的3D模型。利用多种传感器或用户提供的数据作为输入,有助于降低设计应用门槛。然而,现有方法通常只针对单一模态(如点云、图像或文本),限制了泛化性和鲁棒性。我们基于视觉语言模型(VLM)提出一种可同时处理三种输入模态的多模态CAD重建模型。受大语言模型(LLM)训练范式启发,采用两阶段流程:首先在大规模程序生成数据上进行监督微调(SFT),然后通过程序化获取的在线反馈进行强化学习(RL)微调。此外,我们首次探索将强化学习用于CAD任务,证明在线算法如组相对偏好优化(GRPO)优于离线方案。在DeepCAD基准测试中,SFT模型在三种输入模态下均优于现有单模态方法。更重要的是,经过RL微调后,cadrille在三个具有挑战性的数据集上达到新状态,包括一个真实世界数据集。代码已开源:https://github.com/col14m/cadrille。
原文摘要 · Abstract (English)
Computer-Aided Design (CAD) plays a central role in engineering and manufacturing, making it possible to create precise and editable 3D models. Using a variety of sensor or user-provided data as inputs for CAD reconstruction can democratize access to design applications. However, existing methods typically focus on a single input modality, such as point clouds, images, or text, which limits their generalizability and robustness. Leveraging recent advances in vision-language models (VLM), we propose a multi-modal CAD reconstruction model that simultaneously processes all three input modalities. Inspired by large language model (LLM) training paradigms, we adopt a two-stage pipeline: supervised fine-tuning (SFT) on large-scale procedurally generated data, followed by reinforcement learning (RL) fine-tuning using online feedback, obtained programatically. Furthermore, we are the first to explore RL fine-tuning of LLMs for CAD tasks demonstrating that online RL algorithms such as Group Relative Preference Optimization (GRPO) outperform offline alternatives. In the DeepCAD benchmark, our SFT model outperforms existing single-modal approaches in all three input modalities simultaneously. More importantly, after RL fine-tuning, cadrille sets new state-of-the-art on three challenging datasets, including a real-world one. Code is avaliable at https://github.com/col14m/cadrille .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。