高级检索

分层神经编码和离散扩散模型的CAD模型生成方法

CAD Model Generation with Hierarchical Neural Coding and Discrete Diffusion Models

  • 摘要: 针对现有CAD模型生成方法难以同时实现高质量生成与设计意图保持, 对隐式设计约束的处理能力尤为不足的问题, 提出基于离散扩散的层次化CAD生成方法, 将CAD模型表示为三级神经码树, 从局部曲线几何到全局部件排列. 首先使用带自适应掩模跳连接的向量量化变分自编码器提取多层级设计特征; 然后结合分层噪声调度策略动态地调整各层级码本的噪声分布; 最后通过两阶段级联自回归Transformer, 实现从码树到完整CAD命令序列的转换. 在扩展DeepCAD数据集上的大量实验结果表明, 所提方法在无条件生成任务中表现卓越, 各个指标上达到最优和次优效果. 尤其在点云评估指标Rcoverage达到89.15%, DJSD达到0.61, 均超越现有先进方法. 同时, 在条件生成任务中展现出更强的设计意图保持能力, 且用户调研进一步验证了方法在设计意图保持能力方面的有效性, 为CAD自动化提供了新的解决方案.

     

    Abstract: To address the challenge that existing CAD model generation methods struggle to achieve both high-quality generation and design intent preservation, particularly showing insufficient capability in handling implicit design constraints, this paper proposes a hierarchical CAD generation method based on discrete diffusion. The method represents CAD models as three-level neural code trees, spanning from local curve geometry to global component arrangement. First, a vector quantized variational autoencoder with adaptive mask skip connections was employed to extract multi-level design features. Then, a hierarchical noise scheduling strategy was combined to dynamically adjust the noise distribution of codebooks at each level. Finally, a two-stage cascaded autoregressive Transformer was utilized to convert code trees into complete CAD command sequences. Extensive experiments on the extended DeepCAD dataset demonstrate that the proposed method exhibits excellent performance in unconditional generation tasks, achieving optimal and sub-optimal results across all metrics. Notably, the method achieves 89.15% in point cloud evaluation metric Rcoverage and 0.61 in DJSD, both surpassing existing state-of-the-art methods. Furthermore, the method shows stronger design intent preservation capability in conditional generation tasks. User studies further validate the effectiveness of the method in maintaining design intent, providing a novel solution for CAD automation.

     

/

返回文章
返回