用分层离散扩散模型生成更准确的CAD设计,解决传统方法语义错误问题。
Computer-Aided Design Generation by Cascaded Discrete Diffusion Model

- 分两阶段生成:先用离散扩散生成命令,再根据命令条件生成参数。
- 在深达100个符号的DeepCAD数据集上,生成质量超越自回归与连续扩散模型。
- 适合需要高精度、可控性强的工业设计自动化研究者使用。
近年来,深度学习通过将模型表示为离散命令与参数序列,并利用自回归模型或欧氏空间中的连续扩散模型来实现CAD生成。然而,连续扩散在连续欧氏域中扰动表示,无法反映CAD标记固有的离散性和异质性,常导致生成语义无效的符号。为此,我们提出一种分层离散扩散框架,包含命令扩散与条件参数扩散。前向过程直接作用于类别分布,采用精细的转移矩阵:命令使用吸收态转移矩阵逐步污染至指定符号;参数则针对不同属性设计专用核:坐标使用高斯核保证连续性,尺寸使用尺度不变核,布尔值使用先验保持核。反向过程由两个去噪网络实现:基于Transformer的编码器恢复命令,参数网络引入局部自注意力和交叉注意力以支持命令级交互与条件注入。在DeepCAD数据集上的实验表明,该方法在无条件生成指标上优于现有自回归与连续扩散模型,定性结果也验证了其在条件生成任务中的有效可控性。源代码将公开。
原文摘要 · Abstract (English)
Recent deep learning approaches seek to automate CAD creation by representing a model as a sequence of discrete commands and parameters, and then generating them using autoregressive models or continuous diffusion operating in Euclidean embedding space. However, continuous diffusion perturbs representations in a continuous Euclidean domain that does not reflect the inherently discrete and heterogeneous nature of CAD tokens, often producing perturbed representations that map to semantically invalid symbols. To overcome this limitation, we propose a cascaded discrete diffusion framework for CAD generation, which consists of a command diffusion for generating CAD commands and a parameter diffusion conditioned on CAD commands. Unlike isotropic Gaussian perturbation, the forward process of our approach operates directly over categorical token distributions using delicate transition matrices. For commands, we adopt an absorbing-state transition matrix that progressively corrupts tokens to a designated symbol; for parameters, we introduce specific transition matrices tailored to heterogeneous attributes: a Gaussian kernel for coordinate continuity, a scale-invariant kernel for dimensional values, and a prior-preserving kernel for boolean attributes. The reverse process is achieved by two denoising networks: a Transformer-based encoder for command recovery, and a parameter network with extra local self-attention for command-level interaction and cross-attention for conditional injection. Experiments on the DeepCAD dataset show that the proposed approach surpasses existing autoregressive and continuous diffusion models on unconditional generation metrics, while qualitative results validate effective controllability in conditional generation tasks. Source codes will be released.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。