GeoCAD：基于大语言模型的局部几何可控CAD生成
计算机视觉与模式识别
2025-10-21 v2
摘要
局部几何可控的计算机辅助设计（CAD）生成旨在自动修改CAD模型的局部部件，提升设计效率。它还确保新生成的局部部件遵循用户特定的几何指令（例如等腰直角三角形或一角被切掉的矩形）。然而，现有方法在实现这一目标时面临挑战。具体而言，它们要么缺乏遵循文本指令的能力，要么无法聚焦于局部部件。为解决这一局限，我们引入了GeoCAD，一种用户友好且局部几何可控的CAD生成方法。具体而言，我们首先提出了一种互补描述策略，用于为局部部件生成几何指令。该策略涉及基于顶点的描述和基于VLLM的描述，分别用于系统标注简单和复杂的部件。通过这种方式，我们总共标注了约221k个不同的局部部件。在训练阶段，给定一个CAD模型，我们随机遮蔽一个局部部件。然后，利用其几何指令和其余部件作为输入，我们提示大语言模型（LLMs）来预测被遮蔽的部件。在推理阶段，用户可以指定任何局部部件进行修改，同时遵循各种预定义的几何指令。广泛的实验证明了GeoCAD在生成质量、有效性和文本到CAD一致性方面的有效性。代码将在https://github.com/Zhanwei-Z/GeoCAD上公开。
引用
@article{arxiv.2506.10337,
title = {GeoCAD: Local Geometry-Controllable CAD Generation with Large Language Models},
author = {Zhanwei Zhang and Kaiyuan Liu and Junjie Liu and Wenxiao Wang and Binbin Lin and Liang Xie and Chen Shen and Deng Cai},
journal= {arXiv preprint arXiv:2506.10337},
year = {2025}
}
备注
Accepted by NeurIPS 2025