StepCAD:让CAD代码每做一步都看几何,再用搜索修正
从网格到可执行建模程序;高IoU、有效实体与可用设计约束不是同一件事
StepCAD先用目标点云与当前实体共同条件化Qwen2.5-Coder,逐步生成并执行CadQuery操作,再用几何重合度指导局部搜索。搜索不只调半径和深度,还会替换截面、补切多余材料、删去有害操作。配套ARCADE-1.5M提供150万合成程序及1,250万中间状态;困难形状重建提升明显,但优化的目标是形状相似,尚不保证原始特征历史、参数语义或制造约束恢复。
图解主要方法
在真实几何反馈下生成并搜索CAD程序,能改善多少重建,又留下哪些可编辑性问题?

- 1
生成带中间实体的训练程序
以ABC草图环为基础,94种启发式策略组合拉伸、旋转、放样、扫掠、布尔、壳体和收尾特征;每步要求单实体有效且正体积,程序在新命名空间重执行验证后导出STEP。150多“操作”含草图原语,不能与策略动作/API调用混算。
- 2
把目标与当前形状同时送给代码策略
每种几何采1,024点,共享Point-MAE经FPS成64组、每组32邻点,再生成64几何token;额外中心/尺度token保留归一化丢失的姿态。Qwen2.5-Coder-1.5B读取这些token和前序代码,生成下一构造动作。
- 3
执行并反馈实际产生的实体
每步CadQuery执行、重新采点条件化;IoU提升低于0.001、结束词或12动作上限停止。一个动作可含草图、实体及布尔组合,因此12动作不是12条API,也不证明支持训练集中全部最长程序。
- 4
用四类编辑开展窄束搜索
beam宽2,按构造顺序试连续参数BO、目标截面替换草图、差集补切、删除操作。BO用6随机点加10轮每轮4候选,普通参数局部半径受限;草图只取最大有效闭环,补切至多两轮,均依赖真实执行和IoU评分。
- 5
将重建结果送入工程核查
输出可执行且有参数的程序,不声称找回唯一原始历史。还要测局部尺寸、特征含义和修改稳定性;网格来自图像生成器时,其形状错误也可能进入CAD结果。
实验与证据
43页PDF含正文、附录A–K和检查清单已读;图2从第6页渲染截取并检查;核对官方项目、仓库递归目录和README,未复现。
来源证据ARCADE150万程序,平均40.3计数操作、8.5实体操作,约1,250万中间状态;扫掠覆盖25.37%。 3.2表1;附录G/H ↗
我的解读操作词汇并非新发明,贡献在组合分布与中间监督;覆盖率指至少出现一次的程序比例,不是操作频率。
来源证据八张H100训练12小时、100k步、batch32;LLM学习率2×10⁻⁴,点编码器5×10⁻⁵,1k暖启后余弦衰减。 附录I.7、A.2 ↗
我的解读这是论文报告的设定,未独立复现;模型尺度消融只比较前10k步最佳验证损失,不能作完整最终重建结论。
来源证据CADBench各来源3,000,ABC每难度2,000;ABC Hard IoU0.73、CD0.0263、有效形状99.95%。 第4节表2 ↗
我的解读有效只说明执行并产出合法几何,不说明形状准确或可制造;不同公开模型训练语料不匹配。
来源证据同数据single-shot IoU0.644,策略0.682,完整搜索0.816;去中间几何为0.632。 表3 ↗
我的解读分开看数据、条件化与搜索;公开CADEvolve0.568与新训练基线差异不能全部归数据。
来源证据局部搜索平均66秒,归因提升+0.142;草图替换41.5%、补切30.3%、连续调参16.2%、删操作12.0%。 附录A.3表6 ↗
我的解读这些份额由每步获胜候选归因相加而来,不是逐项删除的独立因果贡献;连续参数阶段28.4秒最贵。
来源证据全基准IoU≥0.95占39.9%,≥0.99占15.3%;最好对照CADReasoner为27.2%/7.4%。 附录K.1 ↗
我的解读高均值仍留下大量不精确重建;体积门槛不保证细小孔槽或原参数正确。
来源证据在1,200样本上,不用策略权重的搜索使CAD-Recode种子0.597→0.745。 附录K.2 ↗
我的解读说明兼容CadQuery程序可迁移,不代表任意CAD语言和任意依赖结构都可编辑。
来源证据顶点/边/面准确分数20.8/12.6/8.1,优于CADEvolve15.7/9.7/6.4。 附录K.3 ↗
我的解读匹配受参考B-rep分解影响,仍有较大缺口;不能把合法的不同历史简单判成同一结构恢复。
来源证据搜索平均计数操作16.2→31.7;DeepCAD恢复/参考长度比0.91,IoU0.89。 附录K.4–K.5 ↗
我的解读搜索以增加复杂度换拟合,不普遍简化;较短程序也可能只是缺失几何,不能单看长度。
来源证据算法1未显式把父程序加入候选,图10说明从父程序与提案中保留前2。 算法1;图10 ↗
我的解读公开描述存在实现细节差异,发布代码后应核验是否保证每步不退化;本文不把示意伪码当已验证实现。
开放情况与使用许可
官方ghadinehme/StepCAD递归目录仅README和展示图;README明确训练推理代码、检查点和ARCADE-1.5M均Coming soon。项目的Code按钮只是该占位仓库,不能称方法已开源。
LICENSE论文采用arXiv非独占分发许可;图2必要方法评论保留原貌,仅裁去页面正文。仓库未见LICENSE,网页CC BY-SA 4.0不代表训练数据、模型或软件许可。
我的判断
独立分析 · 未复现实验正方 · 为什么值得投入
ABC Hard平均IoU由最好先前基线0.39到0.73,困难几何下的改进较大;策略本身0.52,搜索还有0.21绝对提升。
相同ARCADE数据下去掉中间几何token,IoU0.682→0.632;独立CAD-Recode种子也可由搜索提升0.597→0.745。
反方 · 哪些结论还不够
87.2%是ABC Hard的相对IoU增幅,不是87.2%的绝对精度或参数恢复率;全基准IoU≥0.99仅15.3%。
搜索可能删除有意义但参数不对的倒角以改善体积;平均程序由16.2增至31.7计数操作,不能一概称更简洁或更好编辑。
公开模型基线用不同训练集,单步策略与搜索算力也不同;无误差条,实体匹配协议在本论文只援引外部定义,不能自称已独立复核。
综合判断
证据支持“逐步几何条件+执行后搜索”改善可执行形状重建;把它用于工程资产时,仍须另验尺寸、细节、实体选择和编辑后的行为。
我会先做的验证
建议实验(尚未执行):待代码数据发布后,固定目标网格和计算预算复现完整基准;除IoU/CD外测孔径、壁厚、小圆角、约束与编辑成功率,并分别测试删特征惩罚和精确实体选择能否减少几何指标投机。