Mamba-CAD：用于3D计算机辅助设计生成建模的状态空间模型
计算机视觉与模式识别
2026-03-03 v1 人工智能
摘要
计算机辅助设计（CAD）生成建模在行业内具有强烈且长期的应用前景。最近，参数化CAD序列作为物体设计逻辑受到序列模型的广泛关注。然而，工业CAD模型，特别是组件对象，细粒度且复杂，需要更长的参数化CAD序列才能定义。为此，本文引入Mamba-CAD，这是一种用于工业复杂CAD模型的自监督生成建模方法，可建模更长的参数化CAD序列。具体而言，我们首先基于Mamba架构设计一个编码器-解码器框架，并配合CAD重建任务进行预训练以建模CAD模型的潜在表示；随后利用所学表示来指导生成对抗网络生成CAD模型的假表示，这些表示最终通过MambaCAD的解码器恢复为参数化CAD序列。为训练Mamba-CAD，我们进一步创建了一个包含77,078个具有更长参数化CAD序列的新数据集。进行全面实验表明，在各种评估指标上，本模型都表现出色，尤其是在有效参数化CAD序列生成长度方面。代码和数据集可从 https://github.com/Sunny-Hack/Code-for-Mamba-CAD-AAAI-2025- 获取。
引用
@article{arxiv.2603.00439,
title = {Mamba-CAD: State Space Model For 3D Computer-Aided Design Generative Modeling},
author = {Xueyang Li and Yunzhong Lou and Yu Song and Xiangdong Zhou},
journal= {arXiv preprint arXiv:2603.00439},
year = {2026}
}
备注
Accepted to AAAI 2025