Towards High-Fidelity CAD Generation via LLM-Driven Program Generation and Text-Based B-Rep Primitive Grounding
Organizations: Fudan University · Shanghai Jiao Tong University
Abstract
The field of Computer-Aided Design (CAD) generation has made significant progress in recent years. Existing methods typically fall into two separate categories: parametric CAD modeling and direct boundary representation (B-Rep) synthesis. In modern feature-based CAD systems, parametric modeling and B-Rep are inherently intertwined, as advanced parametric operations (e.g., fillet and chamfer) require explicit selection of B-Rep geometric primitives, and the B-Rep itself is derived from parametric operations. Consequently, this paradigm gap remains a critical factor limiting AI-driven CAD modeling for complex industrial product design. This paper presents FutureCAD, a novel text-to-CAD framework that leverages large language models (LLMs) and a B-Rep grounding transformer (BRepGround) for high-fidelity CAD generation. Our method generates executable CadQuery scripts, and introduces a text-based query mechanism that enables the LLM to specify geometric selections via natural language, which BRepGround then grounds to the target primitives. To train our framework, we construct a new dataset comprising real-world CAD models. For the LLM, we apply supervised fine-tuning (SFT) to establish fundamental CAD generation capabilities, followed by reinforcement learning (RL) to improve generalization. Experiments show that FutureCAD achieves state-of-the-art CAD generation performance. Code and dataset are available at: https://github.com/JohanStackk/FutureCAD
Figures & tables
| Methods | FutureCAD Dataset | Fusion 360 (Out-of-Distribution) | ||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 92pt. | Advanced | Standard | ||||||||||||||
| G -F1 | R -F1 | Median CD | Mean CD | IR | G -F1 | Median CD | Mean CD | IR | G -F1 | Median CD | Mean CD | IR | ||||
| CAD-Translator | - | - | - | - | - | 79.08 | 94.23 | 152.60 | 3.57 | 76.82 | 159.98 | 211.72 | 3.86 | |||
| Text2CAD | - | - | - | - | - | 81.39 | 77.65 | 132.04 | 1.90 | 80.24 | 112.77 | 178.42 | 3.56 | |||
| Text-to-CadQuery | - | - | - | - | - | 87.04 | 109.83 | 173.69 | 8.38 | 84.36 | 188.88 | 232.56 | 11.70 | |||
| cadrille | - | - | - | - | - | 84.13 | 71.89 | 125.58 | 2.57 | 84.53 | 143.04 | 206.93 | 5.02 | |||
| Methods | Recall@3 | Recall@5 | Recall@10 | mAP | F1 |
|---|---|---|---|---|---|
| LLM-Query | - | - | - | - | 21.51 |
| CLIP-DE | 51.81 | 62.57 | 75.01 | 57.53 | 31.05 |
| Late Fusion | 51.43 | 63.55 | 76.72 | 58.18 | 43.32 |
| Ours | 56.39 | 65.86 | 78.47 | 63.07 | 50.23 |
| Methods | G -F1 | R -F1 | Median CD | Mean CD | IR(%) |
|---|---|---|---|---|---|
| SFT only | 79.82 | 91.44 | 44.00 | 85.15 | 42.67 |
| w/o BRepGround | 80.36 | 82.22 | 38.50 | 76.01 | 8.29 |
| Ours | 80.54 | 85.35 | 31.12 | 71.39 | 1.01 |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
Text-to-CAD Generation
Text-to-CAD generation turns natural-language design intent into executable, editable parametric 3D models and assemblies.
See all 48 papers