

























工业CAD模型尤其是组件对象具有细粒度高、复杂度大的特点,需要更长的参数化CAD序列来定义。现有序列模型在处理长序列时面临挑战,因此Mamba-CAD被提出作为自监督生成模型,专门针对工业中复杂CAD模型的长期参数化序列建模问题,以提升生成能力。
Mamba-CAD采用基于Mamba架构的编码器-解码器框架,并通过CAD重建任务进行预训练,以学习CAD模型的潜在表示。随后,利用该表示指导生成对抗网络产生虚假表示,最终通过解码器将虚假表示恢复为参数化CAD序列,实现完整的生成流程。
为训练Mamba-CAD,研究团队新创建了一个包含77,078个CAD模型的数据集,这些模型均具有更长的参数化CAD序列。综合实验表明,模型在多种评估指标下表现有效,尤其在有效参数化CAD序列的生成长度方面展现出显著优势,代码和数据集已开源。
Computer-Aided Design (CAD) generative modeling has a strong and long-term application in the industry. Recently, the parametric CAD sequence as the design logic of an object has been widely mined by sequence models. However, the industrial CAD models, especially in component objects, are fine-grained and complex, requiring a longer parametric CAD sequence to define. To address the problem, we introduce Mamba-CAD, a self-supervised generative modeling for complex CAD models in the industry, which can model on a longer parametric CAD sequence. Specifically, we first design an encoder-decoder framework based on a Mamba architecture and pair it with a CAD reconstruction task for pre-training to model the latent representation of CAD models; and then we utilize the learned representation to guide a generative adversarial network to produce the fake representation of CAD models, which would be finally recovered into parametric CAD sequences via the decoder of MambaCAD. To train Mamba-CAD, we further create a new dataset consisting of 77,078 CAD models with longer parametric CAD sequences. Comprehensive experiments are conducted to demonstrate the effectiveness of our model under various evaluation metrics, especially in the generation length of valid parametric CAD sequences. The code and dataset can be achieved from https://github.com/Sunny-Hack/Code-for-Mamba-CAD-AAAI-2025-.
此内容由惯性聚合(RSS阅读器)自动聚合整理,仅供阅读参考。 原文来自 — 版权归原作者所有。