用视觉反馈提升大模型生成更符合设计意图的CAD参数序列
Text-to-CAD Generation Through Infusing Visual Feedback in Large Language Models
- 大模型分两阶段训练:先学参数逻辑,再根据渲染图反馈优化
- 相比仅用参数序列训练,生成结果在视觉上更符合人类偏好
- 适合需要文本快速生成可制造3D模型的设计人员
创建计算机辅助设计(CAD)模型需大量专业知识和时间。文本到CAD任务将文字描述转为参数化序列,有助于简化流程。现有研究依赖真实参数序列作为监督信号,但CAD本身是多模态的,包含参数序列与对应渲染图像。从参数序列到图像的映射是多对一的。因此,同时利用参数序列和视觉信号对有效训练至关重要。本文提出CADFusion框架,以大语言模型(LLM)为骨干,交替进行参数学习(SL)和视觉反馈(VF)两个阶段:在SL阶段,使用真实参数序列训练模型生成逻辑连贯的序列;在VF阶段,奖励能生成视觉上更优对象的序列,惩罚不符合视觉偏好的序列,使模型学习视觉感知与评价机制。两阶段交替训练,实现双信号均衡学习。实验表明,CADFusion显著提升生成质量,定性与定量表现均优于基线。
原文摘要 · Abstract (English)
Creating Computer-Aided Design (CAD) models requires significant expertise and effort. Text-to-CAD, which converts textual descriptions into CAD parametric sequences, is crucial in streamlining this process. Recent studies have utilized ground-truth parametric sequences, known as sequential signals, as supervision to achieve this goal. However, CAD models are inherently multimodal, comprising parametric sequences and corresponding rendered visual objects. Besides,the rendering process from parametric sequences to visual objects is many-to-one. Therefore, both sequential and visual signals are critical for effective training. In this work, we introduce CADFusion, a framework that uses Large Language Models (LLMs) as the backbone and alternates between two training stages: the sequential learning (SL) stage and the visual feedback (VF) stage. In the SL stage, we train LLMs using ground-truth parametric sequences, enabling the generation of logically coherent parametric sequences. In the VF stage, we reward parametric sequences that render into visually preferred objects and penalize those that do not, allowing LLMs to learn how rendered visual objects are perceived and evaluated. These two stages alternate throughout the training, ensuring balanced learning and preserving benefits of both signals. Experiments demonstrate that CADFusion significantly improves performance, both qualitatively and quantitatively.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。