跳到主要导航 跳到搜索 跳到主要内容

CAD-Coder: Text-to-CAD Generation with Chain-of-Thought and Geometric Reward

  • Yandong Guan
  • , Xilin Wang
  • , Ximing Xing
  • , Jing Zhang
  • , Dong Xu
  • , Qian Yu*
  • *此作品的通讯作者
  • Beihang University
  • The University of Hong Kong

科研成果: 书/报告/会议事项章节 › 会议稿件 › 同行评审

摘要

In this work, we introduce CAD-Coder, a novel framework that reformulates text-to-CAD as the generation of CadQuery scripts-a Python-based, parametric CAD language. This representation enables direct geometric validation, a richer modeling vocabulary, and seamless integration with existing LLMs. To further enhance code validity and geometric fidelity, we propose a two-stage learning pipeline: (1) supervised fine-tuning on paired text-CadQuery data, and (2) reinforcement learning with Group Reward Policy Optimization (GRPO), guided by a CAD-specific reward comprising both a geometric reward (Chamfer Distance) and a format reward. We also introduce a chain-of-thought (CoT) planning process to improve model reasoning, and construct a large-scale, high-quality dataset of 110K text-CadQuery-3D model triplets and 1.5K CoT samples via an automated pipeline. Extensive experiments demonstrate that CAD-Coder enables LLMs to generate diverse, valid, and complex CAD models directly from natural language, advancing the state of the art of text-to-CAD generation and geometric reasoning.

源语言英语
主期刊名39th Conference on Neural Information Processing Systems, NeurIPS 2025
出版商Neural information processing systems foundation
页66550-66574
页数25
ISBN(电子版)9798331338275
出版状态已出版 - 2025
活动39th Conference on Neural Information Processing Systems, NeurIPS 2025 - San Diego, 美国
期限: 2 12月 2025 → 7 12月 2025

丛书

姓名Advances in Neural Information Processing Systems
卷38
ISSN(印刷版)1049-5258

会议

会议39th Conference on Neural Information Processing Systems, NeurIPS 2025
国家/地区美国
市San Diego
时期2/12/25 → 7/12/25

学术指纹

探究 'CAD-Coder: Text-to-CAD Generation with Chain-of-Thought and Geometric Reward' 的科研主题。它们共同构成独一无二的学术指纹。

引用此