arXiv:2603.21978cs.CVcs.GR2026-03中稿 · CVPR

用几何状态空间扩散模型生成复杂3D设计,解决长序列命令生成难题。

GeoFusion-CAD: Structure-Aware Diffusion with Geometric State Space for Parametric 3D Design

  • 将CAD指令建模为分层树结构,通过状态空间扩散统一捕捉几何与拓扑关系。
  • 在长度达240的指令序列上保持高精度,优于主流Transformer模型。
  • 适合需要生成复杂、长序列参数化3D设计的工程与自动化建模场景。

参数化计算机辅助设计(CAD)是现代3D建模的核心,但现有方法在处理长命令序列时仍面临挑战,尤其在复杂的几何与拓扑依赖下表现不佳。尽管基于Transformer的架构因其强大的依赖建模能力占据主导地位,但其二次注意力开销和有限上下文窗口限制了其在长程序上的扩展性。本文提出GeoFusion-CAD,一种端到端的扩散框架,实现可扩展且结构感知的生成。该方法将CAD程序编码为分层树结构,在状态空间扩散过程中联合捕捉几何与拓扑信息。具体而言,轻量级C-Mamba模块通过选择性状态转移建模长程结构依赖,支持跨长序列的连贯生成。为支持长序列评估,我们引入DeepCAD-240基准,将序列长度从40扩展至240,同时保留ABC数据集中的草图-拉伸语义。大量实验表明,GeoFusion-CAD在短序列和长序列上均表现优异,保持高几何保真度和拓扑一致性,而传统Transformer模型在此类任务中性能下降。本方法在长序列参数化CAD生成任务中达到新最佳水平,为下一代CAD建模系统奠定可扩展基础。代码与数据集已开源于GitHub。

原文摘要 · Abstract (English)

Parametric Computer-Aided Design (CAD) is fundamental to modern 3D modeling, yet existing methods struggle to generate long command sequences, especially under complex geometric and topological dependencies. Transformer-based architectures dominate CAD sequence generation due to their strong dependency modeling, but their quadratic attention cost and limited context windowing hinder scalability to long programs. We propose GeoFusion-CAD, an end-to-end diffusion framework for scalable and structure-aware generation. Our proposal encodes CAD programs as hierarchical trees, jointly capturing geometry and topology within a state-space diffusion process. Specifically, a lightweight C-Mamba block models long-range structural dependencies through selective state transitions, enabling coherent generation across extended command sequences. To support long-sequence evaluation, we introduce DeepCAD-240, an extended benchmark that increases the sequence length ranging from 40 to 240 while preserving sketch-extrusion semantics from the ABC dataset. Extensive experiments demonstrate that GeoFusion-CAD achieves superior performance on both short and long command ranges, maintaining high geometric fidelity and topological consistency where Transformer-based models degrade. Our approach sets new state-of-the-art scores for long-sequence parametric CAD generation, establishing a scalable foundation for next-generation CAD modeling systems. Code and datasets are available at GitHub.

3D设计扩散模型CAD生成长序列建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。