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

TL;DR

GeoFusion-CAD通过几何状态空间扩展长序列CAD生成,提升240步命令的准确性。

cs.CV 🔴 高级 2026-03-23 30 次浏览
Xiaolei Zhou Chuangjie Fang Jie Wu Jingyi Yang Boyi Lin Jianwei Zheng
CAD 扩散模型 几何建模 长序列 结构感知

核心发现

方法论

GeoFusion-CAD是一种端到端的扩散框架,利用几何状态空间进行结构感知CAD生成。采用轻量级G-Mamba模块,通过选择性状态转换建模长程结构依赖,能够在长命令序列中实现一致生成。该方法将CAD程序编码为层次树,结合几何与拓扑信息。

关键结果

  • GeoFusion-CAD在DeepCAD-240数据集上实现了73.9的COV和1.12的MMD,显著优于基于Transformer的方法。
  • 在长序列生成中保持高几何保真度和拓扑一致性,超越现有方法。
  • 通过消融实验验证了G-Mamba模块在长序列生成中的关键作用。

研究意义

该研究为下一代CAD建模系统奠定了可扩展的基础,通过在长序列生成中保持高几何保真度和拓扑一致性,解决了现有方法在长序列生成中的退化问题。其在学术界和工业界的影响深远,尤其在复杂设计任务中。

技术贡献

GeoFusion-CAD通过引入几何状态空间和G-Mamba模块,解决了自注意力机制的计算瓶颈,提供了新的理论保证和工程可能性。与现有方法相比,显著提升了长序列生成的效率和一致性。

新颖性

GeoFusion-CAD首次将几何状态空间应用于CAD生成,结合扩散模型和层次树结构,提供了一种新颖的结构感知生成方法。

局限性

  • 在极端复杂的几何和拓扑结构下,可能存在生成不稳定的情况。
  • 对数据集的多样性和规模有一定依赖。

未来方向

未来工作可探索更复杂的几何结构和拓扑关系,进一步优化G-Mamba模块的效率和稳定性,并扩展至其他3D建模领域。

AI 总览摘要

现代3D建模中,参数化CAD设计是关键。然而,现有方法在生成长命令序列时面临挑战,特别是在复杂几何和拓扑依赖下。GeoFusion-CAD提出了一种端到端的扩散框架,通过几何状态空间实现可扩展的结构感知生成。该方法将CAD程序编码为层次树,结合几何与拓扑信息。实验表明,GeoFusion-CAD在DeepCAD-240数据集上实现了优异的性能,尤其在长序列生成中保持高几何保真度和拓扑一致性。该研究为下一代CAD建模系统奠定了可扩展的基础,具有广泛的学术和工业应用潜力。尽管如此,在极端复杂的几何和拓扑结构下,生成稳定性仍需进一步优化。

深度分析

研究背景

参数化CAD设计是现代3D建模的基础,广泛应用于机械、工业和产品设计中。传统方法如Sketch-Extrusion和B-Rep在生成长序列时面临挑战,特别是在复杂几何和拓扑依赖下。

核心问题

现有基于Transformer的CAD序列生成方法由于自注意力机制的计算瓶颈,在长序列生成中表现不佳,难以保持高几何保真度和拓扑一致性。

核心创新

GeoFusion-CAD通过引入几何状态空间和G-Mamba模块,结合扩散模型和层次树结构,提供了一种新颖的结构感知生成方法,解决了现有方法的计算瓶颈。

方法详解

  • �� 将CAD程序编码为层次树,结合几何与拓扑信息。
  • �� 采用轻量级G-Mamba模块,通过选择性状态转换建模长程结构依赖。
  • �� 利用扩散模型实现稳定优化和长序列生成。

实验设计

在DeepCAD-240数据集上进行实验,评估GeoFusion-CAD在长序列生成中的性能。对比基线包括Transformer-based方法,使用COV、MMD等指标。

结果分析

GeoFusion-CAD在DeepCAD-240数据集上实现了73.9的COV和1.12的MMD,显著优于基于Transformer的方法。消融实验验证了G-Mamba模块在长序列生成中的关键作用。

应用场景

GeoFusion-CAD可用于复杂设计任务中的长序列CAD生成,特别是在需要高几何保真度和拓扑一致性的场景中。

局限与展望

在极端复杂的几何和拓扑结构下,可能存在生成不稳定的情况。对数据集的多样性和规模有一定依赖,未来可探索更复杂的几何结构和拓扑关系。

通俗解读 非专业人士也能看懂

想象你在搭建一个乐高模型。每个乐高块就像CAD中的一个命令,组合在一起形成一个完整的模型。GeoFusion-CAD就像一个聪明的助手,它能帮你更快更好地找到合适的乐高块,并把它们组合在一起,即使是很长的搭建步骤也能轻松应对。这样,你就能在短时间内搭建出一个复杂又精美的乐高模型。

简单解释 像给14岁少年讲一样

想象你在玩一个超级复杂的拼图游戏,每块拼图都有自己的形状和颜色。GeoFusion-CAD就像一个超级智能的拼图助手,它能帮你快速找到合适的拼图块,并把它们放在正确的位置。即使是很长的拼图步骤,它也能轻松搞定,让你在短时间内完成整个拼图游戏。是不是很酷?

术语表

GeoFusion-CAD (几何融合CAD)

一种用于参数化3D设计的结构感知扩散框架,结合几何状态空间实现长序列生成。

用于生成长命令序列的CAD程序。

G-Mamba模块

一种轻量级模块,通过选择性状态转换建模长程结构依赖。

在GeoFusion-CAD中用于处理长序列生成。

DeepCAD-240

一个扩展的数据集,包含40到240步的命令序列。

用于评估GeoFusion-CAD在长序列生成中的性能。

COV (覆盖率)

衡量生成模型覆盖真实数据分布的能力。

用于评估GeoFusion-CAD的生成性能。

MMD (最大平均差异)

用于比较两个分布之间的差异。

用于评估GeoFusion-CAD的生成性能。

开放问题 这项研究留下的未解疑问

  • 1 如何在极端复杂的几何和拓扑结构下保持生成稳定性?目前的方法在这些场景下可能表现不佳。
  • 2 如何进一步优化G-Mamba模块的效率和稳定性?
  • 3 如何扩展GeoFusion-CAD至其他3D建模领域?

应用场景

近期应用

复杂设计任务

GeoFusion-CAD可用于复杂设计任务中的长序列CAD生成,特别是在需要高几何保真度和拓扑一致性的场景中。

远期愿景

下一代CAD建模系统

GeoFusion-CAD为下一代CAD建模系统奠定了可扩展的基础,具有广泛的学术和工业应用潜力。

原文摘要

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.

cs.CV cs.GR