用状态空间模型生成复杂3D设计,支持更长的参数序列。
Mamba-CAD: State Space Model For 3D Computer-Aided Design Generative Modeling
- 基于Mamba架构构建编码器-解码器,预训练建模CAD潜在表示。
- 生成对抗网络产出假表示,经解码后还原出有效参数序列。
- 适用于工业级复杂3D组件生成,适合设计自动化研究者。
计算机辅助设计(CAD)生成建模在工业界有长期应用价值。近期,以参数化CAD序列为设计逻辑的方法被序列模型广泛挖掘。然而,工业级CAD模型尤其是组件对象,具有细粒度与高复杂性,需更长的参数化序列定义。为此,本文提出Mamba-CAD,一种面向工业复杂CAD模型的自监督生成建模方法,可处理更长参数化序列。具体地,首先设计基于Mamba架构的编码器-解码器框架,并结合CAD重建任务进行预训练,以建模CAD模型的潜在表示;随后利用该表示引导生成对抗网络生成虚假表示,再通过MambaCAD解码器恢复为参数化CAD序列。为训练模型,我们构建了一个包含77,078个具有较长参数序列的CAD模型的新数据集。大量实验表明,该模型在多种评估指标下表现优异,尤其在有效参数序列生成长度上显著领先。代码与数据集可在https://github.com/Sunny-Hack/Code-for-Mamba-CAD-AAAI-2025-获取。
原文摘要 · Abstract (English)
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-.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。