arXiv:2601.12641cs.AI2026-01中稿 · the Design, Automa…被引 3

用大模型直接从自然语言生成可制造的CAD模型

STEP-LLM: Generating CAD STEP Models from Natural Language with Large Language Models

  • 构建4万对自然语言与STEP模型数据,改进格式以适配大模型生成
  • 通过检索增强和强化学习,生成模型几何精度显著提升
  • 适合制造业非专家快速实现设计构想,推动设计民主化

计算机辅助设计(CAD)在现代制造中至关重要,但建模仍依赖专业技能且耗时。现有基于大模型的文本转CAD方法多输出命令序列或脚本(如CadQuery),但此类格式依赖特定内核,通用性差。相比之下,标准产品数据交换格式(STEP,ISO 10303)是广泛使用的中性边界表示(B-rep)格式,直接兼容制造流程,但其图结构与跨引用特性给自回归大模型带来挑战。为此,我们构建了约4万组STEP-描述对,并提出针对STEP格式的新型预处理:基于深度优先搜索的重序列化,线性化跨引用同时保持局部性;以及链式思维风格的结构标注,引导全局一致性。引入检索增强生成(RAG)进行监督微调,以相关实例为依据;并通过基于切比雪夫距离的几何奖励,利用强化学习优化生成质量。实验表明,相比Text2CAD基线,STEP-LLM在几何保真度上持续提升:RAG模块显著增强完整性和可渲染性,DFS重序列化提高整体准确性,强化学习进一步减少几何偏差。定量与视觉对比均证实,STEP-LLM生成形状的保真度更高。结果证明了大模型从自然语言生成可制造的STEP模型的可行性,具有推动制造业设计民主化的潜力。

原文摘要 · Abstract (English)

Computer-aided design (CAD) is vital to modern manufacturing, yet model creation remains labor-intensive and expertise-heavy. To enable non-experts to translate intuitive design intent into manufacturable artifacts, recent large language models-based text-to-CAD efforts focus on command sequences or script-based formats like CadQuery. However, these formats are kernel-dependent and lack universality for manufacturing. In contrast, the Standard for the Exchange of Product Data (STEP, ISO 10303) file is a widely adopted, neutral boundary representation (B-rep) format directly compatible with manufacturing, but its graph-structured, cross-referenced nature poses unique challenges for auto-regressive LLMs. To address this, we curate a dataset of ~40K STEP-caption pairs and introduce novel preprocessing tailored for the graph-structured format of STEP, including a depth-first search-based reserialization that linearizes cross-references while preserving locality and chain-of-thought(CoT)-style structural annotations that guide global coherence. We integrate retrieval-augmented generation to ground predictions in relevant examples for supervised fine-tuning, and refine generation quality through reinforcement learning with a specific Chamfer Distance-based geometric reward. Experiments demonstrate consistent gains of our STEP-LLM in geometric fidelity over the Text2CAD baseline, with improvements arising from multiple stages of our framework: the RAG module substantially enhances completeness and renderability, the DFS-based reserialization strengthens overall accuracy, and the RL further reduces geometric discrepancy. Both metrics and visual comparisons confirm that STEP-LLM generates shapes with higher fidelity than Text2CAD. These results show the feasibility of LLM-driven STEP model generation from natural language, showing its potential to democratize CAD design for manufacturing.

CAD生成大模型STEP格式工业设计

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。